folly/test/HistogramBenchmark.cpp

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