From d58f804b8f5a03c77104e5bf1ebf4a2cb6c418dd Mon Sep 17 00:00:00 2001
From: fuzhe1989 <fuzhe@pingcap.com>
Date: Thu, 17 Mar 2022 16:16:20 +0800
Subject: [PATCH] update

Signed-off-by: fuzhe1989 <fuzhe@pingcap.com>
---
 dbms/src/Common/TiFlashMetrics.h        |  2 ++
 dbms/src/Flash/Mpp/ExchangeReceiver.cpp | 12 ++++++++++++
 2 files changed, 14 insertions(+)

diff --git a/dbms/src/Common/TiFlashMetrics.h b/dbms/src/Common/TiFlashMetrics.h
index 8c87591328a..964f2f9fd11 100644
--- a/dbms/src/Common/TiFlashMetrics.h
+++ b/dbms/src/Common/TiFlashMetrics.h
@@ -180,6 +180,8 @@ namespace DB
         F(type_max_threads_of_raw, {"type", "total_max"}),                                                                                \
         F(type_total_threads_of_raw, {"type", "total"}),                                                                                  \
         F(type_threads_of_client_cq_pool, {"type", "rpc_client_cq_pool"}),                                                                \
+        F(type_threads_of_receiver_read_loop, {"type", "rpc_receiver_read_loop"}),                                                        \
+        F(type_threads_of_receiver_reactor, {"type", "rpc_receiver_reactor"}),                                                            \
         F(type_max_threads_of_establish_mpp, {"type", "rpc_establish_mpp_max"}),                                                          \
         F(type_active_threads_of_establish_mpp, {"type", "rpc_establish_mpp"}),                                                           \
         F(type_max_threads_of_dispatch_mpp, {"type", "rpc_dispatch_mpp_max"}),                                                            \
diff --git a/dbms/src/Flash/Mpp/ExchangeReceiver.cpp b/dbms/src/Flash/Mpp/ExchangeReceiver.cpp
index 8f5b2819402..b7873fd5e62 100644
--- a/dbms/src/Flash/Mpp/ExchangeReceiver.cpp
+++ b/dbms/src/Flash/Mpp/ExchangeReceiver.cpp
@@ -14,6 +14,7 @@
 
 #include <Common/CPUAffinityManager.h>
 #include <Common/ThreadFactory.h>
+#include <Common/TiFlashMetrics.h>
 #include <Flash/Coprocessor/CoprocessorReader.h>
 #include <Flash/Mpp/ExchangeReceiver.h>
 #include <Flash/Mpp/MPPTunnel.h>
@@ -350,6 +351,12 @@ template <typename RPCContext>
 void ExchangeReceiverBase<RPCContext>::reactor(const std::vector<Request> & async_requests)
 {
     using AsyncHandler = AsyncRequestHandler<RPCContext>;
+
+    GET_METRIC(tiflash_thread_count, type_threads_of_receiver_reactor).Increment();
+    SCOPE_EXIT({
+        GET_METRIC(tiflash_thread_count, type_threads_of_receiver_reactor).Decrement();
+    });
+
     CPUAffinityManager::getInstance().bindSelfQueryThread();
 
     size_t alive_async_connections = async_requests.size();
@@ -406,6 +413,11 @@ void ExchangeReceiverBase<RPCContext>::reactor(const std::vector<Request> & asyn
 template <typename RPCContext>
 void ExchangeReceiverBase<RPCContext>::readLoop(const Request & req)
 {
+    GET_METRIC(tiflash_thread_count, type_threads_of_receiver_read_loop).Increment();
+    SCOPE_EXIT({
+        GET_METRIC(tiflash_thread_count, type_threads_of_receiver_read_loop).Decrement();
+    });
+
     CPUAffinityManager::getInstance().bindSelfQueryThread();
     bool meet_error = false;
     String local_err_msg;