test/cpp/microbenchmarks/bm_channel.cc

   1 /*
   2  *
   3  * Copyright 2017 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 channel */
  20
  21 #include <benchmark/benchmark.h>
  22
  23 #include <grpc/grpc.h>
  24
  25 #include "test/core/util/test_config.h"
  26 #include "test/cpp/microbenchmarks/helpers.h"
  27 #include "test/cpp/util/test_config.h"
  28
  29 class ChannelDestroyerFixture {
  30  public:
  31   ChannelDestroyerFixture() {}
  32   virtual ~ChannelDestroyerFixture() {
  33     if (channel_) {
  34       grpc_channel_destroy(channel_);
  35     }
  36   }
  37   virtual void Init() = 0;
  38
  39  protected:
  40   grpc_channel* channel_ = nullptr;
  41 };
  42
  43 class InsecureChannelFixture : public ChannelDestroyerFixture {
  44  public:
  45   InsecureChannelFixture() {}
  46   void Init() override {
  47     channel_ = grpc_insecure_channel_create("localhost:1234", nullptr, nullptr);
  48   }
  49 };
  50
  51 class LameChannelFixture : public ChannelDestroyerFixture {
  52  public:
  53   LameChannelFixture() {}
  54   void Init() override {
  55     channel_ = grpc_lame_client_channel_create(
  56         "localhost:1234", GRPC_STATUS_UNAUTHENTICATED, "blah");
  57   }
  58 };
  59
  60 template <class Fixture>
  61 static void BM_InsecureChannelCreateDestroy(benchmark::State& state) {
  62   // In order to test if channel creation time is affected by the number of
  63   // already existing channels, we create some initial channels here.
  64   Fixture initial_channels[512];
  65   for (int i = 0; i < state.range(0); i++) {
  66     initial_channels[i].Init();
  67   }
  68   for (auto _ : state) {
  69     Fixture channel;
  70     channel.Init();
  71   }
  72 }
  73 BENCHMARK_TEMPLATE(BM_InsecureChannelCreateDestroy, InsecureChannelFixture)
  74     ->Range(0, 512);
  75 ;
  76 BENCHMARK_TEMPLATE(BM_InsecureChannelCreateDestroy, LameChannelFixture)
  77     ->Range(0, 512);
  78 ;
  79
  80 // Some distros have RunSpecifiedBenchmarks under the benchmark namespace,
  81 // and others do not. This allows us to support both modes.
  82 namespace benchmark {
  83 void RunTheBenchmarksNamespaced() { RunSpecifiedBenchmarks(); }
  84 }  // namespace benchmark
  85
  86 int main(int argc, char** argv) {
  87   grpc::testing::TestEnvironment env(argc, argv);
  88   LibraryInitializer libInit;
  89   ::benchmark::Initialize(&argc, argv);
  90   ::grpc::testing::InitTest(&argc, &argv, false);
  91   benchmark::RunTheBenchmarksNamespaced();
  92   return 0;
  93 }