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