|
4 | 4 | using System.Threading.Tasks;
|
5 | 5 | using System.Web;
|
6 | 6 | using WampSharp.V2;
|
| 7 | +using WampSharp.V2.Realm; |
7 | 8 |
|
8 | 9 | namespace Jojatekok.PoloniexAPI.LiveTools
|
9 | 10 | {
|
@@ -31,14 +32,53 @@ public ObservableDictionary<CurrencyPair, MarketData> Tickers {
|
31 | 32 | public void Start()
|
32 | 33 | {
|
33 | 34 | WampChannel = new DefaultWampChannelFactory().CreateJsonChannel(Helper.ApiUrlWssBase, "realm1");
|
| 35 | + WampChannel.RealmProxy.Monitor.ConnectionBroken += OnConnectionBroken; |
| 36 | + |
34 | 37 | WampChannelOpenTask = WampChannel.Open();
|
35 | 38 | }
|
36 | 39 |
|
37 | 40 | public void Stop()
|
38 | 41 | {
|
| 42 | + foreach (var subscription in ActiveSubscriptions.Values) { |
| 43 | + subscription.Dispose(); |
| 44 | + } |
| 45 | + ActiveSubscriptions.Clear(); |
| 46 | + |
39 | 47 | WampChannel.Close();
|
40 | 48 | }
|
41 | 49 |
|
| 50 | + private void OnConnectionBroken(object sender, WampSessionCloseEventArgs e) |
| 51 | + { |
| 52 | + if (e.CloseType != SessionCloseType.Disconnection) { |
| 53 | + var subscriptions = new string[ActiveSubscriptions.Count]; |
| 54 | + var i = 0; |
| 55 | + foreach (var subjectName in ActiveSubscriptions.Keys) { |
| 56 | + subscriptions[i] = subjectName; |
| 57 | + i++; |
| 58 | + } |
| 59 | + ActiveSubscriptions.Clear(); |
| 60 | + |
| 61 | + // Re-initialize WampChannel |
| 62 | + Start(); |
| 63 | + |
| 64 | + // Re-subscribe to subjects |
| 65 | +#pragma warning disable 4014 |
| 66 | + for (var j = subscriptions.Length - 1; j >= 0; j--) { |
| 67 | + var subjectName = subscriptions[j]; |
| 68 | + switch (subjectName) { |
| 69 | + case SubjectNameTicker: |
| 70 | + SubscribeToTickerAsync(); |
| 71 | + break; |
| 72 | + |
| 73 | + case SubjectNameTrollbox: |
| 74 | + SubscribeToTrollboxAsync(); |
| 75 | + break; |
| 76 | + } |
| 77 | + } |
| 78 | +#pragma warning restore 4014 |
| 79 | + } |
| 80 | + } |
| 81 | + |
42 | 82 | public async Task SubscribeToTickerAsync()
|
43 | 83 | {
|
44 | 84 | if (!ActiveSubscriptions.ContainsKey(SubjectNameTicker)) {
|
|
0 commit comments