Browse Source

Use cancelTokens in LargeServerDownloader

pull/57/head
RogueException 9 years ago
parent
commit
7606dd650f
2 changed files with 7 additions and 7 deletions
  1. +5
    -5
      src/Discord.Net.Shared/TaskExtensions.cs
  2. +2
    -2
      src/Discord.Net/DiscordClient.cs

+ 5
- 5
src/Discord.Net.Shared/TaskExtensions.cs View File

@@ -53,15 +53,15 @@ namespace Discord
} }
} }


public static async Task Wait(this CancellationTokenSource tokenSource)
public static async Task Wait(this CancellationTokenSource cancelTokenSource)
{ {
var token = tokenSource.Token;
try { await Task.Delay(-1, token).ConfigureAwait(false); }
var cancelToken = cancelTokenSource.Token;
try { await Task.Delay(-1, cancelToken).ConfigureAwait(false); }
catch (OperationCanceledException) { } //Expected catch (OperationCanceledException) { } //Expected
} }
public static async Task Wait(this CancellationToken token)
public static async Task Wait(this CancellationToken cancelToken)
{ {
try { await Task.Delay(-1, token).ConfigureAwait(false); }
try { await Task.Delay(-1, cancelToken).ConfigureAwait(false); }
catch (OperationCanceledException) { } //Expected catch (OperationCanceledException) { } //Expected
} }
} }


+ 2
- 2
src/Discord.Net/DiscordClient.cs View File

@@ -1079,7 +1079,7 @@ namespace Discord
ulong[] serverIds = new ulong[batchSize]; ulong[] serverIds = new ulong[batchSize];


while (!cancelToken.IsCancellationRequested && State == ConnectionState.Connecting) while (!cancelToken.IsCancellationRequested && State == ConnectionState.Connecting)
await Task.Delay(100).ConfigureAwait(false);
await Task.Delay(1000, cancelToken).ConfigureAwait(false);


while (!cancelToken.IsCancellationRequested && State == ConnectionState.Connected) while (!cancelToken.IsCancellationRequested && State == ConnectionState.Connected)
{ {
@@ -1092,7 +1092,7 @@ namespace Discord
if (count > 0) if (count > 0)
GatewaySocket.SendRequestMembers(serverIds.Take(count), "", 0); GatewaySocket.SendRequestMembers(serverIds.Take(count), "", 0);
} }
await Task.Delay(1250).ConfigureAwait(false);
await Task.Delay(1250, cancelToken).ConfigureAwait(false);
} }
} }
catch (OperationCanceledException) { } catch (OperationCanceledException) { }


Loading…
Cancel
Save