[native] Report driver level runtime statistics

prestodb · Mar 26, 2024 · 290876b · 290876b
1 parent 8b8c9e4
commit 290876b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/presto-native-execution/presto_cpp/main/PrestoTask.cpp b/presto-native-execution/presto_cpp/main/PrestoTask.cpp
@@ -548,6 +548,12 @@ protocol::TaskInfo PrestoTask::updateInfoLocked() {
           pipelineOut.outputDataSizeInBytes;
     }
 
+    for (const auto& driverStat : pipeline.driverStats) {
+      for (const auto& [name, value] : driverStat.runtimeStats) {
+        addRuntimeMetric(taskRuntimeStats, name, value);
+      }
+    }
+
     for (auto j = 0; j < pipeline.operatorStats.size(); ++j) {
       auto& opOut = pipelineOut.operatorSummaries[j];
       auto& op = pipeline.operatorStats[j];