@envelop/response-cache-cloudflare-kv
Version:
- Supports [Cloudflare KV](https://developers.cloudflare.com/kv/) cache for `@envelop/response-cache` plugin - Suitable for graphql servers running on [Cloudflare Workers](https://workers.cloudflare.com/)
45 lines (44 loc) • 1.91 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.getAllKvKeysForPrefix = exports.invalidateCacheEntityRecord = exports.invalidate = void 0;
const cache_key_js_1 = require("./cache-key.js");
async function invalidate(entities, KV, keyPrefix) {
const kvPromises = []; // Collecting all the KV operations so we can await them all at once
const entityInvalidationPromises = []; // Parallelize invalidation of each entity
for (const entity of entities) {
entityInvalidationPromises.push(invalidateCacheEntityRecord(entity, kvPromises, KV, keyPrefix));
}
await Promise.allSettled(entityInvalidationPromises);
await Promise.allSettled(kvPromises);
}
exports.invalidate = invalidate;
async function invalidateCacheEntityRecord(entity,
/** Collect all inner promises to batch await all async operations outside the function */
kvPromiseCollection, KV, keyPrefix) {
const entityKey = (0, cache_key_js_1.buildEntityKey)(entity.typename, entity.id, keyPrefix);
for await (const kvKey of getAllKvKeysForPrefix(entityKey, KV)) {
if (kvKey.metadata?.operationKey) {
kvPromiseCollection.push(KV.delete(kvKey.metadata?.operationKey));
kvPromiseCollection.push(KV.delete(kvKey.name));
}
}
}
exports.invalidateCacheEntityRecord = invalidateCacheEntityRecord;
async function* getAllKvKeysForPrefix(prefix, KV) {
let keyListComplete = false;
let cursor;
do {
const kvListResponse = await KV.list({
prefix,
cursor,
});
keyListComplete = kvListResponse.list_complete;
if (!kvListResponse.list_complete) {
cursor = kvListResponse.cursor;
}
for (const keyResult of kvListResponse.keys) {
yield keyResult;
}
} while (!keyListComplete);
}
exports.getAllKvKeysForPrefix = getAllKvKeysForPrefix;