| /* |
| * |
| * Copyright 2019 gRPC authors. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| * |
| */ |
| |
| #ifndef TEST_CPP_MICROBENCHMARKS_CALLBACK_STREAMING_PING_PONG_H |
| #define TEST_CPP_MICROBENCHMARKS_CALLBACK_STREAMING_PING_PONG_H |
| |
| #include <benchmark/benchmark.h> |
| #include <sstream> |
| #include "src/core/lib/profiling/timers.h" |
| #include "src/proto/grpc/testing/echo.grpc.pb.h" |
| #include "test/cpp/microbenchmarks/callback_test_service.h" |
| #include "test/cpp/microbenchmarks/fullstack_context_mutators.h" |
| #include "test/cpp/microbenchmarks/fullstack_fixtures.h" |
| |
| namespace grpc { |
| namespace testing { |
| |
| /******************************************************************************* |
| * BENCHMARKING KERNELS |
| */ |
| |
| template <class Fixture, class ClientContextMutator, class ServerContextMutator> |
| static void BM_CallbackBidiStreaming(benchmark::State& state) { |
| const int message_size = state.range(0); |
| const int max_ping_pongs = state.range(1); |
| CallbackStreamingTestService service; |
| std::unique_ptr<Fixture> fixture(new Fixture(&service)); |
| std::unique_ptr<EchoTestService::Stub> stub_( |
| EchoTestService::NewStub(fixture->channel())); |
| EchoRequest request; |
| EchoResponse response; |
| if (message_size > 0) { |
| request.set_message(std::string(message_size, 'a')); |
| } else { |
| request.set_message(""); |
| } |
| if (state.KeepRunning()) { |
| GPR_TIMER_SCOPE("BenchmarkCycle", 0); |
| std::mutex mu; |
| std::condition_variable cv; |
| bool done = false; |
| BidiClient* test = |
| new BidiClient(&state, stub_.get(), &request, &response, |
| &mu, &cv, &done); |
| test->StartNewRpc(); |
| std::unique_lock<std::mutex> l(mu); |
| while (!done) { |
| cv.wait(l); |
| } |
| } |
| fixture->Finish(state); |
| fixture.reset(); |
| state.SetBytesProcessed(2 * message_size * max_ping_pongs * |
| state.iterations()); |
| } |
| |
| } // namespace testing |
| } // namespace grpc |
| #endif // TEST_CPP_MICROBENCHMARKS_CALLBACK_STREAMING_PING_PONG_H |