Command map

Author: gedem

src/NetCoreStack.WebSockets.ProxyClient/ClientWebSocketConnector.cs

@@ -3,6 +3,7 @@
 using NetCoreStack.WebSockets.Interfaces;
 using NetCoreStack.WebSockets.Internal;
 using System;
+using System.Collections.Generic;
 using System.Linq;
 using System.Net.WebSockets;
 using System.Threading;
@@ -31,6 +32,19 @@ public ClientWebSocketConnector(IOptions<ProxyOptions> options,
             Options = options.Value;
         }
 
+        private WebSocketMessageContext CreateConnectionContext()
+        {
+            var context = new WebSocketMessageContext();
+            context.MessageType = WebSocketMessageType.Text;
+            context.Command = WebSocketCommands.Connect;
+            context.Header = new Dictionary<string, object>
+            {
+                ["Name"] = Options.ConnectorName
+            };
+
+            return context;
+        }
+
         public async Task ConnectAsync()
         {
             try

src/NetCoreStack.WebSockets/Internal/InvocatorRegistry.cs

@@ -23,8 +23,8 @@ public List<IWebSocketCommandInvocator> GetInvocators<TOptions>(WebSocketMessage
                 var commands = context.Command.GetUniqueFlags().OfType<WebSocketCommands>().ToList();
                 foreach (var value in commands)
                 {
-                    Type type;
-                    if (options.Map.TryGetValue(value, out type))
+                    List<Type> types = options.Map.Where(c => c.Command == value).Select(x => x.Invocator).ToList();
+                    foreach (var type in types)
                     {
                         var service = _serviceProvider.GetService(type);
                         var invocator = service as IWebSocketCommandInvocator;

src/NetCoreStack.WebSockets/Internal/WebSocketCommandMap.cs

@@ -0,0 +1,16 @@
+using System;
+
+namespace NetCoreStack.WebSockets.Internal
+{
+    public class WebSocketCommandMap
+    {
+        public WebSocketCommands Command { get; }
+        public Type Invocator { get; }
+
+        public WebSocketCommandMap(WebSocketCommands command, Type invocator)
+        {
+            Command = command;
+            Invocator = invocator;
+        }
+    }
+}

src/NetCoreStack.WebSockets/SocketsOptions.cs

@@ -1,18 +1,19 @@
-using System;
+using NetCoreStack.WebSockets.Internal;
+using System;
 using System.Collections.Generic;
 using System.Linq;
 
 namespace NetCoreStack.WebSockets
 {
     public class SocketsOptions
     {
-        internal readonly Dictionary<WebSocketCommands, Type> Map = 
-            new Dictionary<WebSocketCommands, Type>();
+        internal readonly List<WebSocketCommandMap> Map;
 
         public List<Type> Invocators { get; }
 
         public SocketsOptions()
         {
+            Map = new List<WebSocketCommandMap>();
             Invocators = new List<Type>();
         }
 
@@ -22,7 +23,8 @@ protected void Registry(Type invocatorType, WebSocketCommands commands)
             var values = commands.GetUniqueFlags().OfType<WebSocketCommands>().ToList();
             foreach (var value in values)
             {
-                Map.Add(value, invocatorType);
+                var commandMap = new WebSocketCommandMap(value, invocatorType);
+                Map.Add(commandMap);
             }
         }
     }