[BugFix] reset context start time for each hyper query job's sql (#56196

) Signed-off-by: stephen <stephen5217@163.com>
StarRocks · Feb 24, 2025 · 6a6102b · 6a6102b
1 parent b358399
commit 6a6102b
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/fe/fe-core/src/main/java/com/starrocks/statistic/hyper/HyperQueryJob.java b/fe/fe-core/src/main/java/com/starrocks/statistic/hyper/HyperQueryJob.java
@@ -134,6 +134,8 @@ protected List<TStatisticData> executeStatisticsQuery(String sql, ConnectContext
             LOG.error(message, e);
             lastFailure = new RuntimeException(message, e);
             return Collections.emptyList();
+        } finally {
+            context.setStartTime();
         }
     }
 

diff --git a/fe/fe-core/src/test/java/com/starrocks/statistic/hyper/HyperJobTest.java b/fe/fe-core/src/test/java/com/starrocks/statistic/hyper/HyperJobTest.java
@@ -254,6 +254,26 @@ public List<TabletStats> sample() {
         Assert.assertEquals(5550000, sampler.getSampleInfo(pid).getSampleRowCount());
     }
 
+    @Test
+    public void testHyperQueryJobContextStartTime() {
+        Pair<List<String>, List<Type>> pair = initColumn(List.of("c1", "c2", "c3"));
+
+        new MockUp<SampleInfo>() {
+            @Mock
+            public List<TabletStats> getMediumHighWeightTablets() {
+                return List.of(new TabletStats(1, pid, 5000000));
+            }
+        };
+
+        List<HyperQueryJob> jobs = HyperQueryJob.createSampleQueryJobs(connectContext, db, table, pair.first,
+                pair.second, List.of(pid), 1, sampler);
+        long startTime = connectContext.getStartTime();
+        for (HyperQueryJob job : jobs) {
+            job.queryStatistics();
+            Assert.assertNotEquals(startTime, connectContext.getStartTime());
+        }
+    }
+
     @AfterClass
     public static void afterClass() {
         FeConstants.runningUnitTest = false;