Correctly record the rawInputBytes metric in TableScan

facebookincubator · Jan 4, 2024 · 1a65207 · 1a65207
1 parent fefd586
commit 1a65207
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/velox/connectors/hive/HiveDataSource.h b/velox/connectors/hive/HiveDataSource.h
@@ -54,7 +54,7 @@ class HiveDataSource : public DataSource {
       const std::shared_ptr<common::Filter>& filter) override;
 
   uint64_t getCompletedBytes() override {
-    return ioStats_->rawBytesRead();
+    return ioStats_->read().sum();
   }
 
   uint64_t getCompletedRows() override {

diff --git a/velox/exec/tests/TableScanTest.cpp b/velox/exec/tests/TableScanTest.cpp
@@ -242,6 +242,7 @@ TEST_F(TableScanTest, allColumns) {
   auto it = planStats.find(scanNodeId);
   ASSERT_TRUE(it != planStats.end());
   ASSERT_TRUE(it->second.peakMemoryBytes > 0);
+  ASSERT_TRUE(it->second.rawInputBytes > 0);
   EXPECT_LT(0, exec::TableScan::ioWaitNanos());
 }