aljaz/btcpayserver
1
0
Fork 0
mirror of https://github.com/aljazceru/btcpayserver.git synced 2025-12-17 05:54:26 +01:00
Code Issues Packages Projects Releases Wiki Activity

Keep websocket connection alive on notifications and vault

Browse Source
This commit is contained in:
nicolas.dorier
2021-10-05 14:29:49 +09:00
parent 5c8e62bd90
commit b0ef98dd63
2 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
BTCPayServer/Controllers/NotificationsController.cs
View File

@@ -1,5 +1,6 @@
using System; using System;
using System.Linq; using System.Linq;
using System.Net.WebSockets;
using System.Threading; using System.Threading;
using System.Threading.Tasks; using System.Threading.Tasks;
using BTCPayServer.Abstractions.Constants; using BTCPayServer.Abstractions.Constants;
@@ -69,14 +70,15 @@ namespace BTCPayServer.Controllers
} }
}); });
while (!cancellationToken.IsCancellationRequested) await websocketHelper.NextMessageAsync(cancellationToken);
{
await Task.Delay(2000, cancellationToken);
}
} }
catch (TaskCanceledException) catch (TaskCanceledException)
{ {
// ignored // ignored
}
catch (WebSocketException)
{
} }
finally finally
{ {

4
BTCPayServer/WebsocketExtensions.cs
View File

@@ -23,10 +23,10 @@ wait:
if (completed == waiting) if (completed == waiting)
{ {
await webSocket.SendAsync(Encoding.UTF8.GetBytes("ping"), WebSocketMessageType.Text, true, cancellationToken); await webSocket.SendAsync(Encoding.UTF8.GetBytes("ping"), WebSocketMessageType.Text, true, cancellationToken);
waiting = Task.Delay(TimeSpan.FromSeconds(30), cancellationToken); waiting = Task.Delay(TimeSpan.FromSeconds(25), cancellationToken);
goto wait; goto wait;
} }
return receiving.Result; return await receiving;
} }
} }
} }