55 lines
1.8 KiB
TypeScript
55 lines
1.8 KiB
TypeScript
import { KvPrimitives } from "./lib/kv_primitives.ts";
|
|
import { Plug } from "./plug.ts";
|
|
import { Manifest } from "./types.ts";
|
|
|
|
export interface ManifestCache<T> {
|
|
getManifest(plug: Plug<T>, hash: number): Promise<Manifest<T>>;
|
|
}
|
|
|
|
export class KVPrimitivesManifestCache<T> implements ManifestCache<T> {
|
|
constructor(private kv: KvPrimitives, private manifestPrefix: string) {
|
|
}
|
|
|
|
async getManifest(plug: Plug<T>, hash: number): Promise<Manifest<T>> {
|
|
const [cached] = await this.kv.batchGet([[
|
|
this.manifestPrefix,
|
|
plug.name,
|
|
]]);
|
|
if (cached && cached.hash === hash) {
|
|
// console.log("Using KV cached manifest for", plug.name);
|
|
return cached.manifest;
|
|
}
|
|
await plug.sandbox.init();
|
|
const manifest = plug.sandbox.manifest!;
|
|
await this.kv.batchSet([{
|
|
key: [this.manifestPrefix, plug.name],
|
|
// Deliverately removing the assets from the manifest to preserve space, will be re-added upon load of actual worker
|
|
value: { manifest: { ...manifest, assets: undefined }, hash },
|
|
}]);
|
|
return manifest;
|
|
}
|
|
}
|
|
|
|
export class InMemoryManifestCache<T> implements ManifestCache<T> {
|
|
private cache = new Map<string, {
|
|
manifest: Manifest<T>;
|
|
hash: number;
|
|
}>();
|
|
|
|
async getManifest(plug: Plug<T>, hash: number): Promise<Manifest<T>> {
|
|
const cached = this.cache.get(plug.workerUrl.href);
|
|
if (cached && cached.hash === hash) {
|
|
// console.log("Using memory cached manifest for", plug.name);
|
|
return cached.manifest;
|
|
}
|
|
await plug.sandbox.init();
|
|
const manifest = plug.sandbox.manifest!;
|
|
// Deliverately removing the assets from the manifest to preserve space, will be re-added upon load of actual worker
|
|
this.cache.set(plug.name!, {
|
|
manifest: { ...manifest, assets: undefined },
|
|
hash,
|
|
});
|
|
return manifest;
|
|
}
|
|
}
|