Mirror/SponsorBlockServer
1
0
Fork 0
mirror of https://github.com/ajayyy/SponsorBlockServer.git synced 2025-12-25 08:58:23 +03:00
Code Issues Packages Projects Releases Wiki Activity

Support dragonfly with in memory cache

Browse Source
This commit is contained in:
Ajay
2024-02-09 18:16:28 -05:00
parent 9fa248037a
commit 6d11e1c601
3 changed files with 16 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/utils/redis.ts
View File

@@ -180,6 +180,12 @@ if (config.redis?.enabled) {
const del = client.del.bind(client);
exportClient.del = (...keys) => {
if (config.redis.dragonflyMode) {
for (const key of keys) {
void client.publish("__redis__:invalidate", key);
}
}
if (config.redis.useCompression) {
return del(...keys.map((key) => createKeyName(key)) as [RedisCommandArgument]);
} else {
@@ -367,12 +373,15 @@ export function getRedisStats(): RedisStats {
async function setupCacheClientListener(cacheClient: RedisClientType,
cache: LRUCache<RedisCommandArgument, string>) {
cacheConnectionClientId = String(await cacheClient.clientId());
if (!config.redis.dragonflyMode) {
cacheConnectionClientId = String(await cacheClient.clientId());
}
cacheClient.subscribe("__redis__:invalidate", (keys) => {
if (keys) {
cacheClient.subscribe("__redis__:invalidate", (message) => {
if (message) {
lastInvalidationMessage = Date.now();
const keys = Buffer.isBuffer(message) ? [message.toString()] : message;
for (let key of keys) {
if (config.redis.useCompression) key = key.replace(/.c$/, "");
@@ -392,7 +401,7 @@ async function setupCacheClientListener(cacheClient: RedisClientType,
async function setupCacheClientTracking(client: RedisClientType,
cacheClient: RedisClientType) {
if (!client || !cacheClient.isReady) return;
if (!client || !cacheClient.isReady || config.redis.dragonflyMode) return;
await client.sendCommand(["CLIENT", "TRACKING", "ON", "REDIRECT", cacheConnectionClientId, "BCAST"]);
}