@capagents/forgeqe 1.1.1 → 1.1.3

package/src/cache.ts

@@ -0,0 +1,92 @@
+import { createHash } from "node:crypto"
+
+import { createClient } from "redis"
+
+import type { CacheConfig } from "./config.ts"
+
+type CacheAdapter = {
+  get(key: string): Promise<string | null>
+  set(key: string, value: string, ttlSeconds: number): Promise<void>
+}
+
+let factoryOverride: ((config: CacheConfig) => Promise<CacheAdapter | null>) | undefined
+const adapters = new Map<string, Promise<CacheAdapter | null>>()
+
+function keyFor(config: CacheConfig): string {
+  return `${config.redisUrl}\n${config.keyPrefix}\n${config.connectTimeoutMs}`
+}
+
+function fullKey(config: CacheConfig, namespace: string, parts: unknown[]): string {
+  const digest = createHash("sha256").update(JSON.stringify(parts)).digest("hex")
+  return `${config.keyPrefix}:${namespace}:${digest}`
+}
+
+async function connectRedis(config: CacheConfig): Promise<CacheAdapter | null> {
+  const client = createClient({
+    url: config.redisUrl,
+    socket: { connectTimeout: config.connectTimeoutMs },
+  })
+  client.on("error", () => {})
+  try {
+    await client.connect()
+  } catch {
+    try {
+      client.disconnect()
+    } catch {}
+    return null
+  }
+  return {
+    async get(key) {
+      try {
+        return await client.get(key)
+      } catch {
+        return null
+      }
+    },
+    async set(key, value, ttlSeconds) {
+      try {
+        await client.set(key, value, { EX: ttlSeconds })
+      } catch {}
+    },
+  }
+}
+
+async function adapter(config: CacheConfig | undefined): Promise<CacheAdapter | null> {
+  if (!config?.enabled || !config.redisUrl) return null
+  const id = keyFor(config)
+  if (!adapters.has(id)) {
+    adapters.set(id, (factoryOverride || connectRedis)(config))
+  }
+  return adapters.get(id)!
+}
+
+export async function cachedJson<T>(
+  config: CacheConfig | undefined,
+  opts: {
+    namespace: string
+    keyParts: unknown[]
+    ttlSeconds?: number
+    load: () => Promise<T>
+  },
+): Promise<T> {
+  const ttlSeconds = opts.ttlSeconds || config?.defaultTtlSeconds || 3600
+  const store = await adapter(config)
+  if (!store) return opts.load()
+  const key = fullKey(config!, opts.namespace, opts.keyParts)
+  const hit = await store.get(key)
+  if (hit !== null) {
+    try {
+      return JSON.parse(hit) as T
+    } catch {
+      // Bad JSON should not break the current run; reload and overwrite it.
+    }
+  }
+  const value = await opts.load()
+  await store.set(key, JSON.stringify(value), ttlSeconds)
+  return value
+}
+
+export function setCacheFactoryForTests(factory?: (config: CacheConfig) => Promise<CacheAdapter | null>): void {
+  factoryOverride = factory
+  adapters.clear()
+}