Browse Source

Added Sharding support

tags/1.0-rc
matt 9 years ago
parent
commit
80bfe56322
3 changed files with 8 additions and 4 deletions
  1. +4
    -2
      src/Discord.Net/API/DiscordSocketApiClient.cs
  2. +2
    -0
      src/Discord.Net/API/Gateway/IdentifyParams.cs
  3. +2
    -2
      src/Discord.Net/DiscordSocketClient.cs

+ 4
- 2
src/Discord.Net/API/DiscordSocketApiClient.cs View File

@@ -179,7 +179,7 @@ namespace Discord.API
{ {
return await SendAsync<GetGatewayResponse>("GET", "gateway", options: options).ConfigureAwait(false); return await SendAsync<GetGatewayResponse>("GET", "gateway", options: options).ConfigureAwait(false);
} }
public async Task SendIdentifyAsync(int largeThreshold = 100, bool useCompression = true, RequestOptions options = null)
public async Task SendIdentifyAsync(int largeThreshold = 100, bool useCompression = true, int shardID = 0, int totalShards = 1, RequestOptions options = null)
{ {
var props = new Dictionary<string, string> var props = new Dictionary<string, string>
{ {
@@ -190,8 +190,10 @@ namespace Discord.API
Token = _authToken, Token = _authToken,
Properties = props, Properties = props,
LargeThreshold = largeThreshold, LargeThreshold = largeThreshold,
UseCompression = useCompression
UseCompression = useCompression,
ShardingParams = totalShards > 1 ? new int[] { shardID, totalShards } : null
}; };

await SendGatewayAsync(GatewayOpCode.Identify, msg, options: options).ConfigureAwait(false); await SendGatewayAsync(GatewayOpCode.Identify, msg, options: options).ConfigureAwait(false);
} }
public async Task SendResumeAsync(string sessionId, int lastSeq, RequestOptions options = null) public async Task SendResumeAsync(string sessionId, int lastSeq, RequestOptions options = null)


+ 2
- 0
src/Discord.Net/API/Gateway/IdentifyParams.cs View File

@@ -14,5 +14,7 @@ namespace Discord.API.Gateway
public int LargeThreshold { get; set; } public int LargeThreshold { get; set; }
[JsonProperty("compress")] [JsonProperty("compress")]
public bool UseCompression { get; set; } public bool UseCompression { get; set; }
[JsonProperty("shard", NullValueHandling = NullValueHandling.Ignore)]
public int[] ShardingParams { get; set; }
} }
} }

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

@@ -166,7 +166,7 @@ namespace Discord
if (_sessionId != null) if (_sessionId != null)
await ApiClient.SendResumeAsync(_sessionId, _lastSeq).ConfigureAwait(false); await ApiClient.SendResumeAsync(_sessionId, _lastSeq).ConfigureAwait(false);
else else
await ApiClient.SendIdentifyAsync().ConfigureAwait(false);
await ApiClient.SendIdentifyAsync(shardID: ShardId, totalShards:TotalShards).ConfigureAwait(false);


await _connectTask.Task.ConfigureAwait(false); await _connectTask.Task.ConfigureAwait(false);
_canReconnect = true; _canReconnect = true;
@@ -534,7 +534,7 @@ namespace Discord


_sessionId = null; _sessionId = null;
_lastSeq = 0; _lastSeq = 0;
await ApiClient.SendIdentifyAsync().ConfigureAwait(false);
await ApiClient.SendIdentifyAsync(shardID: ShardId, totalShards: TotalShards).ConfigureAwait(false);
} }
break; break;
case GatewayOpCode.Reconnect: case GatewayOpCode.Reconnect:


Loading…
Cancel
Save