50 lines
1.4 KiB
TypeScript
50 lines
1.4 KiB
TypeScript
/*
|
|
* SPDX-FileCopyrightText: syuilo and other misskey contributors
|
|
* SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
import { Inject, Injectable, OnApplicationShutdown } from '@nestjs/common';
|
|
import * as Redis from 'ioredis';
|
|
import type { MiUser } from '@/models/User.js';
|
|
import type { UserKeypairsRepository } from '@/models/_.js';
|
|
import { RedisKVCache } from '@/misc/cache.js';
|
|
import type { MiUserKeypair } from '@/models/UserKeypair.js';
|
|
import { DI } from '@/di-symbols.js';
|
|
import { bindThis } from '@/decorators.js';
|
|
|
|
@Injectable()
|
|
export class UserKeypairService implements OnApplicationShutdown {
|
|
private cache: RedisKVCache<MiUserKeypair>;
|
|
|
|
constructor(
|
|
@Inject(DI.redis)
|
|
private redisClient: Redis.Redis,
|
|
|
|
@Inject(DI.userKeypairsRepository)
|
|
private userKeypairsRepository: UserKeypairsRepository,
|
|
) {
|
|
this.cache = new RedisKVCache<MiUserKeypair>(this.redisClient, 'userKeypair', {
|
|
lifetime: 1000 * 60 * 60 * 24, // 24h
|
|
memoryCacheLifetime: Infinity,
|
|
fetcher: (key) => this.userKeypairsRepository.findOneByOrFail({ userId: key }),
|
|
toRedisConverter: (value) => JSON.stringify(value),
|
|
fromRedisConverter: (value) => JSON.parse(value),
|
|
});
|
|
}
|
|
|
|
@bindThis
|
|
public async getUserKeypair(userId: MiUser['id']): Promise<MiUserKeypair> {
|
|
return await this.cache.fetch(userId);
|
|
}
|
|
|
|
@bindThis
|
|
public dispose(): void {
|
|
this.cache.dispose();
|
|
}
|
|
|
|
@bindThis
|
|
public onApplicationShutdown(signal?: string | undefined): void {
|
|
this.dispose();
|
|
}
|
|
}
|