Update incRawBytesRead metric in DirectBufferedInput

facebookincubator · Jan 10, 2024 · c3b8668 · c3b8668
1 parent 25be5d6
commit c3b8668
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_->read().sum();
+    return ioStats_->rawBytesRead();
   }
 
   uint64_t getCompletedRows() override {

diff --git a/velox/dwio/common/DirectBufferedInput.cpp b/velox/dwio/common/DirectBufferedInput.cpp
@@ -274,6 +274,7 @@ std::vector<cache::CachePin> DirectCoalescedLoad::loadData(bool isPrefetch) {
   }
   input_->read(buffers, requests_[0].region.offset, LogType::FILE);
   ioStats_->read().increment(size);
+  ioStats_->incRawBytesRead(size);
   ioStats_->incRawOverreadBytes(overread);
   if (isPrefetch) {
     ioStats_->prefetch().increment(size);