folly/stats/test/HistogramBenchmark.cpp

   1 /*
   2  * Copyright 2012-present 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 #include <folly/stats/Histogram.h>
  17
  18 #include <folly/Benchmark.h>
  19 #include <folly/container/Foreach.h>
  20 #include <folly/portability/GFlags.h>
  21
  22 using folly::Histogram;
  23
  24 void addValue(unsigned int n, int64_t bucketSize, int64_t min, int64_t max) {
  25   Histogram<int64_t> hist(bucketSize, min, max);
  26   int64_t num = min;
  27   FOR_EACH_RANGE (i, 0, n) {
  28     hist.addValue(num);
  29     ++num;
  30     if (num > max) {
  31       num = min;
  32     }
  33   }
  34 }
  35
  36 BENCHMARK_NAMED_PARAM(addValue, 0_to_100, 1, 0, 100);
  37 BENCHMARK_NAMED_PARAM(addValue, 0_to_1000, 10, 0, 1000);
  38 BENCHMARK_NAMED_PARAM(addValue, 5k_to_20k, 250, 5000, 20000);
  39
  40 int main(int argc, char* argv[]) {
  41   gflags::ParseCommandLineFlags(&argc, &argv, true);
  42   folly::runBenchmarks();
  43   return 0;
  44 }