Handshake transport registration - TryAdd

Author: Gencebay Demir

…ockets/DefaultHandshakeStateTransport.cs …ternal/DefaultHandshakeStateTransport.cssrc/NetCoreStack.WebSockets/DefaultHandshakeStateTransport.cs renamed to src/NetCoreStack.WebSockets/Internal/DefaultHandshakeStateTransport.cs src/NetCoreStack.WebSockets/DefaultHandshakeStateTransport.cs renamed to src/NetCoreStack.WebSockets/Internal/DefaultHandshakeStateTransport.cs

@@ -1,4 +1,5 @@
 using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.DependencyInjection.Extensions;
 using NetCoreStack.WebSockets.Common;
 using NetCoreStack.WebSockets.Internal;
 using System;
@@ -14,7 +15,7 @@ public static void AddNativeWebSockets(this IServiceCollection services)
                 throw new ArgumentNullException(nameof(services));
             }
 
-            services.AddTransient<IHandshakeStateTransport, DefaultHandshakeStateTransport>();
+            services.TryAdd(ServiceDescriptor.Transient<IHandshakeStateTransport, DefaultHandshakeStateTransport>());
             services.AddSingleton<IConnectionManager, ConnectionManager>();
         }
     }

src/NetCoreStack.WebSockets/SocketServiceCollectionExtensions.cs

@@ -0,0 +1,21 @@
+using NetCoreStack.WebSockets.Common;
+using System.Threading.Tasks;
+
+namespace ServerTestApp
+{
+    public class MyHandshakeStateTransport : IHandshakeStateTransport
+    {
+        public Task<object> GetStateAsync()
+        {
+            return Task.FromResult<object>(0);
+        }
+    }
+
+    public class CustomHandshakeStateTransport : IHandshakeStateTransport
+    {
+        public Task<object> GetStateAsync()
+        {
+            return Task.FromResult<object>(0);
+        }
+    }
+}

test/ServerTestApp/MyHandshakeStateTransport.cs

@@ -4,6 +4,7 @@
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Logging;
 using NetCoreStack.WebSockets;
+using NetCoreStack.WebSockets.Common;
 using Swashbuckle.Swagger.Model;
 using System.IO;
 
@@ -38,9 +39,11 @@ public void ConfigureServices(IServiceCollection services)
                 c.DescribeAllEnumsAsStrings();
             });
 
+            services.AddTransient<IHandshakeStateTransport, MyHandshakeStateTransport>();
+
             // Add Net Core Stack socket services.
             services.AddNativeWebSockets();
-
+            
             // Add framework services.
             services.AddMvc();
         }