create-windy 0.6.14 → 0.6.16

package/template/packages/storage/src/memory-blob-store.ts

@@ -5,6 +5,7 @@ import {
   type ArtifactRef,
   type BlobStore,
   type PutBlobInput,
+  type PutBlobOnceInput,
   type ReadableBlob,
   type StoredBlob,
 } from "./blob-store.js";
@@ -19,39 +20,25 @@ export class MemoryBlobStore implements BlobStore {
   private readonly blobs = new Map<ArtifactRef, MemoryBlob>();
 
   async put(input: PutBlobInput): Promise<StoredBlob> {
-    const chunks: Uint8Array[] = [];
-    const hash = createSha256Accumulator();
-    let byteSize = 0;
-    const reader = input.body.getReader();
-    while (true) {
-      const result = await reader.read();
-      if (result.done) break;
-      const copy = Uint8Array.from(result.value);
-      chunks.push(copy);
-      hash.update(copy);
-      byteSize += copy.byteLength;
-    }
-    if (
-      input.expectedByteSize !== undefined &&
-      input.expectedByteSize !== byteSize
-    ) {
-      throw new BlobStoreError("SIZE_MISMATCH", "Blob 大小与声明不一致");
-    }
-    const contentHash = hash.digest();
-    if (
-      input.expectedContentHash &&
-      input.expectedContentHash.value !== contentHash.value
-    ) {
-      throw new BlobStoreError("HASH_MISMATCH", "Blob SHA-256 与声明不一致");
-    }
-    const ref = artifactRef(`blob_${randomUUID().replaceAll("-", "")}`);
-    const blob: MemoryBlob = {
-      ref,
-      byteSize,
-      contentHash,
-      createdAt: new Date().toISOString(),
-      bytes: join(chunks, byteSize),
-    };
+    return this.write(
+      artifactRef(`blob_${randomUUID().replaceAll("-", "")}`),
+      input,
+    );
+  }
+
+  async putOnce(input: PutBlobOnceInput): Promise<StoredBlob> {
+    const ref = artifactRef(input.ref);
+    const existing = this.blobs.get(ref);
+    if (existing) return requireExactBlob(existing, input);
+    const candidate = await readBlob(ref, input);
+    const concurrent = this.blobs.get(ref);
+    if (concurrent) return requireExactBlob(concurrent, input);
+    this.blobs.set(ref, candidate);
+    return metadata(candidate);
+  }
+
+  private async write(ref: ArtifactRef, input: PutBlobInput) {
+    const blob = await readBlob(ref, input);
     this.blobs.set(ref, blob);
     return metadata(blob);
   }
@@ -84,6 +71,59 @@ export class MemoryBlobStore implements BlobStore {
   }
 }
 
+async function readBlob(
+  ref: ArtifactRef,
+  input: PutBlobInput,
+): Promise<MemoryBlob> {
+  const chunks: Uint8Array[] = [];
+  const hash = createSha256Accumulator();
+  let byteSize = 0;
+  const reader = input.body.getReader();
+  while (true) {
+    const result = await reader.read();
+    if (result.done) break;
+    const copy = Uint8Array.from(result.value);
+    chunks.push(copy);
+    hash.update(copy);
+    byteSize += copy.byteLength;
+  }
+  if (
+    input.expectedByteSize !== undefined &&
+    input.expectedByteSize !== byteSize
+  ) {
+    throw new BlobStoreError("SIZE_MISMATCH", "Blob 大小与声明不一致");
+  }
+  const contentHash = hash.digest();
+  if (
+    input.expectedContentHash &&
+    input.expectedContentHash.value !== contentHash.value
+  ) {
+    throw new BlobStoreError("HASH_MISMATCH", "Blob SHA-256 与声明不一致");
+  }
+  return {
+    ref,
+    byteSize,
+    contentHash,
+    createdAt: new Date().toISOString(),
+    bytes: join(chunks, byteSize),
+  };
+}
+
+function requireExactBlob(blob: MemoryBlob, input: PutBlobOnceInput) {
+  if (
+    blob.ref !== input.ref ||
+    blob.byteSize !== input.expectedByteSize ||
+    blob.contentHash.algorithm !== input.expectedContentHash.algorithm ||
+    blob.contentHash.value !== input.expectedContentHash.value
+  ) {
+    throw new BlobStoreError(
+      "BLOB_IDENTITY_CONFLICT",
+      `Blob 稳定坐标冲突：${input.ref}`,
+    );
+  }
+  return metadata(blob);
+}
+
 function metadata(blob: MemoryBlob): StoredBlob {
   const { bytes: _bytes, ...value } = blob;
   return { ...value, contentHash: { ...value.contentHash } };