From 316a4d76886e117659e98328a5c6aeeb458f7bb1 Mon Sep 17 00:00:00 2001
From: Savorboard <yangxiaodong1214@126.com>
Date: Fri, 14 Jun 2019 10:38:33 +0800
Subject: [PATCH]  Improved RabbitMQOptions to provide extensions option to
 configure the client original configuration. #350

---
 .../CAP.RabbiMQOptions.cs                     | 29 ++++++-------------
 .../IConnectionChannelPool.Default.cs         | 10 +++----
 2 files changed, 13 insertions(+), 26 deletions(-)

diff --git a/src/DotNetCore.CAP.RabbitMQ/CAP.RabbiMQOptions.cs b/src/DotNetCore.CAP.RabbitMQ/CAP.RabbiMQOptions.cs
index afb2fd2..d52fe61 100644
--- a/src/DotNetCore.CAP.RabbitMQ/CAP.RabbiMQOptions.cs
+++ b/src/DotNetCore.CAP.RabbitMQ/CAP.RabbiMQOptions.cs
@@ -2,15 +2,14 @@
 // Licensed under the MIT License. See License.txt in the project root for license information.
 
 // ReSharper disable once CheckNamespace
+
+using System;
+using RabbitMQ.Client;
+
 namespace DotNetCore.CAP
 {
     public class RabbitMQOptions
     {
-        /// <summary>
-        /// Default value for connection attempt timeout, in milliseconds.
-        /// </summary>
-        public const int DefaultConnectionTimeout = 30 * 1000;
-
         /// <summary>
         /// Default password (value: "guest").
         /// </summary>
@@ -63,21 +62,6 @@ namespace DotNetCore.CAP
         /// </summary>
         public string ExchangeName { get; set; } = DefaultExchangeName;
 
-        /// <summary>
-        /// Timeout setting for connection attempts (in milliseconds).
-        /// </summary>
-        public int RequestedConnectionTimeout { get; set; } = DefaultConnectionTimeout;
-
-        /// <summary>
-        /// Timeout setting for socket read operations (in milliseconds).
-        /// </summary>
-        public int SocketReadTimeout { get; set; } = DefaultConnectionTimeout;
-
-        /// <summary>
-        /// Timeout setting for socket write operations (in milliseconds).
-        /// </summary>
-        public int SocketWriteTimeout { get; set; } = DefaultConnectionTimeout;
-
         /// <summary>
         /// The port to connect on.
         /// </summary>
@@ -87,5 +71,10 @@ namespace DotNetCore.CAP
         /// Gets or sets queue message automatic deletion time (in milliseconds). Default 864000000 ms (10 days).
         /// </summary>
         public int QueueMessageExpires { get; set; } = 864000000;
+
+        /// <summary>
+        /// RabbitMQ native connection factory options
+        /// </summary>
+        public Action<ConnectionFactory> ConnectionFactoryOptions { get; set; }
     }
 }
\ No newline at end of file
diff --git a/src/DotNetCore.CAP.RabbitMQ/IConnectionChannelPool.Default.cs b/src/DotNetCore.CAP.RabbitMQ/IConnectionChannelPool.Default.cs
index 7b48a01..0b39f39 100644
--- a/src/DotNetCore.CAP.RabbitMQ/IConnectionChannelPool.Default.cs
+++ b/src/DotNetCore.CAP.RabbitMQ/IConnectionChannelPool.Default.cs
@@ -6,7 +6,6 @@ using System.Collections.Concurrent;
 using System.Diagnostics;
 using System.Threading;
 using Microsoft.Extensions.Logging;
-using Newtonsoft.Json;
 using RabbitMQ.Client;
 
 namespace DotNetCore.CAP.RabbitMQ
@@ -42,7 +41,7 @@ namespace DotNetCore.CAP.RabbitMQ
                 Exchange = options.ExchangeName + "." + capOptions.Version;
             }
 
-            _logger.LogDebug("RabbitMQ configuration of CAP :{0} {1}", Environment.NewLine, JsonConvert.SerializeObject(options, Formatting.Indented));
+            _logger.LogDebug($"RabbitMQ configuration:'HostName:{options.HostName}, Port:{options.Port}, UserName:{options.UserName}, Password:{options.Password}, ExchangeName:{options.ExchangeName}'");
         }
 
         IModel IConnectionChannelPool.Rent()
@@ -88,19 +87,18 @@ namespace DotNetCore.CAP.RabbitMQ
                 UserName = options.UserName,
                 Port = options.Port,
                 Password = options.Password,
-                VirtualHost = options.VirtualHost,
-                RequestedConnectionTimeout = options.RequestedConnectionTimeout,
-                SocketReadTimeout = options.SocketReadTimeout,
-                SocketWriteTimeout = options.SocketWriteTimeout
+                VirtualHost = options.VirtualHost
             };
 
             if (options.HostName.Contains(","))
             {
+                options.ConnectionFactoryOptions?.Invoke(factory);
                 return () => factory.CreateConnection(
                     options.HostName.Split(new[] { "," }, StringSplitOptions.RemoveEmptyEntries));
             }
 
             factory.HostName = options.HostName;
+            options.ConnectionFactoryOptions?.Invoke(factory);
             return () => factory.CreateConnection();
         }