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

package/dist/index.cjs

@@ -129,11 +129,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;
@@ -417,7 +430,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 }
@@ -443,6 +460,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(
@@ -457,9 +475,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 }
@@ -485,6 +505,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(
@@ -499,6 +520,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 };
 }
@@ -1532,18 +1554,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) {