Write Listener (Rev 4)

Author: scottf

src/main/java/io/nats/client/impl/NatsConnectionWriter.java

@@ -15,7 +15,6 @@
 
 import io.nats.client.Options;
 import io.nats.client.StatisticsCollector;
-import io.nats.client.WriteListener;
 import io.nats.client.support.ByteArrayBuilder;
 
 import java.io.IOException;
@@ -34,7 +33,7 @@
 import static io.nats.client.support.NatsConstants.*;
 
 class NatsConnectionWriter implements Runnable {
-    enum Mode {
+    public enum Mode {
         Normal, Reconnect, WaitingForEndReconnect
     }
     private static final int BUFFER_BLOCK_SIZE = 256;
@@ -179,7 +178,7 @@ void sendMessageBatch(NatsMessage msg, DataPort dataPort, StatisticsCollector st
                 stats.incrementOut(size);
                 if (writeListener != null) {
                     NatsMessage finalMsg = msg;
-                    writeListener.submit(() -> writeListener.buffered(finalMsg));
+                    writeListener.submit(() -> writeListener.buffered(finalMsg, mode.get()));
                 }
 
                 if (msg.flushImmediatelyAfterPublish) {

src/main/java/io/nats/client/WriteListener.java renamed to src/main/java/io/nats/client/impl/WriteListener.java

@@ -11,9 +11,8 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-package io.nats.client;
+package io.nats.client.impl;
 
-import io.nats.client.impl.NatsMessage;
 import org.jspecify.annotations.Nullable;
 
 import java.util.concurrent.ExecutorService;
@@ -36,5 +35,5 @@ public final void submit(Runnable runnable) {
         executorService.submit(runnable);
     }
 
-    public abstract void buffered(NatsMessage msg);
+    public abstract void buffered(NatsMessage msg, NatsConnectionWriter.Mode mode);
 }