ConnectionId default header set for proxy client

Author: gedem

src/NetCoreStack.WebSockets.ProxyClient/ClientWebSocketConnector.cs

@@ -85,9 +85,29 @@ public async Task ConnectAsync()
             }
         }
 
+        private ArraySegment<byte> CreateTextSegment(WebSocketMessageContext context)
+        {
+            if (context == null)
+            {
+                throw new ArgumentNullException(nameof(context));
+            }
+
+            object connectionId = string.Empty;
+            if (context.Header.TryGetValue(SocketsConstants.ConnectionId, out connectionId))
+            {
+                var id = connectionId as string;
+                if (string.IsNullOrEmpty(id))
+                    throw new InvalidOperationException(nameof(connectionId));
+            }
+            else
+                context.Header.Add(SocketsConstants.ConnectionId, ConnectionId);
+
+            return context.ToSegment();
+        }
+
         public async Task SendAsync(WebSocketMessageContext context)
         {
-            var segments = context.ToSegment();
+            var segments = CreateTextSegment(context);
             await _webSocket.SendAsync(segments, WebSocketMessageType.Text, true, CancellationToken.None);
         }
 

src/NetCoreStack.WebSockets/Internal/SocketsConstants.cs

@@ -6,5 +6,6 @@ public static class SocketsConstants
         public const int ChunkSize = 1024 * 4;
         public const string CompressedKey = "Compressed";
         public const string ConnectorName = "ConnectorName";
+        public const string ConnectionId = "ConnectionId";
     }
 }