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

feat: use redis cache for non-id message lookups

Browse Source
This commit is contained in:
spiral
2022-11-24 06:32:55 +00:00
parent bf7747ab34
commit e9673a6704
9 changed files with 75 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
PluralKit.Core/Services/RedisService.cs
View File

@@ -11,4 +11,41 @@ public class RedisService
if (config.RedisAddr != null)
Connection = await ConnectionMultiplexer.ConnectAsync(config.RedisAddr);
}
private string LastMessageKey(ulong userId, ulong channelId) => $"user_last_message:{userId}:{channelId}";
public Task SetLastMessage(ulong userId, ulong channelId, ulong mid)
=> Connection.GetDatabase().UlongSetAsync(LastMessageKey(userId, channelId), mid, expiry: TimeSpan.FromMinutes(10));
public Task<ulong?> GetLastMessage(ulong userId, ulong channelId)
=> Connection.GetDatabase().UlongGetAsync(LastMessageKey(userId, channelId));
private string LoggerCleanKey(ulong userId, ulong guildId) => $"log_cleanup:{userId}:{guildId}";
public Task SetLogCleanup(ulong userId, ulong guildId)
=> Connection.GetDatabase().StringSetAsync(LoggerCleanKey(userId, guildId), 1, expiry: TimeSpan.FromSeconds(3));
public Task<bool> HasLogCleanup(ulong userId, ulong guildId)
=> Connection.GetDatabase().KeyExistsAsync(LoggerCleanKey(userId, guildId));
// note: these methods are named weird - they actually get the proxied mid from the original mid
// but anything else would've been more confusing
private string OriginalMidKey(ulong original_mid) => $"original_mid:{original_mid}";
public Task SetOriginalMid(ulong original_mid, ulong proxied_mid)
=> Connection.GetDatabase().UlongSetAsync(OriginalMidKey(original_mid), proxied_mid, expiry: TimeSpan.FromMinutes(30));
public Task<ulong?> GetOriginalMid(ulong original_mid)
=> Connection.GetDatabase().UlongGetAsync(OriginalMidKey(original_mid));
}
public static class RedisExt
{
public static async Task<ulong?> UlongGetAsync(this StackExchange.Redis.IDatabase database, string key)
{
var data = await database.StringGetAsync(key);
if (data == RedisValue.Null) return null;
if (ulong.TryParse(data, out var value))
return value;
return null;
}
public static Task UlongSetAsync(this StackExchange.Redis.IDatabase database, string key, ulong value, TimeSpan? expiry = null)
=> database.StringSetAsync(key, value.ToString(), expiry);
}