feat(waterfall): some more processing to handle waterfall collapse nodes

SigNoz · Jan 8, 2025 · 2025365 · 2025365
1 parent 0cb710e
commit 2025365
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/frontend/src/api/trace/getTraceV2.tsx b/frontend/src/api/trace/getTraceV2.tsx
@@ -12,9 +12,19 @@ const getTraceV2 = async (
 	props: GetTraceV2PayloadProps,
 ): Promise<SuccessResponse<GetTraceV2SuccessResponse> | ErrorResponse> => {
 	try {
+		let uncollapsedNodes = [...props.uncollapsedNodes];
+		if (!props.isInterestedSpanIdUnCollapsed) {
+			uncollapsedNodes = uncollapsedNodes.filter(
+				(node) => node !== props.interestedSpanId,
+			);
+		}
+		const postData: GetTraceV2PayloadProps = {
+			...props,
+			uncollapsedNodes,
+		};
 		const response = await axios.post<GetTraceV2SuccessResponse>(
 			`/traces/${props.traceId}`,
-			omit(props, 'traceId'),
+			omit(postData, 'traceId'),
 		);
 
 		return {