e2net/iceshrimp
3
0
Fork 0
forked from mirrors/iceshrimp
Code Pull requests Activity

Fix Cache<T> .getAll and .delete functions

Browse source
This commit is contained in:
Laura Hausmann 2023-09-04 21:30:26 +02:00
parent 0ad31334e2
commit 9e5f96b9ad
Signed by: zotan
GPG key ID: D044E84C5BE01605
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
packages/backend/src/misc/cache.ts
View file

@ -1,6 +1,7 @@
import { redisClient } from "@/db/redis.js";
import { encode, decode } from "msgpackr";
import { ChainableCommander } from "ioredis";
import config from "@/config/index.js";
export class Cache<T> {
private ttl: number;
@ -37,12 +38,14 @@ export class Cache<T> {
}
public async getAll(renew = false): Promise<Map<string, T>> {
const keys = await redisClient.keys(`${this.prefix}*`);
const finalPrefix = `${config.cacheServer?.prefix ?? config.redis.prefix}:${this.prefix}:`;
const keys = (await redisClient.keys(`${finalPrefix}*`)).map(p => p.substring(finalPrefix.length));
const prefixedKeys = keys.map(p => this.prefixedKey(p));
const map = new Map<string, T>();
if (keys.length === 0) {
return map;
}
const values = await redisClient.mgetBuffer(keys);
const values = await redisClient.mgetBuffer(prefixedKeys);
for (const [i, key] of keys.entries()) {
const val = values[i];
@ -54,7 +57,7 @@ export class Cache<T> {
if (renew) {
const trans = redisClient.multi();
for (const key of map.keys()) {
trans.expire(key, this.ttl);
trans.expire(this.prefixedKey(key), this.ttl);
}
await trans.exec();
}
@ -64,7 +67,7 @@ export class Cache<T> {
public async delete(...keys: (string | null)[]): Promise<void> {
if (keys.length > 0) {
const _keys = keys.map(this.prefixedKey);
const _keys = keys.map(p => this.prefixedKey(p));
await redisClient.del(_keys);
}
}