@utaba/deep-memory-storage-cosmosdb 0.17.0 → 0.18.0

package/dist/index.cjs

@@ -1251,10 +1251,10 @@ async function getEntityRelationships(conn, repositoryId, entityId, options) {
   };
   let edgeTraversal;
   switch (direction) {
-    case "outbound":
+    case "out":
       edgeTraversal = "g.V().has('repositoryId', rid).hasId(eid).has('entityType').outE()";
       break;
-    case "inbound":
+    case "in":
       edgeTraversal = "g.V().has('repositoryId', rid).hasId(eid).has('entityType').inE()";
       break;
     case "both":
@@ -1273,9 +1273,9 @@ async function getEntityRelationships(conn, repositoryId, entityId, options) {
     typeFilter = `.hasLabel(${typeParams.join(", ")})`;
   }
   let unionQuery = null;
-  if (direction === "outbound") {
+  if (direction === "out") {
     unionQuery = `g.V().has('repositoryId', rid).hasId(eid).has('entityType').union(outE()${typeFilter}, inE()${typeFilter}.has('bidirectional', true))`;
-  } else if (direction === "inbound") {
+  } else if (direction === "in") {
     unionQuery = `g.V().has('repositoryId', rid).hasId(eid).has('entityType').union(inE()${typeFilter}, outE()${typeFilter}.has('bidirectional', true))`;
   }
   const baseQuery = unionQuery ?? `${edgeTraversal}${typeFilter}`;
@@ -2364,11 +2364,11 @@ Verify the ARM/Bicep template that provisioned the container.`
    *
    * The server-side step direction is fixed to `'both'` (catches every edge
    * in either direction). The directional + bidirectional filter is applied
-   * client-side per hop — i.e. `direction: 'outbound'` includes inbound edges
+   * client-side per hop — i.e. `direction: 'out'` includes inbound edges
    * where `bidirectional` is true. The CosmosDB Gremlin compiler does not
    * natively express the `union(outE, inE.has(bidirectional))` shape that
    * would push this filter server-side, so it stays client-side; the
-   * observable contract (which edges count toward "outbound" given
+   * observable contract (which edges count toward `'out'` given
    * bidirectionality) is preserved.
    *
    * Round-trips per call: `options.depth` (one per layer). The previous BFS
@@ -2414,17 +2414,17 @@ Verify the ARM/Bicep template that provisioned the container.`
           const isTarget = rel.targetEntityId === fv;
           let matchesDirection = false;
           let connectedId;
-          if (isSource && (options.direction === "outbound" || options.direction === "both")) {
+          if (isSource && (options.direction === "out" || options.direction === "both")) {
             matchesDirection = true;
             connectedId = rel.targetEntityId;
-          } else if (isTarget && (options.direction === "inbound" || options.direction === "both")) {
+          } else if (isTarget && (options.direction === "in" || options.direction === "both")) {
             matchesDirection = true;
             connectedId = rel.sourceEntityId;
           } else if (rel.bidirectional) {
-            if (isSource && options.direction === "inbound") {
+            if (isSource && options.direction === "in") {
               matchesDirection = true;
               connectedId = rel.targetEntityId;
-            } else if (isTarget && options.direction === "outbound") {
+            } else if (isTarget && options.direction === "out") {
               matchesDirection = true;
               connectedId = rel.sourceEntityId;
             }
@@ -2666,7 +2666,7 @@ Verify the ARM/Bicep template that provisioned the container.`
       }
       return projected;
     };
-    const projectStoredRelationship = (rel, direction = "outbound") => ({
+    const projectStoredRelationship = (rel, direction = "out") => ({
       id: rel.id,
       type: rel.relationshipType,
       sourceEntityId: rel.sourceEntityId,
@@ -2677,10 +2677,27 @@ Verify the ARM/Bicep template that provisioned the container.`
     let entities = [];
     let relationships;
     let paths;
+    const rangeRowCount = spec.returnMode === "all" ? raw.allEntities.length + raw.allRelationships.length : 0;
     if (spec.returnMode === "terminal") {
       entities = raw.terminalEntities.map(projectStoredEntity);
       relationships = void 0;
     } else if (spec.returnMode === "all") {
+      const missingIds = /* @__PURE__ */ new Set();
+      for (const rel of raw.allRelationships) {
+        if (!raw.entityMap.has(rel.sourceEntityId)) {
+          missingIds.add(rel.sourceEntityId);
+        }
+        if (!raw.entityMap.has(rel.targetEntityId)) {
+          missingIds.add(rel.targetEntityId);
+        }
+      }
+      if (missingIds.size > 0) {
+        const fetched = await this.getEntities(repositoryId, [...missingIds]);
+        for (const stored of fetched.values()) {
+          raw.allEntities.push(stored);
+          raw.entityMap.set(stored.id, stored);
+        }
+      }
       entities = raw.allEntities.map(projectStoredEntity);
       relationships = raw.allRelationships.map((r) => projectStoredRelationship(r));
     } else {
@@ -2708,7 +2725,7 @@ Verify the ARM/Bicep template that provisioned the container.`
         })
       }));
       relationships = Array.from(raw.relationshipMap.values()).map(
-        (rel) => projectStoredRelationship(rel, raw.pathRelFirstDirection.get(rel.id) ?? "outbound")
+        (rel) => projectStoredRelationship(rel, raw.pathRelFirstDirection.get(rel.id) ?? "out")
       );
     }
     const limit = spec.limit ?? 50;
@@ -2720,6 +2737,8 @@ Verify the ARM/Bicep template that provisioned the container.`
     } else {
       total = entities.length;
     }
+    const paginationSignal = spec.returnMode === "all" ? rangeRowCount : total;
+    const truncated = paginationSignal >= limit;
     const queryMetadata = {
       executionTimeMs: raw.executionTimeMs,
       resourceCost: raw.requestCharge != null ? { units: "RU", value: raw.requestCharge } : void 0,
@@ -2729,8 +2748,8 @@ Verify the ARM/Bicep template that provisioned the container.`
         maxResults: limit,
         maxDepth: spec.steps?.length
       },
-      truncated: total >= limit,
-      truncationReason: total >= limit ? "result_limit" : void 0
+      truncated,
+      truncationReason: truncated ? "result_limit" : void 0
     };
     return {
       entities,
@@ -2738,7 +2757,7 @@ Verify the ARM/Bicep template that provisioned the container.`
       paths,
       total,
       returned: total,
-      hasMore: total >= limit,
+      hasMore: truncated,
       queryMetadata
     };
   }
@@ -2829,7 +2848,7 @@ Verify the ARM/Bicep template that provisioned the container.`
             const stored = relationshipFromGremlin(props);
             raw.relationshipMap.set(stored.id, stored);
             pathRelIds.push(stored.id);
-            const direction = lastVertexId === stored.sourceEntityId ? "outbound" : "inbound";
+            const direction = lastVertexId === stored.sourceEntityId ? "out" : "in";
             pathRelDirections.push(direction);
             if (!raw.pathRelFirstDirection.has(stored.id)) {
               raw.pathRelFirstDirection.set(stored.id, direction);