initial implementation.

1 files changed, 15 insertions, 0 deletions

diff --git a/test/cpp/qps/client_async.cc b/test/cpp/qps/client_async.cc
index 29a79e7343..2e4b7acba7 100644
--- a/test/cpp/qps/client_async.cc
+++ b/test/cpp/qps/client_async.cc
@@ -54,6 +54,10 @@
 #include "test/cpp/qps/usage_timer.h"
 #include "test/cpp/util/create_test_channel.h"
 
+extern "C" {
+#include "src/core/lib/surface/completion_queue.h"
+}
+
 namespace grpc {
 namespace testing {
 
@@ -205,6 +209,17 @@ class AsyncClient : public ClientImpl<StubType, RequestType> {
     }
   }
 
+int GetPollCount() {
+  int count = 0;
+  int i = 0;
+  for (auto cq = cli_cqs_.begin(); cq != cli_cqs_.end(); cq++) {
+    int k = (int)grpc_get_cq_poll_num((*cq)->cq());
+    gpr_log(GPR_INFO, "%d: per cq poll:%d", i++, k);
+    count += k;
+  }
+  return count;
+}
+
  protected:
   const int num_async_threads_;