Add Publish Timeout on Disconnect

src/NATS.Client.Core/NatsConnection.Publish.cs

@@ -67,7 +67,17 @@ public ValueTask PublishAsync<T>(in NatsMsg<T> msg, INatsSerialize<T>? serialize
 
     private async ValueTask ConnectAndPublishAsync<T>(string subject, T? data, NatsHeaders? headers, string? replyTo, INatsSerialize<T> serializer, CancellationToken cancellationToken)
     {
-        await ConnectAsync().AsTask().WaitAsync(cancellationToken).ConfigureAwait(false);
+        if (Opts.PublishTimeoutOnDisconnected)
+        {
+            using var cts1 = new CancellationTokenSource(Opts.CommandTimeout);
+            using var cts2 = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken, cts1.Token);
+            await ConnectAsync().AsTask().WaitAsync(cts2.Token).ConfigureAwait(false);
+        }
+        else
+        {
+            await ConnectAsync().AsTask().WaitAsync(cancellationToken).ConfigureAwait(false);
+        }
+
         await CommandWriter.PublishAsync(subject, data, headers, replyTo, serializer, cancellationToken).ConfigureAwait(false);
     }
 }

src/NATS.Client.Core/NatsOpts.cs

@@ -206,6 +206,14 @@ public sealed record NatsOpts
     /// </remarks>
     public bool RetryOnInitialConnect { get; init; }
 
+    /// <summary>
+    /// Gets or sets a value indicating whether publish would throw an exception
+    /// when the connection is disconnected and the <see cref="CommandTimeout"/> is reached.
+    /// The default is <c>false</c>, meaning publish will not throw on disconnected state
+    /// and will wait to publish the message until reconnected.
+    /// </summary>
+    public bool PublishTimeoutOnDisconnected { get; set; } = false;
+
     internal NatsUri[] GetSeedUris(bool suppressRandomization = false)
     {
         var urls = Url.Split(',');