Add support for WebSocket server via ASP.NET Core 2.0

2026-10-16 17:23:04 +08:00 · 2017-10-21 00:14:09 +02:00
parent 11ade8c3df
commit f9eb6ed85e
10 changed files with 219 additions and 9 deletions
@@ -0,0 +1,95 @@
+using System;
+using System.IO;
+using System.Net.WebSockets;
+using System.Threading;
+using System.Threading.Tasks;
+using MQTTnet.Core.Adapter;
+using MQTTnet.Core.Channel;
+using MQTTnet.Core.Diagnostics;
+using MQTTnet.Core.Serializer;
+using MQTTnet.Core.Server;
+using MQTTnet.Implementations;
+
+namespace MQTTnet.TestApp.AspNetCore2
+{
+    public class MqttWebSocketServerAdapter : IMqttServerAdapter, IDisposable
+    {
+        private readonly MqttNetTrace _trace;
+
+        public MqttWebSocketServerAdapter(MqttNetTrace trace)
+        {
+            _trace = trace ?? throw new ArgumentNullException(nameof(trace));
+        }
+
+        public event EventHandler<MqttServerAdapterClientAcceptedEventArgs> ClientAccepted;
+
+        public Task StartAsync(MqttServerOptions options)
+        {
+            return Task.CompletedTask;
+        }
+
+        public Task StopAsync()
+        {
+            return Task.FromResult(0);
+        }
+
+        public Task AcceptWebSocketAsync(WebSocket webSocket)
+        {
+            if (webSocket == null) throw new ArgumentNullException(nameof(webSocket));
+
+            var channel = new MqttWebSocketServerChannel(webSocket);
+            var clientAdapter = new MqttChannelCommunicationAdapter(channel, new MqttPacketSerializer(), _trace);
+
+            var eventArgs = new MqttServerAdapterClientAcceptedEventArgs(clientAdapter);
+            ClientAccepted?.Invoke(this, eventArgs);
+            return eventArgs.SessionTask;
+        }
+        
+        public void Dispose()
+        {
+            StopAsync();
+        }
+
+        private class MqttWebSocketServerChannel : IMqttCommunicationChannel, IDisposable
+        {
+            private readonly WebSocket _webSocket;
+
+            public MqttWebSocketServerChannel(WebSocket webSocket)
+            {
+                _webSocket = webSocket ?? throw new ArgumentNullException(nameof(webSocket));
+
+                RawReceiveStream = new WebSocketStream(_webSocket);
+            }
+
+            public Stream SendStream => RawReceiveStream;
+            public Stream ReceiveStream => RawReceiveStream;
+            public Stream RawReceiveStream { get; }
+
+            public Task ConnectAsync()
+            {
+                return Task.CompletedTask;
+            }
+
+            public Task DisconnectAsync()
+            {
+                RawReceiveStream?.Dispose();
+
+                if (_webSocket == null)
+                {
+                    return Task.CompletedTask;
+                }
+
+                return _webSocket.CloseAsync(WebSocketCloseStatus.NormalClosure, string.Empty, CancellationToken.None);
+            }
+
+            public void Dispose()
+            {
+                RawReceiveStream?.Dispose();
+                SendStream?.Dispose();
+                ReceiveStream?.Dispose();
+
+                _webSocket?.Dispose();
+            }
+        }
+    }
+}