Return total time as a double

2 files changed, 3 insertions, 8 deletions

diff --git a/spark-common/src/main/java/me/lucko/spark/common/sampler/node/AbstractNode.java b/spark-common/src/main/java/me/lucko/spark/common/sampler/node/AbstractNode.java
index 8c5975a..221b9cb 100644
--- a/spark-common/src/main/java/me/lucko/spark/common/sampler/node/AbstractNode.java
+++ b/spark-common/src/main/java/me/lucko/spark/common/sampler/node/AbstractNode.java
@@ -30,7 +30,6 @@ import java.util.Collections;
 import java.util.List;
 import java.util.Map;
 import java.util.concurrent.ConcurrentHashMap;
-import java.util.concurrent.TimeUnit;
 import java.util.concurrent.atomic.LongAdder;
 
 /**
@@ -55,12 +54,8 @@ public abstract class AbstractNode {
      *
      * @return the total time
      */
-    public long getTotalTime() {
-        long millis = TimeUnit.MICROSECONDS.toMillis(this.totalTime.longValue());
-        if (millis == 0) {
-            return 1;
-        }
-        return millis;
+    public double getTotalTime() {
+        return this.totalTime.longValue() / 1000d;
     }
 
     private AbstractNode resolveChild(String className, String methodName, int lineNumber) {
diff --git a/spark-common/src/main/java/me/lucko/spark/common/sampler/node/StackTraceNode.java b/spark-common/src/main/java/me/lucko/spark/common/sampler/node/StackTraceNode.java
index c4e7ac4..80338e1 100644
--- a/spark-common/src/main/java/me/lucko/spark/common/sampler/node/StackTraceNode.java
+++ b/spark-common/src/main/java/me/lucko/spark/common/sampler/node/StackTraceNode.java
@@ -75,7 +75,7 @@ public final class StackTraceNode extends AbstractNode implements Comparable<Sta
 
     @Override
     public int compareTo(StackTraceNode that) {
-        int i = -Long.compare(this.getTotalTime(), that.getTotalTime());
+        int i = -Double.compare(this.getTotalTime(), that.getTotalTime());
         if (i != 0) {
             return i;
         }