eightball/PluralKit
1
0
Fork 0
mirror of https://github.com/PluralKit/PluralKit.git synced 2026-02-16 10:40:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add basic support for multi-node clustering

Browse source
This commit is contained in:
Ske 2021-06-09 14:49:12 +02:00
parent b09a34d9cd
commit d61f61fead
3 changed files with 53 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

18
Myriad/Gateway/Cluster.cs
View file

@ -32,17 +32,19 @@ namespace Myriad.Gateway
public async Task Start(GatewayInfo.Bot info)
{
var concurrency = GetActualShardConcurrency(info.SessionStartLimit.MaxConcurrency);
_ratelimiter = new(_logger, concurrency);
await Start(info.Url, info.Shards);
await Start(info.Url, 0, info.Shards - 1, info.Shards, info.SessionStartLimit.MaxConcurrency);
}
public async Task Start(string url, int shardCount)
public async Task Start(string url, int shardMin, int shardMax, int shardTotal, int concurrency)
{
_logger.Information("Starting {ShardCount} shards at {Url}", shardCount, url);
for (var i = 0; i < shardCount; i++)
CreateAndAddShard(url, new ShardInfo(i, shardCount));
concurrency = GetActualShardConcurrency(concurrency);
_ratelimiter = new(_logger, concurrency);
var shardCount = shardMax - shardMin + 1;
_logger.Information("Starting {ShardCount} of {ShardTotal} shards (#{ShardMin}-#{ShardMax}) at {Url}",
shardCount, shardTotal, shardMin, shardMax, url);
for (var i = shardMin; i <= shardMax; i++)
CreateAndAddShard(url, new ShardInfo(i, shardTotal));
await StartShards();
}