Add service name to RabbitMQ connection alias name

dotnetcore · Jun 21, 2019 · 9fd0e37 · 9fd0e37
1 parent a02ccc7
commit 9fd0e37
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/DotNetCore.CAP.RabbitMQ/IConnectionChannelPool.Default.cs b/src/DotNetCore.CAP.RabbitMQ/IConnectionChannelPool.Default.cs
@@ -4,6 +4,7 @@
 using System;
 using System.Collections.Concurrent;
 using System.Diagnostics;
+using System.Reflection;
 using System.Threading;
 using Microsoft.Extensions.Logging;
 using RabbitMQ.Client;
@@ -90,24 +91,26 @@ public void Dispose()
 
         private static Func<IConnection> CreateConnection(RabbitMQOptions options)
         {
+            var serviceName = Assembly.GetEntryAssembly()?.GetName().Name.ToLower();
+
             var factory = new ConnectionFactory
             {
                 UserName = options.UserName,
                 Port = options.Port,
                 Password = options.Password,
                 VirtualHost = options.VirtualHost
             };
-
+            
             if (options.HostName.Contains(","))
             {
                 options.ConnectionFactoryOptions?.Invoke(factory);
                 return () => factory.CreateConnection(
-                    options.HostName.Split(new[] { "," }, StringSplitOptions.RemoveEmptyEntries));
+                    options.HostName.Split(new[] { "," }, StringSplitOptions.RemoveEmptyEntries), serviceName);
             }
-
+           
             factory.HostName = options.HostName;
             options.ConnectionFactoryOptions?.Invoke(factory);
-            return () => factory.CreateConnection();
+            return () => factory.CreateConnection(serviceName);
         }
 
         private void RabbitMQ_ConnectionShutdown(object sender, ShutdownEventArgs e)