test/cpp/microbenchmarks/bm_fullstack_streaming_pump.cc

   1 /*
   2  *
   3  * Copyright 2016 gRPC authors.
   4  *
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *     http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  *
  17  */
  18
  19 /* Benchmark gRPC end2end in various configurations */
  20
  21 #include "test/cpp/microbenchmarks/fullstack_streaming_pump.h"
  22 #include "test/cpp/util/test_config.h"
  23
  24 namespace grpc {
  25 namespace testing {
  26
  27 /*******************************************************************************
  28  * CONFIGURATIONS
  29  */
  30
  31 // force library initialization
  32 auto& force_library_initialization = Library::get();
  33
  34 BENCHMARK_TEMPLATE(BM_PumpStreamClientToServer, TCP)
  35     ->Range(0, 128 * 1024 * 1024);
  36 BENCHMARK_TEMPLATE(BM_PumpStreamClientToServer, UDS)
  37     ->Range(0, 128 * 1024 * 1024);
  38 BENCHMARK_TEMPLATE(BM_PumpStreamClientToServer, InProcess)
  39     ->Range(0, 128 * 1024 * 1024);
  40 BENCHMARK_TEMPLATE(BM_PumpStreamClientToServer, SockPair)
  41     ->Range(0, 128 * 1024 * 1024);
  42 BENCHMARK_TEMPLATE(BM_PumpStreamClientToServer, InProcessCHTTP2)
  43     ->Range(0, 128 * 1024 * 1024);
  44 BENCHMARK_TEMPLATE(BM_PumpStreamServerToClient, TCP)
  45     ->Range(0, 128 * 1024 * 1024);
  46 BENCHMARK_TEMPLATE(BM_PumpStreamServerToClient, UDS)
  47     ->Range(0, 128 * 1024 * 1024);
  48 BENCHMARK_TEMPLATE(BM_PumpStreamServerToClient, InProcess)
  49     ->Range(0, 128 * 1024 * 1024);
  50 BENCHMARK_TEMPLATE(BM_PumpStreamServerToClient, SockPair)
  51     ->Range(0, 128 * 1024 * 1024);
  52 BENCHMARK_TEMPLATE(BM_PumpStreamServerToClient, InProcessCHTTP2)
  53     ->Range(0, 128 * 1024 * 1024);
  54 BENCHMARK_TEMPLATE(BM_PumpStreamClientToServer, MinTCP)->Arg(0);
  55 BENCHMARK_TEMPLATE(BM_PumpStreamClientToServer, MinUDS)->Arg(0);
  56 BENCHMARK_TEMPLATE(BM_PumpStreamClientToServer, MinInProcess)->Arg(0);
  57 BENCHMARK_TEMPLATE(BM_PumpStreamClientToServer, MinSockPair)->Arg(0);
  58 BENCHMARK_TEMPLATE(BM_PumpStreamClientToServer, MinInProcessCHTTP2)->Arg(0);
  59 BENCHMARK_TEMPLATE(BM_PumpStreamServerToClient, MinTCP)->Arg(0);
  60 BENCHMARK_TEMPLATE(BM_PumpStreamServerToClient, MinUDS)->Arg(0);
  61 BENCHMARK_TEMPLATE(BM_PumpStreamServerToClient, MinInProcess)->Arg(0);
  62 BENCHMARK_TEMPLATE(BM_PumpStreamServerToClient, MinSockPair)->Arg(0);
  63 BENCHMARK_TEMPLATE(BM_PumpStreamServerToClient, MinInProcessCHTTP2)->Arg(0);
  64
  65 }  // namespace testing
  66 }  // namespace grpc
  67
  68 // Some distros have RunSpecifiedBenchmarks under the benchmark namespace,
  69 // and others do not. This allows us to support both modes.
  70 namespace benchmark {
  71 void RunTheBenchmarksNamespaced() { RunSpecifiedBenchmarks(); }
  72 }  // namespace benchmark
  73
  74 int main(int argc, char** argv) {
  75   ::benchmark::Initialize(&argc, argv);
  76   ::grpc::testing::InitTest(&argc, &argv, false);
  77   benchmark::RunTheBenchmarksNamespaced();
  78   return 0;
  79 }