folly/io/async/test/EventBaseBenchmark.cpp

   1 /*
   2  * Copyright 2015 Facebook, Inc.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *   http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #include <folly/Benchmark.h>
  18 #include <folly/io/async/EventBase.h>
  19
  20 #include <gflags/gflags.h>
  21
  22 using namespace folly;
  23
  24 class CountedLoopCallback : public EventBase::LoopCallback {
  25  public:
  26   CountedLoopCallback(EventBase* eventBase, unsigned int count)
  27     : eventBase_(eventBase)
  28     , count_(count) {}
  29
  30   void runLoopCallback() noexcept override {
  31     --count_;
  32     if (count_ > 0) {
  33       eventBase_->runInLoop(this);
  34     }
  35   }
  36
  37  private:
  38   EventBase* eventBase_;
  39   unsigned int count_;
  40 };
  41
  42 BENCHMARK(timeMeasurementsOn, n) {
  43   EventBase eventBase;
  44
  45   while (n--) {
  46     CountedLoopCallback c(&eventBase, 10);
  47     eventBase.runInLoop(&c);
  48     eventBase.loop();
  49   }
  50 }
  51
  52 BENCHMARK_RELATIVE(timeMeasurementsOff, n) {
  53   EventBase eventBase(/* enableTimeMeasurement */ false);
  54
  55   while (n--) {
  56     CountedLoopCallback c(&eventBase, 10);
  57     eventBase.runInLoop(&c);
  58     eventBase.loop();
  59   }
  60 }
  61
  62 /**
  63  * --bm_min_iters=1000000
  64  *
  65  * ============================================================================
  66  * folly/io/async/test/EventBaseBenchmark.cpp      relative  time/iter  iters/s
  67  * ============================================================================
  68  * timeMeasurementsOn                                           2.02us  494.57K
  69  * timeMeasurementsOff                              231.19%   874.58ns    1.14M
  70  * ============================================================================
  71  */
  72
  73 int main(int argc, char** argv) {
  74   gflags::ParseCommandLineFlags(&argc, &argv, true);
  75   runBenchmarks();
  76 }