feat(cache): add AniwatchAPICache class to implement API caching layer
This commit is contained in:
@@ -0,0 +1,50 @@
|
|||||||
|
import { config } from "dotenv";
|
||||||
|
import { Redis } from "ioredis";
|
||||||
|
|
||||||
|
config();
|
||||||
|
|
||||||
|
export class AniwatchAPICache {
|
||||||
|
private _client: Redis | null;
|
||||||
|
public isOptional: boolean = true;
|
||||||
|
|
||||||
|
static DEFAULT_CACHE_EXPIRY_SECONDS = 60 as const;
|
||||||
|
static CACHE_EXPIRY_HEADER_NAME = "X-ANIWATCH-CACHE-EXPIRY" as const;
|
||||||
|
|
||||||
|
constructor() {
|
||||||
|
const redisConnURL = process.env?.ANIWATCH_API_REDIS_CONN_URL;
|
||||||
|
this.isOptional = !Boolean(redisConnURL);
|
||||||
|
this._client = this.isOptional ? null : new Redis(String(redisConnURL));
|
||||||
|
}
|
||||||
|
|
||||||
|
set(key: string | Buffer, value: string | Buffer | number) {
|
||||||
|
if (this.isOptional) return;
|
||||||
|
return this._client?.set(key, value);
|
||||||
|
}
|
||||||
|
|
||||||
|
get(key: string | Buffer) {
|
||||||
|
if (this.isOptional) return;
|
||||||
|
return this._client?.get(key);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @param expirySeconds set to 60 by default
|
||||||
|
*/
|
||||||
|
async getOrSet<T>(
|
||||||
|
key: string | Buffer,
|
||||||
|
setCB: () => Promise<T>,
|
||||||
|
expirySeconds: number = AniwatchAPICache.DEFAULT_CACHE_EXPIRY_SECONDS
|
||||||
|
) {
|
||||||
|
const cachedData = this.isOptional
|
||||||
|
? null
|
||||||
|
: (await this._client?.get(key)) || null;
|
||||||
|
let data = JSON.parse(String(cachedData)) as T;
|
||||||
|
|
||||||
|
if (!data) {
|
||||||
|
data = await setCB();
|
||||||
|
await this._client?.set(key, JSON.stringify(data), "EX", expirySeconds);
|
||||||
|
}
|
||||||
|
return data;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export const cache = new AniwatchAPICache();
|
||||||
Reference in New Issue
Block a user