From 825d2dfcbbee08c37e621f12e98a42c075ea7735 Mon Sep 17 00:00:00 2001
From: SagarRajput-7 <162284829+SagarRajput-7@users.noreply.github.com>
Date: Fri, 27 Dec 2024 12:06:18 +0530
Subject: [PATCH] fix: added handling for new key - duration_nano in traces
 (#6658)

Co-authored-by: Srikanth Chekuri <srikanth.chekuri92@gmail.com>
---
 frontend/src/container/TimeSeriesView/index.tsx          | 3 ++-
 frontend/src/container/Trace/TraceTable/util.ts          | 2 +-
 frontend/src/container/TracesExplorer/ListView/utils.tsx | 2 +-
 3 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/frontend/src/container/TimeSeriesView/index.tsx b/frontend/src/container/TimeSeriesView/index.tsx
index b619e64b028a..c285ab70b664 100644
--- a/frontend/src/container/TimeSeriesView/index.tsx
+++ b/frontend/src/container/TimeSeriesView/index.tsx
@@ -29,7 +29,8 @@ function TimeSeriesViewContainer({
 		currentQuery.builder.queryData.forEach(
 			({ aggregateAttribute, aggregateOperator }) => {
 				const isExistDurationNanoAttribute =
-					aggregateAttribute.key === 'durationNano';
+					aggregateAttribute.key === 'durationNano' ||
+					aggregateAttribute.key === 'duration_nano';
 
 				const isCountOperator =
 					aggregateOperator === 'count' || aggregateOperator === 'count_distinct';
diff --git a/frontend/src/container/Trace/TraceTable/util.ts b/frontend/src/container/Trace/TraceTable/util.ts
index 08276636c11a..4b2e8c7c13a5 100644
--- a/frontend/src/container/Trace/TraceTable/util.ts
+++ b/frontend/src/container/Trace/TraceTable/util.ts
@@ -1,5 +1,5 @@
 export const getSpanOrderParam = (key: string): string => {
-	if (key === 'durationNano') {
+	if (key === 'durationNano' || key === 'duration_nano') {
 		return 'duration';
 	}
 	if (key === 'timestamp') {
diff --git a/frontend/src/container/TracesExplorer/ListView/utils.tsx b/frontend/src/container/TracesExplorer/ListView/utils.tsx
index 8dd20669a067..35c9f5aa8f62 100644
--- a/frontend/src/container/TracesExplorer/ListView/utils.tsx
+++ b/frontend/src/container/TracesExplorer/ListView/utils.tsx
@@ -96,7 +96,7 @@ export const getListColumns = (
 					);
 				}
 
-				if (key === 'durationNano') {
+				if (key === 'durationNano' || key === 'duration_nano') {
 					return (
 						<BlockLink to={getTraceLink(item)} openInNewTab={false}>
 							<Typography data-testid={key}>{getMs(value)}ms</Typography>