File size: 2,550 Bytes
bf48b89 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 | // Cloudflare Workers KV cache module
import type { KVNamespace } from '@cloudflare/workers-types';
import { config } from '@/config';
import type CacheModule from './base';
let kvNamespace: KVNamespace | null = null;
const status = { available: false };
const getCacheTtlKey = (key: string) => {
if (key.startsWith('rsshub:cacheTtl:')) {
throw new Error('"rsshub:cacheTtl:" prefix is reserved for the internal usage, please change your cache key');
}
return `rsshub:cacheTtl:${key}`;
};
export const setKVNamespace = (kv: KVNamespace) => {
kvNamespace = kv;
status.available = true;
};
export const getKVNamespace = () => kvNamespace;
export default {
init: () => {
// KV namespace is set via setKVNamespace from Worker env binding
},
get: async (key: string, refresh = true) => {
if (key && status.available && kvNamespace) {
const cacheTtlKey = getCacheTtlKey(key);
const [value, cacheTtl] = await Promise.all([kvNamespace.get(key), kvNamespace.get(cacheTtlKey)]);
if (value && refresh) {
const ttl = cacheTtl ? Number.parseInt(cacheTtl, 10) : config.cache.contentExpire;
// Refresh TTL by re-setting the value
// KV doesn't have a native expire refresh, so we need to re-put
// Use waitUntil pattern in production for non-blocking refresh
await Promise.all([kvNamespace.put(key, value, { expirationTtl: ttl }), cacheTtl ? kvNamespace.put(cacheTtlKey, cacheTtl, { expirationTtl: ttl }) : Promise.resolve()]);
}
return value || '';
} else {
return null;
}
},
set: async (key: string, value?: string | Record<string, any>, maxAge = config.cache.contentExpire) => {
if (!status.available || !kvNamespace) {
return;
}
if (!value || value === 'undefined') {
value = '';
}
if (typeof value === 'object') {
value = JSON.stringify(value);
}
if (key) {
const promises: Array<Promise<void>> = [kvNamespace.put(key, value, { expirationTtl: maxAge })];
if (maxAge !== config.cache.contentExpire) {
// Store the cache ttl if it is not the default value
promises.push(kvNamespace.put(getCacheTtlKey(key), String(maxAge), { expirationTtl: maxAge }));
}
await Promise.all(promises);
}
},
clients: {},
status,
} as CacheModule;
|