misskey/packages/backend/src/misc/cache.ts

44 lines
957 B
TypeScript
Raw Normal View History

2021-03-18 01:49:14 +00:00
export class Cache<T> {
private cache: Map<string | null, { date: number; value: T; }>;
private lifetime: number;
constructor(lifetime: Cache<never>['lifetime']) {
2021-03-18 01:54:39 +00:00
this.cache = new Map();
2021-03-18 01:49:14 +00:00
this.lifetime = lifetime;
}
2021-03-18 01:55:51 +00:00
public set(key: string | null, value: T): void {
2021-03-18 01:49:14 +00:00
this.cache.set(key, {
date: Date.now(),
2021-12-09 14:58:30 +00:00
value,
2021-03-18 01:49:14 +00:00
});
}
public get(key: string | null): T | undefined {
2021-03-18 01:49:14 +00:00
const cached = this.cache.get(key);
if (cached == null) return undefined;
2021-03-18 01:49:14 +00:00
if ((Date.now() - cached.date) > this.lifetime) {
this.cache.delete(key);
return undefined;
2021-03-18 01:49:14 +00:00
}
return cached.value;
}
public delete(key: string | null) {
this.cache.delete(key);
}
public async fetch(key: string | null, fetcher: () => Promise<T>): Promise<T> {
const cachedValue = this.get(key);
if (cachedValue !== undefined) {
// Cache HIT
return cachedValue;
}
// Cache MISS
const value = await fetcher();
this.set(key, value);
return value;
}
2021-03-18 01:49:14 +00:00
}