ash/iceshrimp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Clear public key cache on user deletion

Browse source
This commit is contained in:
Laura Hausmann 2023-09-05 22:56:35 +02:00
parent 85f396ff49
commit 35f7bde1a4
No known key found for this signature in database
GPG key ID: D044E84C5BE01605
1 changed files with 26 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
packages/backend/src/remote/activitypub/db-resolver.ts
View file

@ -18,6 +18,7 @@ import { uriPersonCache, userByIdCache } from "@/services/user-cache.js";
import type { IObject } from "./type.js"; import type { IObject } from "./type.js";
import { getApId } from "./type.js"; import { getApId } from "./type.js";
import { resolvePerson } from "./models/person.js"; import { resolvePerson } from "./models/person.js";
import {redisClient, subscriber} from "@/db/redis.js";
const publicKeyCache = new Cache<UserPublickey | null>("publicKey", 60 * 30); const publicKeyCache = new Cache<UserPublickey | null>("publicKey", 60 * 30);
const publicKeyByUserIdCache = new Cache<UserPublickey | null>( const publicKeyByUserIdCache = new Cache<UserPublickey | null>(
@ -204,3 +205,28 @@ export default class DbResolver {
}; };
} }
} }
subscriber.on("message", async (_, data) => {
const obj = JSON.parse(data);
if (obj.channel === "internal") {
const { type, body } = obj.message;
switch (type) {
case "remoteUserDeleted":
case "localUserDeleted": {
const toDelete = Array.from(await publicKeyByUserIdCache.getAll())
.filter((v) => v[1]?.userId === body.id)
.map((v) => v[0]);
const toDeleteKey = Array.from(await publicKeyCache.getAll())
.filter((v) => v[1]?.userId === body.id)
.map((v) => v[0]);
await publicKeyByUserIdCache.delete(...toDelete);
await publicKeyCache.delete(...toDeleteKey);
break;
}
default:
break;
}
}
});