Record presto exchange source request duration and num of retries

Author: kewang1024

presto-native-execution/presto_cpp/main/PrestoExchangeSource.cpp

@@ -255,6 +255,10 @@ void PrestoExchangeSource::handleDataResponse(
 
 void PrestoExchangeSource::processDataResponse(
     std::unique_ptr<http::HttpResponse> response) {
+  RECORD_HISTOGRAM_METRIC_VALUE(
+      kCounterExchangeRequestDuration, dataRequestRetryState_.durationMs());
+  RECORD_HISTOGRAM_METRIC_VALUE(
+      kCounterExchangeRequestNumRetries, dataRequestRetryState_.numTries());
   if (closed_.load()) {
     // If PrestoExchangeSource is already closed, just free all buffers
     // allocated without doing any processing. This can happen when a super slow

presto-native-execution/presto_cpp/main/PrestoExchangeSource.h

@@ -56,6 +56,10 @@ class PrestoExchangeSource : public velox::exec::ExchangeSource {
       return velox::getCurrentTimeMs() - startMs_;
     }
 
+    size_t numTries() const {
+      return numTries_;
+    }
+
     // Returns whether we have exhausted all retries. We only retry if we spent
     // less than maxWaitMs_ time after we first started.
     bool isExhausted() const {

presto-native-execution/presto_cpp/main/common/Counters.cpp

@@ -108,6 +108,29 @@ void registerPrestoMetrics() {
       95,
       99,
       100);
+
+  // Tracks exchange request duration in range of [0, 300s] with
+  // 300 buckets and reports P50, P90, P99, and P100.
+  DEFINE_HISTOGRAM_METRIC(
+      kCounterExchangeRequestDuration,
+      1'000,
+      0,
+      300'000,
+      50,
+      90,
+      99,
+      100);
+  // Tracks exchange request num of retris in range of [0, 20] with
+  // 20 buckets and reports P50, P90, P99, and P100.
+  DEFINE_HISTOGRAM_METRIC(
+      kCounterExchangeRequestNumRetries,
+      1,
+      0,
+      20,
+      50,
+      90,
+      99,
+      100);
   DEFINE_METRIC(kCounterMemoryPushbackCount, facebook::velox::StatType::COUNT);
   DEFINE_HISTOGRAM_METRIC(
       kCounterMemoryPushbackLatencyMs, 10'000, 0, 100'000, 50, 90, 99, 100);

presto-native-execution/presto_cpp/main/common/Counters.h

@@ -47,6 +47,11 @@ constexpr folly::StringPiece kCounterHTTPClientTransactionCreateDelay{
 constexpr folly::StringPiece kCounterExchangeSourcePeakQueuedBytes{
     "presto_cpp.exchange_source_peak_queued_bytes"};
 
+constexpr folly::StringPiece kCounterExchangeRequestDuration{
+    "presto_cpp.exchange.request.duration"};
+constexpr folly::StringPiece kCounterExchangeRequestNumRetries{
+    "presto_cpp.exchange.request.num_retries"};
+
 constexpr folly::StringPiece kCounterNumQueryContexts{
     "presto_cpp.num_query_contexts"};