INSANEMODE/IW4M-Admin
1
0
Fork 0
mirror of https://github.com/RaidMax/IW4M-Admin.git synced 2025-06-16 10:09:14 -05:00
Code Issues Releases Wiki Activity

add cancellation token for rcon connection to allow more granular control

Browse Source
This commit is contained in:
RaidMax
2022-02-28 20:44:30 -06:00
parent c22bfed99d
commit 57558f6223
17 changed files with 187 additions and 132 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
Integrations/Source/SourceRConConnection.cs
View File

@ -48,12 +48,12 @@ namespace Integrations.Source
_activeQuery.Dispose();
}
public async Task<string[]> SendQueryAsync(StaticHelpers.QueryType type, string parameters = "")
public async Task<string[]> SendQueryAsync(StaticHelpers.QueryType type, string parameters = "", CancellationToken token = default)
{
try
{
await _activeQuery.WaitAsync();
await WaitForAvailable();
await _activeQuery.WaitAsync(token);
await WaitForAvailable(token);
if (_needNewSocket)
{
@ -66,7 +66,7 @@ namespace Integrations.Source
// ignored
}
await Task.Delay(ConnectionTimeout);
await Task.Delay(ConnectionTimeout, token);
_rconClient = _rconClientFactory.CreateClient(_ipEndPoint);
_authenticated = false;
_needNewSocket = false;
@ -147,12 +147,12 @@ namespace Integrations.Source
}
}
private async Task WaitForAvailable()
private async Task WaitForAvailable(CancellationToken token)
{
var diff = DateTime.Now - _lastQuery;
if (diff < FloodDelay)
{
await Task.Delay(FloodDelay - diff);
await Task.Delay(FloodDelay - diff, token);
}
}