Fix bm_chttp2_transport

8 years ago · 97a0ded7c5
parent 50da9d83b6
commit 97a0ded7c5
1 changed files with 4 additions and 2 deletions
--- a/test/cpp/microbenchmarks/bm_chttp2_transport.cc
+++ b/test/cpp/microbenchmarks/bm_chttp2_transport.cc
@ -572,11 +572,13 @@ static void BM_TransportStreamRecv(benchmark::State &state) {
    } while (grpc_byte_stream_next(exec_ctx, recv_stream,
                                   recv_stream->length - received,
                                   drain_continue.get()) &&
-             GRPC_ERROR_NONE == grpc_byte_stream_pull(exec_ctx, recv_stream,
-                                                      &recv_slice));
+             GRPC_ERROR_NONE ==
+                 grpc_byte_stream_pull(exec_ctx, recv_stream, &recv_slice) &&
+             (received += GRPC_SLICE_LENGTH(recv_slice), true));
  });

  drain_continue = MakeClosure([&](grpc_exec_ctx *exec_ctx, grpc_error *error) {
+    grpc_byte_stream_pull(exec_ctx, recv_stream, &recv_slice);
    received += GRPC_SLICE_LENGTH(recv_slice);
    grpc_slice_unref_internal(exec_ctx, recv_slice);
    grpc_closure_run(exec_ctx, drain.get(), GRPC_ERROR_NONE);