Try to fix NATS reconnection publish issue after restarting server. #…

…1542
dotnetcore · Jun 11, 2024 · 51453d5 · 51453d5
1 parent 1cd3f40
commit 51453d5
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/src/DotNetCore.CAP.NATS/CAP.NATSOptions.cs b/src/DotNetCore.CAP.NATS/CAP.NATSOptions.cs
@@ -18,7 +18,7 @@ public class NATSOptions
         /// Gets or sets the server url/urls used to connect to the NATs server.
         /// </summary>
         /// <remarks>This may contain username/password information.</remarks>
-        public string Servers { get; set; } = "nats://localhost:4222";
+        public string Servers { get; set; } = "nats://127.0.0.1:4222";
 
         /// <summary>
         /// connection pool size, default is 10

diff --git a/src/DotNetCore.CAP.NATS/DotNetCore.CAP.NATS.csproj b/src/DotNetCore.CAP.NATS/DotNetCore.CAP.NATS.csproj
@@ -7,7 +7,7 @@
   </PropertyGroup>
 
   <ItemGroup>
-    <PackageReference Include="NATS.Client" Version="1.1.1" />
+    <PackageReference Include="NATS.Client" Version="1.1.5" />
   </ItemGroup>
 
   <ItemGroup>

diff --git a/src/DotNetCore.CAP.NATS/IConnectionPool.Default.cs b/src/DotNetCore.CAP.NATS/IConnectionPool.Default.cs
@@ -14,6 +14,7 @@ public class ConnectionPool : IConnectionPool, IDisposable
     {
         private readonly NATSOptions _options;
         private readonly ConcurrentQueue<IConnection> _connectionPool;
+
         private readonly ConnectionFactory _connectionFactory;
         private int _pCount;
         private int _maxSize;
@@ -54,14 +55,17 @@ public IConnection RentConnection()
 
         public bool Return(IConnection connection)
         {
-            if (Interlocked.Increment(ref _pCount) <= _maxSize)
+            if (Interlocked.Increment(ref _pCount) <= _maxSize && connection.State == ConnState.CONNECTED)
             {
                 _connectionPool.Enqueue(connection);
 
                 return true;
             }
 
-            connection.Dispose();
+            if (!connection.IsReconnecting())
+            {
+                connection.Dispose();
+            }
 
             Interlocked.Decrement(ref _pCount);
 
@@ -75,7 +79,6 @@ public void Dispose()
             while (_connectionPool.TryDequeue(out var context))
             {
                 context.Dispose();
-
             }
         }
     }