misty.cafe/mistykey
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

manage user keypairs cache

Browse source
This commit is contained in:
Hazelnoot 2025-10-01 12:28:56 -04:00
parent b49ba4b00d
commit 34ea8dcbc2
1 changed files with 23 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

28
packages/backend/src/core/UserKeypairService.ts
View file

@ -5,16 +5,19 @@
import { Inject, Injectable, OnApplicationShutdown } from '@nestjs/common';
import * as Redis from 'ioredis';
import { In } from 'typeorm';
import type { MiUser } from '@/models/User.js';
import type { UserKeypairsRepository } from '@/models/_.js';
import { MemoryKVCache, RedisKVCache } from '@/misc/cache.js';
import type { MiUserKeypair } from '@/models/UserKeypair.js';
import { DI } from '@/di-symbols.js';
import { bindThis } from '@/decorators.js';
import { CacheManagementService, type ManagedQuantumKVCache } from '@/core/CacheManagementService.js';
import { InternalEventService } from '@/core/InternalEventService.js';
import type { InternalEventTypes } from '@/core/GlobalEventService.js';
@Injectable()
export class UserKeypairService implements OnApplicationShutdown {
private cache: MemoryKVCache<MiUserKeypair>;
public readonly userKeypairCache: ManagedQuantumKVCache<MiUserKeypair>;
constructor(
@Inject(DI.redis)
@ -22,18 +25,33 @@ export class UserKeypairService implements OnApplicationShutdown {
@Inject(DI.userKeypairsRepository)
private userKeypairsRepository: UserKeypairsRepository,
private readonly internalEventService: InternalEventService,
cacheManagementService: CacheManagementService,
) {
this.cache = new MemoryKVCache<MiUserKeypair>(1000 * 60 * 60 * 24); // 24h
this.userKeypairCache = cacheManagementService.createQuantumKVCache('userKeypair', {
lifetime: 1000 * 60 * 60, // 1h
fetcher: async userId => await this.userKeypairsRepository.findOneBy({ userId }),
bulkFetcher: async userIds => await this.userKeypairsRepository.findBy({ userId: In(userIds) }).then(ks => ks.map(k => [k.userId, k])),
});
this.internalEventService.on('userChangeDeletedState', this.onUserDeleted);
}
@bindThis
public async getUserKeypair(userId: MiUser['id']): Promise<MiUserKeypair> {
return await this.cache.fetch(userId, () => this.userKeypairsRepository.findOneByOrFail({ userId }));
return await this.userKeypairCache.fetch(userId);
}
@bindThis
private async onUserDeleted(body: InternalEventTypes['userChangeDeletedState']): Promise<void> {
await this.userKeypairCache.delete(body.id);
}
@bindThis
public dispose(): void {
this.cache.dispose();
this.internalEventService.off('userChangeDeletedState', this.onUserDeleted);
}
@bindThis