diff --git a/velox/connectors/hive/HiveDataSource.h b/velox/connectors/hive/HiveDataSource.h index 3d3f6d733d040..dcd9e8983cf0c 100644 --- a/velox/connectors/hive/HiveDataSource.h +++ b/velox/connectors/hive/HiveDataSource.h @@ -54,7 +54,7 @@ class HiveDataSource : public DataSource { const std::shared_ptr& 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 index fc763bf939573..69eb43c1ebc56 100644 --- 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()); }