C++ code changes in response to grpc_completion_queue_create() API change

1 files changed, 16 insertions, 4 deletions

diff --git a/test/cpp/microbenchmarks/bm_cq.cc b/test/cpp/microbenchmarks/bm_cq.cc
index c017474bf4..c7bd8743a5 100644
--- a/test/cpp/microbenchmarks/bm_cq.cc
+++ b/test/cpp/microbenchmarks/bm_cq.cc
@@ -66,7 +66,10 @@ BENCHMARK(BM_CreateDestroyCpp);
 
 static void BM_CreateDestroyCore(benchmark::State& state) {
   while (state.KeepRunning()) {
-    grpc_completion_queue_destroy(grpc_completion_queue_create(NULL));
+    // TODO: sreek Make this a templatized benchmark and pass completion type
+    // and polling type as parameters
+    grpc_completion_queue_destroy(
+        grpc_completion_queue_create(GRPC_CQ_NEXT, DEFAULT_POLLING, NULL));
   }
 }
 BENCHMARK(BM_CreateDestroyCore);
@@ -98,7 +101,10 @@ static void BM_Pass1Cpp(benchmark::State& state) {
 BENCHMARK(BM_Pass1Cpp);
 
 static void BM_Pass1Core(benchmark::State& state) {
-  grpc_completion_queue* cq = grpc_completion_queue_create(NULL);
+  // TODO: sreek Make this templatized benchmark and pass polling_type as a
+  // param
+  grpc_completion_queue* cq =
+      grpc_completion_queue_create(GRPC_CQ_NEXT, DEFAULT_POLLING, NULL);
   gpr_timespec deadline = gpr_inf_future(GPR_CLOCK_MONOTONIC);
   while (state.KeepRunning()) {
     grpc_cq_completion completion;
@@ -114,7 +120,10 @@ static void BM_Pass1Core(benchmark::State& state) {
 BENCHMARK(BM_Pass1Core);
 
 static void BM_Pluck1Core(benchmark::State& state) {
-  grpc_completion_queue* cq = grpc_completion_queue_create(NULL);
+  // TODO: sreek Make this templatized benchmark and pass polling_type as a
+  // param
+  grpc_completion_queue* cq =
+      grpc_completion_queue_create(GRPC_CQ_PLUCK, DEFAULT_POLLING, NULL);
   gpr_timespec deadline = gpr_inf_future(GPR_CLOCK_MONOTONIC);
   while (state.KeepRunning()) {
     grpc_cq_completion completion;
@@ -130,7 +139,10 @@ static void BM_Pluck1Core(benchmark::State& state) {
 BENCHMARK(BM_Pluck1Core);
 
 static void BM_EmptyCore(benchmark::State& state) {
-  grpc_completion_queue* cq = grpc_completion_queue_create(NULL);
+  // TODO: sreek Make this a templatized benchmark and pass polling_type as a
+  // param
+  grpc_completion_queue* cq =
+      grpc_completion_queue_create(GRPC_CQ_NEXT, DEFAULT_POLLING, NULL);
   gpr_timespec deadline = gpr_inf_past(GPR_CLOCK_MONOTONIC);
   while (state.KeepRunning()) {
     grpc_completion_queue_next(cq, deadline, NULL);