refactor(native): Change to new background cpu time var location (#26673)

Summary:

shown in title

Reviewed By: xiaoxmeng

Differential Revision: D87611460

Author: tanjialiang

presto-native-execution/presto_cpp/main/operators/ShuffleWrite.cpp

@@ -120,35 +120,34 @@ class ShuffleWrite : public Operator {
 
   void recordShuffleWriteClientStats() {
     auto lockedStats = stats_.wlock();
-    std::optional<uint64_t> backgroundCpuTimeMsOpt = std::nullopt;
+    std::optional<uint64_t> backgroundCpuTimeNanosOpt = std::nullopt;
     if (shuffle_->supportsMetrics()) {
       const auto shuffleMetrics = shuffle_->metrics();
       for (const auto& [name, metric] : shuffleMetrics) {
         lockedStats->runtimeStats[name] = metric;
       }
 
-      if (shuffleMetrics.contains(ExchangeClient::kBackgroundCpuTimeMs)) {
-        backgroundCpuTimeMsOpt =
-            shuffleMetrics.at(ExchangeClient::kBackgroundCpuTimeMs).sum;
+      if (shuffleMetrics.contains(Operator::kBackgroundCpuTimeNanos)) {
+        backgroundCpuTimeNanosOpt =
+            shuffleMetrics.at(Operator::kBackgroundCpuTimeNanos).sum;
       }
     } else {
       const auto shuffleStats = shuffle_->stats();
       for (const auto& [name, value] : shuffleStats) {
         lockedStats->runtimeStats[name] = RuntimeMetric(value);
       }
 
-      if (shuffleStats.contains(ExchangeClient::kBackgroundCpuTimeMs)) {
-        backgroundCpuTimeMsOpt =
-            shuffleStats.at(ExchangeClient::kBackgroundCpuTimeMs);
+      if (shuffleStats.contains(Operator::kBackgroundCpuTimeNanos)) {
+        backgroundCpuTimeNanosOpt =
+            shuffleStats.at(Operator::kBackgroundCpuTimeNanos);
       }
     }
 
-    if (backgroundCpuTimeMsOpt.has_value()) {
+    if (backgroundCpuTimeNanosOpt.has_value()) {
       const CpuWallTiming backgroundTiming{
-          static_cast<uint64_t>(1),
-          0,
-          static_cast<uint64_t>(*backgroundCpuTimeMsOpt) *
-              Timestamp::kNanosecondsInMillisecond};
+          .count = static_cast<uint64_t>(1),
+          .wallNanos = 0,
+          .cpuNanos = static_cast<uint64_t>(*backgroundCpuTimeNanosOpt)};
       lockedStats->backgroundTiming.clear();
       lockedStats->backgroundTiming.add(backgroundTiming);
     }

presto-native-execution/presto_cpp/main/operators/tests/ShuffleTest.cpp

@@ -42,7 +42,7 @@ namespace facebook::presto::operators::test {
 
 namespace {
 
-static const uint64_t kFakeBackgroundCpuTimeMs = 123;
+static const uint64_t kFakeBackgroundCpuTimeNanos = 123000000;
 
 struct TestShuffleInfo {
   uint32_t numPartitions;
@@ -153,7 +153,7 @@ class TestShuffleWriter : public ShuffleWriter {
   folly::F14FastMap<std::string, int64_t> stats() const override {
     return {
         {"test-shuffle.write", 1002},
-        {exec::ExchangeClient::kBackgroundCpuTimeMs, kFakeBackgroundCpuTimeMs}};
+        {exec::Operator::kBackgroundCpuTimeNanos, kFakeBackgroundCpuTimeNanos}};
   }
 
   std::shared_ptr<std::vector<std::vector<std::unique_ptr<ReadBatch>>>>&
@@ -248,7 +248,7 @@ class TestShuffleReader : public ShuffleReader {
   folly::F14FastMap<std::string, int64_t> stats() const override {
     return {
         {"test-shuffle.read", 1032},
-        {exec::ExchangeClient::kBackgroundCpuTimeMs, kFakeBackgroundCpuTimeMs}};
+        {exec::Operator::kBackgroundCpuTimeNanos, kFakeBackgroundCpuTimeNanos}};
   }
 
  private:
@@ -1033,7 +1033,7 @@ TEST_F(ShuffleTest, endToEnd) {
       numPartitions,
       numMapDrivers,
       {data},
-      kFakeBackgroundCpuTimeMs * Timestamp::kNanosecondsInMillisecond);
+      kFakeBackgroundCpuTimeNanos);
 }
 
 TEST_F(ShuffleTest, endToEndWithSortedShuffle) {
@@ -1074,7 +1074,7 @@ TEST_F(ShuffleTest, endToEndWithSortedShuffle) {
       numPartitions,
       numMapDrivers,
       {batch1, batch2},
-      kFakeBackgroundCpuTimeMs * Timestamp::kNanosecondsInMillisecond,
+      kFakeBackgroundCpuTimeNanos,
       ordering,
       fields,
       expectedSortingOrder);
@@ -1131,7 +1131,7 @@ TEST_F(ShuffleTest, endToEndWithSortedShuffleRowLimit) {
       numPartitions,
       numMapDrivers,
       {data},
-      kFakeBackgroundCpuTimeMs * Timestamp::kNanosecondsInMillisecond,
+      kFakeBackgroundCpuTimeNanos,
       ordering,
       fields,
       expectedSortingOrder,
@@ -1160,7 +1160,7 @@ TEST_F(ShuffleTest, endToEndWithReplicateNullAndAny) {
       numPartitions,
       numMapDrivers,
       {data},
-      kFakeBackgroundCpuTimeMs * Timestamp::kNanosecondsInMillisecond);
+      kFakeBackgroundCpuTimeNanos);
 }
 
 TEST_F(ShuffleTest, replicateNullsAndAny) {