moved RestartAfterListenError

moved RestartAfterListenError to the WebSocketServer class
statianzo · Aug 24, 2016 · 00866bb · 00866bb
1 parent ca00a01
commit 00866bb
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 9 deletions.
diff --git a/src/Fleck/Interfaces/ISocket.cs b/src/Fleck/Interfaces/ISocket.cs
@@ -15,7 +15,6 @@ public interface ISocket
         int RemotePort { get; }
         Stream Stream { get; }
         bool NoDelay { get; set; }
-        bool RestartAfterListenError { get; set; }
         EndPoint LocalEndPoint { get; }
 
         Task<ISocket> Accept(Action<ISocket> callback, Action<Exception> error);

diff --git a/src/Fleck/SocketWrapper.cs b/src/Fleck/SocketWrapper.cs
@@ -16,7 +16,6 @@ public class SocketWrapper : ISocket
         private Stream _stream;
         private CancellationTokenSource _tokenSource;
         private TaskFactory _taskFactory;
-        private bool _restartAfterListenError;
 
         public string RemoteIpAddress
         {
@@ -87,12 +86,6 @@ public bool NoDelay
             set { _socket.NoDelay = value; }
         }
 
-        public bool RestartAfterListenError
-        {
-            get { return _restartAfterListenError; }
-            set { _restartAfterListenError = value; }
-        }
-
         public EndPoint LocalEndPoint
         {
             get { return _socket.LocalEndPoint; }

diff --git a/src/Fleck/WebSocketServer.cs b/src/Fleck/WebSocketServer.cs
@@ -43,6 +43,7 @@ public WebSocketServer(int port, string location)
         public X509Certificate2 Certificate { get; set; }
         public SslProtocols EnabledSslProtocols { get; set; }
         public IEnumerable<string> SupportedSubProtocols { get; set; }
+        public bool RestartAfterListenError {get; set; }
 
         public bool IsSecure
         {
@@ -101,7 +102,7 @@ private void ListenForClients()
         {
             ListenerSocket.Accept(OnClientConnect, e => {
                 FleckLog.Error("Listener socket is closed", e);
-                if(ListenerSocket.RestartAfterListenError){
+                if(RestartAfterListenError){
                     FleckLog.Info("Listener socket restarting");
                     try
                     {