@utaba/deep-memory-storage-cosmosdb 0.9.2 → 0.10.0

package/dist/index.d.cts

@@ -22,6 +22,14 @@ interface CosmosDbProviderConfig {
     defaultTimeoutMs?: number;
     /** Whether to reject unauthorized TLS certs — set false for emulator (default: true) */
     rejectUnauthorized?: boolean;
+    /**
+     * Milliseconds to pause between batches in deleteAllContents/deleteRepository
+     * to pace RU consumption. Required on low-RU tiers (e.g. CosmosDB free tier at
+     * 1000 RU/s), where a single batch drop query can consume the entire per-second
+     * budget and throttle the next batch. Set to 1000+ on constrained tiers.
+     * Default: 0 (no pacing — suitable for autoscale and provisioned throughput).
+     */
+    deleteBatchDelayMs?: number;
     /**
      * Optional usage sink. When provided, the provider emits one
      * {@link OperationUsage} record per public method call with the aggregated

package/dist/index.d.ts

@@ -22,6 +22,14 @@ interface CosmosDbProviderConfig {
     defaultTimeoutMs?: number;
     /** Whether to reject unauthorized TLS certs — set false for emulator (default: true) */
     rejectUnauthorized?: boolean;
+    /**
+     * Milliseconds to pause between batches in deleteAllContents/deleteRepository
+     * to pace RU consumption. Required on low-RU tiers (e.g. CosmosDB free tier at
+     * 1000 RU/s), where a single batch drop query can consume the entire per-second
+     * budget and throttle the next batch. Set to 1000+ on constrained tiers.
+     * Default: 0 (no pacing — suitable for autoscale and provisioned throughput).
+     */
+    deleteBatchDelayMs?: number;
     /**
      * Optional usage sink. When provided, the provider emits one
      * {@link OperationUsage} record per public method call with the aggregated

package/dist/index.js

@@ -92,11 +92,24 @@ function isTransientError(err) {
   return false;
 }
 function getRetryAfterMs(err, attempt) {
-  const retryAfter = err?.["retryAfterMs"];
-  if (typeof retryAfter === "number" && retryAfter > 0) {
-    return retryAfter;
-  }
-  return Math.min(500 * Math.pow(2, attempt), 1e4);
+  const exponentialMs = Math.min(500 * Math.pow(2, attempt), 1e4);
+  const topLevel = err?.["retryAfterMs"];
+  if (typeof topLevel === "number" && topLevel > 0) {
+    return Math.max(topLevel, exponentialMs);
+  }
+  const attrs = err?.statusAttributes;
+  let hintVal;
+  if (attrs instanceof Map) {
+    hintVal = attrs.get("x-ms-retry-after-ms") ?? attrs.get("x-ms-retry-after");
+  } else if (typeof attrs === "object" && attrs !== null) {
+    const record = attrs;
+    hintVal = record["x-ms-retry-after-ms"] ?? record["x-ms-retry-after"];
+  }
+  const hintMs = Number(hintVal);
+  if (Number.isFinite(hintMs) && hintMs > 0) {
+    return Math.max(hintMs, exponentialMs);
+  }
+  return exponentialMs;
 }
 function extractRequestCharge(resultSet) {
   const attrs = resultSet.attributes;
@@ -380,7 +393,11 @@ async function updateRepository(conn, repositoryId, updates) {
   return await getRepository(conn, repositoryId);
 }
 var DELETE_BATCH_SIZE = 500;
-async function deleteRepository(conn, repositoryId, onProgress) {
+function sleep2(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+async function deleteRepository(conn, repositoryId, onProgress, options) {
+  const batchDelayMs = options?.batchDelayMs ?? 0;
   const entityCountResult = await conn.submit(
     "g.V().has('repositoryId', rid).has('entityType').count()",
     { rid: repositoryId }
@@ -406,6 +423,7 @@ async function deleteRepository(conn, repositoryId, onProgress) {
     relationshipsDeleted = totalRelationships - remainingCount;
     await onProgress?.({ entitiesDeleted, relationshipsDeleted, totalEntities, totalRelationships });
     if (remainingCount === 0) break;
+    if (batchDelayMs > 0) await sleep2(batchDelayMs);
   }
   while (true) {
     await conn.submit(
@@ -420,9 +438,11 @@ async function deleteRepository(conn, repositoryId, onProgress) {
     entitiesDeleted = totalEntities - remainingCount;
     await onProgress?.({ entitiesDeleted, relationshipsDeleted, totalEntities, totalRelationships });
     if (remainingCount === 0) break;
+    if (batchDelayMs > 0) await sleep2(batchDelayMs);
   }
 }
-async function deleteAllContents(conn, repositoryId, onProgress) {
+async function deleteAllContents(conn, repositoryId, onProgress, options) {
+  const batchDelayMs = options?.batchDelayMs ?? 0;
   const entityCountResult = await conn.submit(
     "g.V().has('repositoryId', rid).has('entityType').count()",
     { rid: repositoryId }
@@ -448,6 +468,7 @@ async function deleteAllContents(conn, repositoryId, onProgress) {
     relationshipsDeleted = totalRelationships - remainingCount;
     await onProgress?.({ entitiesDeleted, relationshipsDeleted, totalEntities, totalRelationships });
     if (remainingCount === 0) break;
+    if (batchDelayMs > 0) await sleep2(batchDelayMs);
   }
   while (true) {
     await conn.submit(
@@ -462,6 +483,7 @@ async function deleteAllContents(conn, repositoryId, onProgress) {
     entitiesDeleted = totalEntities - remainingCount;
     await onProgress?.({ entitiesDeleted, relationshipsDeleted, totalEntities, totalRelationships });
     if (remainingCount === 0) break;
+    if (batchDelayMs > 0) await sleep2(batchDelayMs);
   }
   return { deletedEntities: totalEntities, deletedRelationships: totalRelationships };
 }
@@ -1495,18 +1517,20 @@ var CosmosDbProvider = class {
   }
   async deleteRepository(repositoryId, onProgress) {
     this.assertValidRepositoryId(repositoryId);
+    const batchDelayMs = this.config.deleteBatchDelayMs ?? 0;
     return this.track(
       "deleteRepository",
       repositoryId,
-      () => deleteRepository(this.conn, repositoryId, onProgress)
+      () => deleteRepository(this.conn, repositoryId, onProgress, { batchDelayMs })
     );
   }
   async deleteAllContents(repositoryId, onProgress) {
     this.assertValidRepositoryId(repositoryId);
+    const batchDelayMs = this.config.deleteBatchDelayMs ?? 0;
     return this.track(
       "deleteAllContents",
       repositoryId,
-      () => deleteAllContents(this.conn, repositoryId, onProgress)
+      () => deleteAllContents(this.conn, repositoryId, onProgress, { batchDelayMs })
     );
   }
   async getRepositoryStats(repositoryId) {