This commit is contained in:
tamaina 2024-02-27 02:15:49 +00:00
parent 13e0a64a77
commit e2a8f4f880
2 changed files with 13 additions and 2 deletions

View File

@ -5,7 +5,7 @@
import { Inject, Injectable, OnApplicationShutdown } from '@nestjs/common'; import { Inject, Injectable, OnApplicationShutdown } from '@nestjs/common';
import { DI } from '@/di-symbols.js'; import { DI } from '@/di-symbols.js';
import type { NotesRepository, UserPublickeysRepository, UsersRepository } from '@/models/_.js'; import type { MiUser, NotesRepository, UserPublickeysRepository, UsersRepository } from '@/models/_.js';
import type { Config } from '@/config.js'; import type { Config } from '@/config.js';
import { MemoryKVCache } from '@/misc/cache.js'; import { MemoryKVCache } from '@/misc/cache.js';
import type { MiUserPublickey } from '@/models/UserPublickey.js'; import type { MiUserPublickey } from '@/models/UserPublickey.js';
@ -190,6 +190,16 @@ export class ApDbResolverService implements OnApplicationShutdown {
}; };
} }
@bindThis
public refreshCacheByUserId(userId: MiUser['id']): void {
this.publicKeyByUserIdCache.delete(userId);
for (const [k, v] of this.publicKeyCache.cache.entries()) {
if (v.value?.userId === userId) {
this.publicKeyCache.delete(k);
}
}
}
@bindThis @bindThis
public dispose(): void { public dispose(): void {
this.publicKeyCache.dispose(); this.publicKeyCache.dispose();

View File

@ -119,7 +119,8 @@ export class InboxProcessorService {
// みたいになっててUserを引っ張れば公開キーも入ることを期待する // みたいになっててUserを引っ張れば公開キーも入ることを期待する
if (activity.signature.creator) { if (activity.signature.creator) {
const candicate = activity.signature.creator.replace(/#.*/, ''); const candicate = activity.signature.creator.replace(/#.*/, '');
await this.apPersonService.resolvePerson(candicate).catch(() => null); const user = await this.apPersonService.resolvePerson(candicate).catch(() => null);
if (user) this.apDbResolverService.refreshCacheByUserId(user.id);
} }
// keyIdからLD-Signatureのユーザーを取得 // keyIdからLD-Signatureのユーザーを取得