SPAWNRYS/bloxstrap
1
0
Fork 0
mirror of https://github.com/bloxstraplabs/bloxstrap.git synced 2025-04-21 10:01:27 -07:00
Code Issues Actions Packages Projects Releases Wiki Activity

use cancellation tokens for connectivity check

Browse Source
This commit is contained in:
bluepilledgreat 2024-08-23 18:35:22 +01:00
parent d15e910904
commit 56016e9723
1 changed files with 15 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
Bloxstrap/RobloxDeployment.cs
View File

@ -18,24 +18,26 @@
{ "https://s3.amazonaws.com/setup.roblox.com", 4 }
};
private static async Task<string?> TestConnection(string url, int priority)
private static async Task<string?> TestConnection(string url, int priority, CancellationToken token)
{
string LOG_IDENT = $"RobloxDeployment::TestConnection.{url}";
await Task.Delay(priority * 1000);
if (BaseUrl is not null)
return null;
await Task.Delay(priority * 1000, token);
App.Logger.WriteLine(LOG_IDENT, "Connecting...");
try
{
var response = await App.HttpClient.GetAsync($"{url}/version");
var response = await App.HttpClient.GetAsync($"{url}/version", token);
if (!response.IsSuccessStatusCode)
throw new HttpResponseException(response);
}
catch (TaskCanceledException)
{
App.Logger.WriteLine(LOG_IDENT, "Connectivity test cancelled.");
throw;
}
catch (Exception ex)
{
App.Logger.WriteException(LOG_IDENT, ex);
@ -59,8 +61,11 @@
if (!String.IsNullOrEmpty(BaseUrl))
return null;
CancellationTokenSource tokenSource = new CancellationTokenSource();
CancellationToken token = tokenSource.Token;
var exceptions = new List<Exception>();
var tasks = (from entry in BaseUrls select TestConnection(entry.Key, entry.Value)).ToList();
var tasks = (from entry in BaseUrls select TestConnection(entry.Key, entry.Value, token)).ToList();
App.Logger.WriteLine(LOG_IDENT, "Testing connectivity...");
@ -79,6 +84,9 @@
break;
}
// stop other running connectivity tests
tokenSource.Cancel();
if (String.IsNullOrEmpty(BaseUrl))
return exceptions[0];