Latency traces

1 files changed, 7 insertions, 0 deletions

diff --git a/test/cpp/qps/client_sync.cc b/test/cpp/qps/client_sync.cc
index beef604856..0523371013 100644
--- a/test/cpp/qps/client_sync.cc
+++ b/test/cpp/qps/client_sync.cc
@@ -59,6 +59,8 @@
 #include "test/cpp/qps/interarrival.h"
 #include "test/cpp/qps/timer.h"
 
+#include "src/core/profiling/timers.h"
+
 namespace grpc {
 namespace testing {
 
@@ -100,8 +102,10 @@ class SynchronousUnaryClient GRPC_FINAL : public SynchronousClient {
     auto* stub = channels_[thread_idx % channels_.size()].get_stub();
     double start = Timer::Now();
     grpc::ClientContext context;
+    GRPC_TIMER_BEGIN(GRPC_PTAG_CLIENT_UNARY_CALL, 0);
     grpc::Status s =
         stub->UnaryCall(&context, request_, &responses_[thread_idx]);
+    GRPC_TIMER_END(GRPC_PTAG_CLIENT_UNARY_CALL, 0);
     histogram->Add((Timer::Now() - start) * 1e9);
     return s.ok();
   }
@@ -136,11 +140,14 @@ class SynchronousStreamingClient GRPC_FINAL : public SynchronousClient {
   bool ThreadFunc(Histogram* histogram, size_t thread_idx) GRPC_OVERRIDE {
     WaitToIssue(thread_idx);
     double start = Timer::Now();
+    GRPC_TIMER_BEGIN(GRPC_PTAG_CLIENT_UNARY_CALL, 0);
     if (stream_[thread_idx]->Write(request_) &&
         stream_[thread_idx]->Read(&responses_[thread_idx])) {
+      GRPC_TIMER_END(GRPC_PTAG_CLIENT_UNARY_CALL, 0);
       histogram->Add((Timer::Now() - start) * 1e9);
       return true;
     }
+    GRPC_TIMER_END(GRPC_PTAG_CLIENT_UNARY_CALL, 0);
     return false;
   }