Fix calc

comet-ml · Dec 12, 2024 · 29941cd · 29941cd
1 parent 290fb13
commit 29941cd
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 13 deletions.
diff --git a/apps/opik-backend/src/main/java/com/comet/opik/api/Span.java b/apps/opik-backend/src/main/java/com/comet/opik/api/Span.java
@@ -1,6 +1,7 @@
 package com.comet.opik.api;
 
 import com.comet.opik.domain.SpanType;
+import com.comet.opik.utils.DurationUtils;
 import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
 import com.fasterxml.jackson.annotation.JsonProperty;
 import com.fasterxml.jackson.annotation.JsonView;
@@ -14,7 +15,6 @@
 import lombok.Builder;
 
 import java.math.BigDecimal;
-import java.time.Duration;
 import java.time.Instant;
 import java.util.List;
 import java.util.Map;
@@ -77,11 +77,6 @@ public static class Public {
     @JsonView({Span.View.Public.class})
     @Schema(accessMode = Schema.AccessMode.READ_ONLY)
     public Double duration() {
-        if (endTime == null) {
-            return null;
-        }
-
-        long micros = Duration.between(startTime, endTime).toNanos() / 1_000;
-        return micros / 1_000.0;
+        return DurationUtils.getDurationInSeconds(startTime, endTime);
     }
 }
diff --git a/apps/opik-backend/src/main/java/com/comet/opik/api/Trace.java b/apps/opik-backend/src/main/java/com/comet/opik/api/Trace.java
@@ -1,5 +1,6 @@
 package com.comet.opik.api;
 
+import com.comet.opik.utils.DurationUtils;
 import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
 import com.fasterxml.jackson.annotation.JsonProperty;
 import com.fasterxml.jackson.annotation.JsonView;
@@ -72,11 +73,6 @@ public static class Public {
     @JsonView({Span.View.Public.class})
     @Schema(accessMode = Schema.AccessMode.READ_ONLY)
     public Double duration() {
-        if (endTime == null) {
-            return null;
-        }
-
-        long micros = Duration.between(startTime, endTime).toNanos() / 1_000;
-        return micros / 1_000.0;
+        return DurationUtils.getDurationInSeconds(startTime, endTime);
     }
 }
diff --git a/apps/opik-backend/src/main/java/com/comet/opik/utils/DurationUtils.java b/apps/opik-backend/src/main/java/com/comet/opik/utils/DurationUtils.java
@@ -0,0 +1,23 @@
+package com.comet.opik.utils;
+
+import lombok.NonNull;
+import lombok.experimental.UtilityClass;
+
+import java.time.Duration;
+import java.time.Instant;
+
+@UtilityClass
+public class DurationUtils {
+
+    public static final Double TIME_UNIT = 1_000.0;
+
+    public static Double getDurationInSeconds(@NonNull Instant startTime, Instant endTime) {
+        if (endTime == null) {
+            return null;
+        }
+
+        long micros = Duration.between(startTime, endTime).toNanos() / TIME_UNIT.longValue();
+        return micros / TIME_UNIT;
+    }
+
+}