Fixing reconnector issues

Author: Elad Zelingher

src/net45/Extensions/WampSharp.WebSockets/WebSockets/WebSocketWrapperConnection.cs

@@ -129,6 +129,7 @@ await mWebSocket.CloseAsync(WebSocketCloseStatus.NormalClosure,
             catch (Exception ex)
             {
                 RaiseConnectionError(ex);
+                RaiseConnectionClosed();
             }
         }
 

src/net45/WampSharp/WAMP2/V2/Api/WampChannelReconnector.cs

@@ -17,6 +17,7 @@ public class WampChannelReconnector : IDisposable
         private IDisposable mDisposable = Disposable.Empty;
         private bool mStarted = false;
         private readonly object mLock = new object();
+        private IDisposable mConnectionBrokenDisposable;
 
         /// <summary>
         /// Initializes a new instance of <see cref="WampChannelReconnector"/>.
@@ -29,20 +30,22 @@ public WampChannelReconnector(IWampChannel channel, Func<Task> connector)
 
             var connectionBrokenObservable =
                 Observable.FromEventPattern<WampSessionCloseEventArgs>
-                    (x => monitor.ConnectionBroken += x,
-                        x => monitor.ConnectionBroken -= x)
-                    .Select(x => Unit.Default);
+                          (x => monitor.ConnectionBroken += x,
+                           x => monitor.ConnectionBroken -= x)
+                          .Select(x => Unit.Default)
+                          .Replay(1);
 
             var onceAndConnectionBroken =
-                Observable.Return(Unit.Default).Concat
-                    (connectionBrokenObservable);
+                connectionBrokenObservable.StartWith(Unit.Default);
 
             IObservable<IObservable<Unit>> reconnect =
                 from connectionBroke in onceAndConnectionBroken
                 let tryReconnect = Observable.FromAsync(connector)
                     .Catch<Unit, Exception>(x => Observable.Empty<Unit>())
                 select tryReconnect;
 
+            mConnectionBrokenDisposable = connectionBrokenObservable.Connect();
+
             mMerged = reconnect.Concat();
         }
 
@@ -78,6 +81,7 @@ public void Dispose()
             {
                 mMerged = null;
                 mDisposable.Dispose();
+                mConnectionBrokenDisposable.Dispose();
             }
         }
     }