eightball/PluralKit
1
0
Fork 0
mirror of https://github.com/PluralKit/PluralKit.git synced 2026-02-07 14:27:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: set shard status to offline when starting up

Browse source
This commit is contained in:
spiral 2021-11-28 20:50:13 -05:00
parent d2f29bbb51
commit d195c80d92
No known key found for this signature in database
GPG key ID: A6059F0CA0E1BD31
2 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
PluralKit.Core/Database/Repository/ModelRepository.Shards.cs
View file

@ -14,6 +14,8 @@ public partial class ModelRepository
"insert into shards (id, status) values (@Id, @Status) on conflict (id) do update set status = @Status",
new { Id = shard, Status = status });
public Task ClearShardStatus() => _db.Execute(conn => conn.ExecuteAsync("update shards set status = 0"));
public Task RegisterShardHeartbeat(IPKConnection conn, int shard, Duration ping) =>
conn.ExecuteAsync(
"insert into shards (id, last_heartbeat, ping) values (@Id, now(), @Ping) on conflict (id) do update set last_heartbeat = now(), ping = @Ping",