@comake/skl-js-engine 1.4.3 → 1.5.0

package/src/util/ReadCacheHelper.ts

@@ -0,0 +1,64 @@
+import crypto from 'crypto';
+import type { ReadCacheOperation } from '../SklEngineOptions';
+
+export interface BuildReadCacheKeyInput {
+  operation: ReadCacheOperation;
+  args: readonly unknown[];
+  endpointUrl?: string;
+  namespace?: string;
+  keyHint?: string;
+}
+
+function stableStringify(value: unknown): string {
+  if (value === null) return 'null';
+
+  const valueType = typeof value;
+  if (valueType === 'number' || valueType === 'boolean') return String(value);
+  if (valueType === 'string') return JSON.stringify(value);
+  if (valueType !== 'object') {
+    const serializedPrimitive = JSON.stringify(value);
+    return serializedPrimitive ?? String(value);
+  }
+
+  if (Array.isArray(value)) {
+    return `[${value.map(stableStringify).join(',')}]`;
+  }
+
+  const objectValue = value as Record<string, unknown>;
+  // eslint-disable-next-line @typescript-eslint/require-array-sort-compare
+  const keys = Object.keys(objectValue).sort();
+  return `{${keys.map((key) => `${JSON.stringify(key)}:${stableStringify(objectValue[key])}`).join(',')}}`;
+}
+
+export function buildReadCacheKey(input: BuildReadCacheKeyInput): string {
+  const keyParts = {
+    namespace: input.namespace ?? '',
+    endpointUrl: input.endpointUrl ?? '',
+    operation: input.operation,
+    keyHint: input.keyHint ?? '',
+    args: input.args
+  };
+  return crypto.createHash('sha256').update(stableStringify(keyParts)).digest('hex');
+}
+
+export class ReadCacheSingleflight {
+  private readonly inflight = new Map<string, Promise<unknown>>();
+
+  public async do<T>(key: string, fn: () => Promise<T>): Promise<T> {
+    const existing = this.inflight.get(key) as Promise<T> | undefined;
+    if (existing) {
+      return existing;
+    }
+
+    const promise = (async(): Promise<T> => {
+      try {
+        return await fn();
+      } finally {
+        this.inflight.delete(key);
+      }
+    })();
+
+    this.inflight.set(key, promise as Promise<unknown>);
+    return promise;
+  }
+}