Add test cases for helper function profileNanos

prestodb · Mar 14, 2024 · 0572190 · 0572190
1 parent e51a143
commit 0572190
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/presto-common/src/main/java/com/facebook/presto/common/RuntimeStats.java b/presto-common/src/main/java/com/facebook/presto/common/RuntimeStats.java
@@ -137,4 +137,11 @@ public <V> V profileNanos(String tag, Supplier<V> supplier)
         addMetricValueIgnoreZero(tag, NANO, System.nanoTime() - startTime);
         return result;
     }
+
+    public void profileNanosVoid(String tag, Runnable runnable)
+    {
+        long startTime = System.nanoTime();
+        runnable.run();
+        addMetricValueIgnoreZero(tag, NANO, System.nanoTime() - startTime);
+    }
 }
diff --git a/presto-common/src/test/java/com/facebook/presto/common/TestRuntimeStats.java b/presto-common/src/test/java/com/facebook/presto/common/TestRuntimeStats.java
@@ -29,7 +29,9 @@ public class TestRuntimeStats
     private static final String TEST_METRIC_NAME_3 = "test3";
     private static final String TEST_METRIC_NAME_NANO_1 = "test_nano_1";
     private static final String TEST_METRIC_NAME_NANO_2 = "test_nano_2";
+    private static final String TEST_METRIC_NAME_NANO_3 = "test_nano_3";
     private static final String TEST_METRIC_NAME_BYTE = "test_byte";
+    private static final long ONE_SECOND_IN_NANOS = 1_000_000_000L;
 
     private void assertRuntimeMetricEquals(RuntimeMetric m1, RuntimeMetric m2)
     {
@@ -234,4 +236,23 @@ public void testReturnUnmodifiedMetrics()
         RuntimeStats stats = new RuntimeStats();
         stats.getMetrics().put(TEST_METRIC_NAME_1, new RuntimeMetric(TEST_METRIC_NAME_1, NONE));
     }
+
+    @Test
+    public void testProfileNano()
+    {
+        RuntimeStats stats = new RuntimeStats();
+        int status = stats.profileNanos(TEST_METRIC_NAME_NANO_3, () -> 1);
+
+        assert stats.getMetric(TEST_METRIC_NAME_NANO_3).getSum() < ONE_SECOND_IN_NANOS;
+        assertEquals(status, 1);
+    }
+
+    @Test
+    public void testProfileNanoVoid()
+    {
+        RuntimeStats stats = new RuntimeStats();
+        stats.profileNanosVoid(TEST_METRIC_NAME_NANO_3, () -> {});
+
+        assert stats.getMetric(TEST_METRIC_NAME_NANO_3).getSum() < ONE_SECOND_IN_NANOS;
+    }
 }