[iotdb] branch improve/time-stat created (now 9a4714b1d3)

[er...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

ericpai pushed a change to branch improve/time-stat
in repository https://gitbox.apache.org/repos/asf/iotdb.git


      at 9a4714b1d3 Add test stats in cluster IT for future use

This branch includes the following new commits:

     new 9a4714b1d3 Add test stats in cluster IT for future use

The 1 revisions listed above as "new" are entirely new to this
repository and will be described in separate emails.  The revisions
listed as "add" were already present in the repository and have only
been added to this reference.

[iotdb] 01/01: Add test stats in cluster IT for future use

[er...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

ericpai pushed a commit to branch improve/time-stat
in repository https://gitbox.apache.org/repos/asf/iotdb.git

commit 9a4714b1d3fd0c072a98658d8fd3ce1b6b95db09
Author: ericpai <er...@hotmail.com>
AuthorDate: Wed Jul 6 17:51:52 2022 +0800

    Add test stats in cluster IT for future use
---
 .../org/apache/iotdb/it/env/IoTDBTestRunner.java   | 45 ++++++++++++++++++++--
 1 file changed, 41 insertions(+), 4 deletions(-)

diff --git a/integration-test/src/main/java/org/apache/iotdb/it/env/IoTDBTestRunner.java b/integration-test/src/main/java/org/apache/iotdb/it/env/IoTDBTestRunner.java
index d865e736fc..88a272ea79 100644
--- a/integration-test/src/main/java/org/apache/iotdb/it/env/IoTDBTestRunner.java
+++ b/integration-test/src/main/java/org/apache/iotdb/it/env/IoTDBTestRunner.java
@@ -18,6 +18,7 @@
  */
 package org.apache.iotdb.it.env;
 
+import org.jetbrains.annotations.NotNull;
 import org.junit.runner.Description;
 import org.junit.runner.notification.RunNotifier;
 import org.junit.runners.BlockJUnit4ClassRunner;
@@ -26,14 +27,29 @@ import org.junit.runners.model.InitializationError;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+
 public class IoTDBTestRunner extends BlockJUnit4ClassRunner {
 
   private static final Logger logger = LoggerFactory.getLogger(IoTDBTestRunner.class);
+  private final List<TestStat> testStats = new ArrayList<>();
 
   public IoTDBTestRunner(Class<?> testClass) throws InitializationError {
     super(testClass);
   }
 
+  @Override
+  public void run(RunNotifier notifier) {
+    super.run(notifier);
+    Collections.sort(testStats);
+    System.out.println("==== Top 30 slowest cases ====");
+    for (int i = 0; i < Math.min(testStats.size(), 30); i++) {
+      System.out.println(testStats.get(i));
+    }
+  }
+
   @Override
   protected void runChild(final FrameworkMethod method, RunNotifier notifier) {
     Description description = describeChild(method);
@@ -41,9 +57,30 @@ public class IoTDBTestRunner extends BlockJUnit4ClassRunner {
     long currentTime = System.currentTimeMillis();
     EnvFactory.getEnv().setTestMethodName(description.getMethodName());
     super.runChild(method, notifier);
-    logger.info(
-        "Done {}. Cost: {}s",
-        description.getMethodName(),
-        (System.currentTimeMillis() - currentTime) / 1000.0);
+    double timeCost = (System.currentTimeMillis() - currentTime) / 1000.0;
+    String testName = description.getClassName() + "." + description.getMethodName();
+    logger.info("Done {}. Cost: {}s", description.getMethodName(), timeCost);
+    testStats.add(new TestStat(testName, timeCost));
+  }
+
+  private static class TestStat implements Comparable<TestStat> {
+    private final String name;
+    private final double seconds;
+
+    public TestStat(String name, double seconds) {
+      this.name = name;
+      this.seconds = seconds;
+    }
+
+    @Override
+    public int compareTo(@NotNull TestStat o) {
+      // Compare in a reverse order
+      return Double.compare(o.seconds, seconds);
+    }
+
+    @Override
+    public String toString() {
+      return String.format("%.3f\t%s", seconds, name);
+    }
   }
 }