liz/PluralKit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: include shard_id in redis channel

Browse Source
This commit is contained in:
spiral
2022-05-10 07:32:14 -04:00
parent abb01aaf2c
commit 799279054d
4 changed files with 23 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
PluralKit.Bot/Services/RedisGatewayService.cs
View File

@@ -23,22 +23,26 @@ public class RedisGatewayService
_logger = logger.ForContext<RedisGatewayService>();
}
public event Func<IGatewayEvent, Task>? OnEventReceived;
public event Func<(int, IGatewayEvent), Task>? OnEventReceived;
public async Task Start()
public async Task Start(int shardId)
{
_redis = await ConnectionMultiplexer.ConnectAsync(_config.RedisGatewayUrl);
var channel = await _redis.GetSubscriber().SubscribeAsync("evt");
channel.OnMessage(Handle);
if (_redis == null)
_redis = await ConnectionMultiplexer.ConnectAsync(_config.RedisGatewayUrl);
_logger.Debug("Subscribing to shard {ShardId} on redis", shardId);
var channel = await _redis.GetSubscriber().SubscribeAsync($"evt-{shardId}");
channel.OnMessage((evt) => Handle(shardId, evt));
}
public async Task Handle(ChannelMessage message)
public async Task Handle(int shardId, ChannelMessage message)
{
var packet = JsonSerializer.Deserialize<GatewayPacket>(message.Message, _jsonSerializerOptions);
if (packet.Opcode != GatewayOpcode.Dispatch) return;
var evt = DeserializeEvent(packet.EventType, (JsonElement)packet.Payload);
if (evt == null) return;
await OnEventReceived(evt);
await OnEventReceived((shardId, evt));
}
private IGatewayEvent? DeserializeEvent(string eventType, JsonElement payload)