preparing WebSocketClient.cs to run as server side mode

alec1o · Nov 15, 2023 · 911e3a3 · 911e3a3
1 parent 1906fc8
commit 911e3a3
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/src/WebSocket/WebSocketClient.cs b/src/WebSocket/WebSocketClient.cs
@@ -22,9 +22,11 @@ public class WebSocketClient : IWebsocketClient
         private EventHandler<Exception> _onError;
         private EventHandler _onOpen;
         private readonly List<(byte[] buffer, BufferType bufferType)> _bufferList;
-        private bool _tryConnecting, _tryClosing;
+        private bool _tryConnecting, _tryClosing, _initServerSide;
+        private readonly bool _serverSide;
         private readonly object _bufferLock;
         private ClientWebSocket _websocket;
+        private WebSocket _ws;
 
 
         public WebSocketClient()
@@ -38,10 +40,22 @@ public WebSocketClient()
             _serverSide = false;
         }
 
+        internal WebSocketClient(WebSocket websocket)
+        {
+            _serverSide = true;
+            _ws = websocket;
+        }
+
+        internal void InitWebSocketServerSide()
+        {
+            if(_initServerSide) return;
+            _ReceiveData();
+            _initServerSide = true;
+        }
 
         public void Open(Uri uri)
         {
-            if (IsOpened || _tryConnecting || _tryClosing) return;
+            if (IsOpened || _tryConnecting || _tryClosing || _serverSide) return;
 
             _tryConnecting = true;