@localstack/appinspector-ui 1.0.54 → 1.0.55

package/dist/index.cjs

@@ -13101,10 +13101,16 @@ var buildSpansEndpoint = (request) => {
     queryParameters.set("pagination_token", request.pagination_token);
   }
   for (const [key, value] of Object.entries(request)) {
-    if (value !== void 0 && key !== "limit" && key !== "pagination_token" && value !== null) {
+    if (value !== void 0 && key !== "limit" && key !== "linked_span_ids" && key !== "linking_span_ids" && key !== "pagination_token" && value !== null) {
       queryParameters.set(key, String(value));
     }
   }
+  if (request.linked_span_ids) {
+    queryParameters.set("linked_span_ids", request.linked_span_ids.join(","));
+  }
+  if (request.linking_span_ids) {
+    queryParameters.set("linking_span_ids", request.linking_span_ids.join(","));
+  }
   const queryString = queryParameters.toString();
   return queryString ? `${API_ENDPOINTS.SPANS}?${queryString}` : API_ENDPOINTS.SPANS;
 };
@@ -14570,6 +14576,13 @@ var TraceGraph = ({
       } else {
         rootSpans.push(currentSpan);
       }
+      for (const linkedSpanId of span.linked_span_ids ?? []) {
+        const linkedSpan = spansMap.get(linkedSpanId);
+        if (linkedSpan) {
+          linkedSpan.children ??= [];
+          linkedSpan.children.push(currentSpan);
+        }
+      }
     }
     let allNodes = [];
     let allEdges = [];
@@ -14753,6 +14766,40 @@ var TraceGraphView = ({
     }
   );
 };
+async function fetchConnectedTraces(api, traceIds, allSpans = /* @__PURE__ */ new Map(), processedTraceIds = /* @__PURE__ */ new Set()) {
+  const newTraceIds = traceIds.filter((id2) => !processedTraceIds.has(id2));
+  if (newTraceIds.length === 0) {
+    return [...allSpans.values()];
+  }
+  for (const id2 of newTraceIds) processedTraceIds.add(id2);
+  const traceResponses = await Promise.all(
+    newTraceIds.map((id2) => api.getSpans({ trace_id: id2 }))
+  );
+  const currentBatchSpans = traceResponses.flatMap((r) => r.spans);
+  const currentSpanIds = currentBatchSpans.map((s) => s.span_id);
+  for (const span of currentBatchSpans) {
+    allSpans.set(span.span_id, span);
+  }
+  const [linkedSpans, linkingSpans] = await Promise.all([
+    api.getSpans({ linked_span_ids: currentSpanIds }),
+    api.getSpans({ linking_span_ids: currentSpanIds })
+  ]);
+  const discoveredTraceIds = /* @__PURE__ */ new Set();
+  for (const span of [...linkedSpans.spans, ...linkingSpans.spans]) {
+    if (!processedTraceIds.has(span.trace_id)) {
+      discoveredTraceIds.add(span.trace_id);
+    }
+  }
+  if (discoveredTraceIds.size > 0) {
+    return fetchConnectedTraces(
+      api,
+      [...discoveredTraceIds],
+      allSpans,
+      processedTraceIds
+    );
+  }
+  return [...allSpans.values()];
+}
 var TraceGraphPage = ({ onClose, spanId, traceId }) => {
   const api = useAppInspectorApi();
   const [spans, setSpans] = (0, import_react60.useState)();
@@ -14761,10 +14808,8 @@ var TraceGraphPage = ({ onClose, spanId, traceId }) => {
   const fetchSpans = (0, import_react60.useCallback)(async () => {
     setLoading(true);
     try {
-      const response = await api.getSpans({
-        trace_id: traceId
-      });
-      setSpans(response.spans);
+      const spans2 = await fetchConnectedTraces(api, [traceId]);
+      setSpans(spans2);
       setFetchError(void 0);
     } catch (error) {
       console.error(error);