Move the release of the object inside the Return method (#969)

src/DotNetCore.CAP.Kafka/IConnectionPool.Default.cs

@@ -60,6 +60,8 @@ namespace DotNetCore.CAP.Kafka
                 return true;
             }
 
+            producer.Dispose();
+
             Interlocked.Decrement(ref _pCount);
 
             return false;

src/DotNetCore.CAP.Kafka/ITransport.Kafka.cs

@@ -64,11 +64,7 @@ namespace DotNetCore.CAP.Kafka
             }
             finally
             {
-                var returned = _connectionPool.Return(producer);
-                if (!returned)
-                {
-                    producer.Dispose();
-                }
+                _connectionPool.Return(producer);
             }
         }
     }

src/DotNetCore.CAP.NATS/IConnectionPool.Default.cs

@@ -17,7 +17,7 @@ namespace DotNetCore.CAP.NATS
         private readonly ConnectionFactory _connectionFactory;
         private int _pCount;
         private int _maxSize;
-        
+
         public ConnectionPool(ILogger<ConnectionPool> logger, IOptions<NATSOptions> options)
         {
             _options = options.Value;
@@ -51,7 +51,7 @@ namespace DotNetCore.CAP.NATS
             {
                 connection = _connectionFactory.CreateConnection(_options.Servers);
             }
-            
+
             return connection;
         }
 
@@ -64,6 +64,8 @@ namespace DotNetCore.CAP.NATS
                 return true;
             }
 
+            connection.Dispose();
+
             Interlocked.Decrement(ref _pCount);
 
             return false;

src/DotNetCore.CAP.NATS/ITransport.NATS.cs

@@ -35,8 +35,8 @@ namespace DotNetCore.CAP.NATS
                 {
                     msg.Header[header.Key] = header.Value;
                 }
-                
-                var reply=  connection.Request(msg);
+
+                var reply = connection.Request(msg);
 
                 if (reply.Data != null && reply.Data[0] == 1)
                 {
@@ -54,11 +54,7 @@ namespace DotNetCore.CAP.NATS
             }
             finally
             {
-                var returned = _connectionPool.Return(connection);
-                if (!returned)
-                {
-                    connection.Dispose();
-                }
+                _connectionPool.Return(connection);
             }
         }
     }

src/DotNetCore.CAP.RabbitMQ/IConnectionChannelPool.Default.cs

@@ -105,7 +105,7 @@ namespace DotNetCore.CAP.RabbitMQ
             if (options.HostName.Contains(","))
             {
                 options.ConnectionFactoryOptions?.Invoke(factory);
- 
+
                 return () => factory.CreateConnection(
                     options.HostName.Split(new[] { "," }, StringSplitOptions.RemoveEmptyEntries));
             }
@@ -149,6 +149,8 @@ namespace DotNetCore.CAP.RabbitMQ
                 return true;
             }
 
+            connection.Dispose();
+
             Interlocked.Decrement(ref _count);
 
             Debug.Assert(_maxSize == 0 || _pool.Count <= _maxSize);

src/DotNetCore.CAP.RabbitMQ/ITransport.RabbitMQ.cs

@@ -67,11 +67,7 @@ namespace DotNetCore.CAP.RabbitMQ
             {
                 if (channel != null)
                 {
-                    var returned = _connectionChannelPool.Return(channel);
-                    if (!returned)
-                    {
-                        channel.Dispose();
-                    }
+                    _connectionChannelPool.Return(channel);
                 }
             }
         }