Merge pull request #1132 from qzwzqty/master

Passes the Endpoint of the client before it disconnected
dotnet · Apr 2, 2021 · 10f7d80 · 10f7d80
2 parents 324fb7e + 85b339e
commit 10f7d80
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 7 deletions.
diff --git a/Source/MQTTnet/Server/MqttClientSessionsManager.cs b/Source/MQTTnet/Server/MqttClientSessionsManager.cs
@@ -1,4 +1,4 @@
-using MQTTnet.Adapter;
+using MQTTnet.Adapter;
 using MQTTnet.Diagnostics;
 using MQTTnet.Exceptions;
 using MQTTnet.Formatter;
@@ -237,11 +237,13 @@ public async Task CleanUpClient(string clientId, IMqttChannelAdapter channelAdap
                 }
             }
 
+            var endpoint = channelAdapter.Endpoint;
+
             await SafeCleanupChannelAsync(channelAdapter).ConfigureAwait(false);
 
             if (clientId != null)
             {
-                await _eventDispatcher.SafeNotifyClientDisconnectedAsync(clientId, disconnectType).ConfigureAwait(false);
+                await _eventDispatcher.SafeNotifyClientDisconnectedAsync(clientId, disconnectType, endpoint).ConfigureAwait(false);
             }
         }
 

diff --git a/Source/MQTTnet/Server/MqttServerClientDisconnectedEventArgs.cs b/Source/MQTTnet/Server/MqttServerClientDisconnectedEventArgs.cs
@@ -1,13 +1,14 @@
-using System;
+using System;
 
 namespace MQTTnet.Server
 {
     public class MqttServerClientDisconnectedEventArgs : EventArgs
     {
-        public MqttServerClientDisconnectedEventArgs(string clientId, MqttClientDisconnectType disconnectType)
+        public MqttServerClientDisconnectedEventArgs(string clientId, MqttClientDisconnectType disconnectType, string endpoint)
         {
             ClientId = clientId ?? throw new ArgumentNullException(nameof(clientId));
             DisconnectType = disconnectType;
+            Endpoint = endpoint;
         }
 
         /// <summary>
@@ -17,5 +18,7 @@ public MqttServerClientDisconnectedEventArgs(string clientId, MqttClientDisconne
         public string ClientId { get; }
 
         public MqttClientDisconnectType DisconnectType { get; }
+
+        public string Endpoint { get; }
     }
 }
diff --git a/Source/MQTTnet/Server/MqttServerEventDispatcher.cs b/Source/MQTTnet/Server/MqttServerEventDispatcher.cs
@@ -1,4 +1,4 @@
-using MQTTnet.Client.Receiving;
+using MQTTnet.Client.Receiving;
 using MQTTnet.Diagnostics;
 using System;
 using System.Threading.Tasks;
@@ -44,7 +44,7 @@ public async Task SafeNotifyClientConnectedAsync(string clientId)
             }
         }
 
-        public async Task SafeNotifyClientDisconnectedAsync(string clientId, MqttClientDisconnectType disconnectType)
+        public async Task SafeNotifyClientDisconnectedAsync(string clientId, MqttClientDisconnectType disconnectType, string endpoint)
         {
             try
             {
@@ -54,7 +54,7 @@ public async Task SafeNotifyClientDisconnectedAsync(string clientId, MqttClientD
                     return;
                 }
 
-                await handler.HandleClientDisconnectedAsync(new MqttServerClientDisconnectedEventArgs(clientId, disconnectType)).ConfigureAwait(false);
+                await handler.HandleClientDisconnectedAsync(new MqttServerClientDisconnectedEventArgs(clientId, disconnectType, endpoint)).ConfigureAwait(false);
             }
             catch (Exception exception)
             {