silverbullet/plugos/manifest_cache.ts

51 lines
1.6 KiB
TypeScript

import { DataStore } from "./lib/datastore.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 ds: DataStore, private manifestPrefix: string) {
}
async getManifest(plug: Plug<T>, hash: number): Promise<Manifest<T>> {
const cached = await this.ds.get([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.ds.set([this.manifestPrefix, plug.name], {
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;
}
}