NextGraph
/
rocksdb
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

pristine code

Browse Source 
Summary:
This commit was generated using `mgt import`.
pristine code for third-party libraries:
third-party/benchmark

upgrade google benchmark to v1.6.1

contains a local patch that reverts [this](https://github.com/google/benchmark/pull/1227?fbclid=IwAR2CCmIJmjU62SPPQQf_t8kdAsMjYv_Pa_GxabYUOdQpGPZUHKwbnYS_1oE) and changs `enum Flags` to be `enum Flags : uint32_t`.

Reviewed By: chadaustin

Differential Revision: D35136540

fbshipit-source-id: f3662f953cd87956e5e9b767e55e3697f99d3b49
main
Jingjing Wang 3 years ago committed by Facebook GitHub Bot
parent 40e3f30a28
commit 5a085d789d
1 changed files with 30 additions and 30 deletions
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Show Stats Download Patch File Download Diff File
  1. 60
      microbench/db_basic_bench.cc

60
microbench/db_basic_bench.cc
Unescape View File

@ -262,10 +262,10 @@ static void DBPut(benchmark::State& state) {
} }
options.compaction_style = compaction_style; options.compaction_style = compaction_style;
auto rnd = Random(301 + state.thread_index); auto rnd = Random(301 + state.thread_index());
KeyGenerator kg(&rnd, key_num); KeyGenerator kg(&rnd, key_num);
if (state.thread_index == 0) { if (state.thread_index() == 0) {
SetupDB(state, options, &db, "DBPut"); SetupDB(state, options, &db, "DBPut");
} }
@ -283,7 +283,7 @@ static void DBPut(benchmark::State& state) {
} }
} }
if (state.thread_index == 0) { if (state.thread_index() == 0) {
auto db_full = static_cast_with_check<DBImpl>(db); auto db_full = static_cast_with_check<DBImpl>(db);
Status s = db_full->WaitForCompact(true); Status s = db_full->WaitForCompact(true);
if (!s.ok()) { if (!s.ok()) {
@ -346,10 +346,10 @@ static void ManualCompaction(benchmark::State& state) {
options.soft_pending_compaction_bytes_limit = 0; options.soft_pending_compaction_bytes_limit = 0;
options.hard_pending_compaction_bytes_limit = 0; options.hard_pending_compaction_bytes_limit = 0;
auto rnd = Random(301 + state.thread_index); auto rnd = Random(301 + state.thread_index());
KeyGenerator kg(&rnd, key_num); KeyGenerator kg(&rnd, key_num);
if (state.thread_index == 0) { if (state.thread_index() == 0) {
SetupDB(state, options, &db, "ManualCompaction"); SetupDB(state, options, &db, "ManualCompaction");
} }
@ -390,7 +390,7 @@ static void ManualCompaction(benchmark::State& state) {
} }
} }
if (state.thread_index == 0) { if (state.thread_index() == 0) {
auto db_full = static_cast_with_check<DBImpl>(db); auto db_full = static_cast_with_check<DBImpl>(db);
s = db_full->WaitForCompact(true); s = db_full->WaitForCompact(true);
if (!s.ok()) { if (!s.ok()) {
@ -466,10 +466,10 @@ static void ManualFlush(benchmark::State& state) {
options.hard_pending_compaction_bytes_limit = 0; options.hard_pending_compaction_bytes_limit = 0;
options.write_buffer_size = 2l << 30; // 2G to avoid auto flush options.write_buffer_size = 2l << 30; // 2G to avoid auto flush
auto rnd = Random(301 + state.thread_index); auto rnd = Random(301 + state.thread_index());
KeyGenerator kg(&rnd, key_num); KeyGenerator kg(&rnd, key_num);
if (state.thread_index == 0) { if (state.thread_index() == 0) {
SetupDB(state, options, &db, "ManualFlush"); SetupDB(state, options, &db, "ManualFlush");
} }
@ -488,7 +488,7 @@ static void ManualFlush(benchmark::State& state) {
} }
} }
if (state.thread_index == 0) { if (state.thread_index() == 0) {
auto db_full = static_cast_with_check<DBImpl>(db); auto db_full = static_cast_with_check<DBImpl>(db);
Status s = db_full->WaitForCompact(true); Status s = db_full->WaitForCompact(true);
if (!s.ok()) { if (!s.ok()) {
@ -541,10 +541,10 @@ static void DBGet(benchmark::State& state) {
options.table_factory.reset(NewBlockBasedTableFactory(table_options)); options.table_factory.reset(NewBlockBasedTableFactory(table_options));
} }
auto rnd = Random(301 + state.thread_index); auto rnd = Random(301 + state.thread_index());
KeyGenerator kg(&rnd, key_num); KeyGenerator kg(&rnd, key_num);
if (state.thread_index == 0) { if (state.thread_index() == 0) {
SetupDB(state, options, &db, "DBGet"); SetupDB(state, options, &db, "DBGet");
// load db // load db
@ -595,7 +595,7 @@ static void DBGet(benchmark::State& state) {
state.counters["neg_qu_pct"] = benchmark::Counter( state.counters["neg_qu_pct"] = benchmark::Counter(
static_cast<double>(not_found * 100), benchmark::Counter::kAvgIterations); static_cast<double>(not_found * 100), benchmark::Counter::kAvgIterations);
if (state.thread_index == 0) { if (state.thread_index() == 0) {
if (enable_statistics) { if (enable_statistics) {
HistogramData histogram_data; HistogramData histogram_data;
options.statistics->histogramData(DB_GET, &histogram_data); options.statistics->histogramData(DB_GET, &histogram_data);
@ -640,10 +640,10 @@ static void SimpleGetWithPerfContext(benchmark::State& state) {
options.create_if_missing = true; options.create_if_missing = true;
options.arena_block_size = 8 << 20; options.arena_block_size = 8 << 20;
auto rnd = Random(301 + state.thread_index); auto rnd = Random(301 + state.thread_index());
KeyGenerator kg(&rnd, 1024); KeyGenerator kg(&rnd, 1024);
if (state.thread_index == 0) { if (state.thread_index() == 0) {
auto env = Env::Default(); auto env = Env::Default();
std::string db_path; std::string db_path;
Status s = env->GetTestDirectory(&db_path); Status s = env->GetTestDirectory(&db_path);
@ -748,7 +748,7 @@ static void SimpleGetWithPerfContext(benchmark::State& state) {
benchmark::Counter(static_cast<double>(get_from_table_nanos), benchmark::Counter(static_cast<double>(get_from_table_nanos),
benchmark::Counter::kAvgIterations); benchmark::Counter::kAvgIterations);
if (state.thread_index == 0) { if (state.thread_index() == 0) {
TeardownDB(state, db, options, kg); TeardownDB(state, db, options, kg);
} }
} }
@ -857,10 +857,10 @@ static void IteratorSeek(benchmark::State& state) {
options.table_factory.reset(NewBlockBasedTableFactory(table_options)); options.table_factory.reset(NewBlockBasedTableFactory(table_options));
} }
auto rnd = Random(301 + state.thread_index); auto rnd = Random(301 + state.thread_index());
KeyGenerator kg(&rnd, key_num); KeyGenerator kg(&rnd, key_num);
if (state.thread_index == 0) { if (state.thread_index() == 0) {
SetupDB(state, options, &db, "IteratorSeek"); SetupDB(state, options, &db, "IteratorSeek");
// load db // load db
@ -905,7 +905,7 @@ static void IteratorSeek(benchmark::State& state) {
} }
} }
if (state.thread_index == 0) { if (state.thread_index() == 0) {
TeardownDB(state, db, options, kg); TeardownDB(state, db, options, kg);
} }
} }
@ -951,10 +951,10 @@ static void IteratorNext(benchmark::State& state) {
Options options; Options options;
options.compaction_style = compaction_style; options.compaction_style = compaction_style;
auto rnd = Random(301 + state.thread_index); auto rnd = Random(301 + state.thread_index());
KeyGenerator kg(&rnd, key_num); KeyGenerator kg(&rnd, key_num);
if (state.thread_index == 0) { if (state.thread_index() == 0) {
SetupDB(state, options, &db, "IteratorNext"); SetupDB(state, options, &db, "IteratorNext");
// load db // load db
auto wo = WriteOptions(); auto wo = WriteOptions();
@ -999,7 +999,7 @@ static void IteratorNext(benchmark::State& state) {
} }
} }
if (state.thread_index == 0) { if (state.thread_index() == 0) {
TeardownDB(state, db, options, kg); TeardownDB(state, db, options, kg);
} }
} }
@ -1025,10 +1025,10 @@ static void IteratorNextWithPerfContext(benchmark::State& state) {
static DB* db; static DB* db;
Options options; Options options;
auto rnd = Random(301 + state.thread_index); auto rnd = Random(301 + state.thread_index());
KeyGenerator kg(&rnd, 1024); KeyGenerator kg(&rnd, 1024);
if (state.thread_index == 0) { if (state.thread_index() == 0) {
SetupDB(state, options, &db, "IteratorNextWithPerfContext"); SetupDB(state, options, &db, "IteratorNextWithPerfContext");
// load db // load db
auto wo = WriteOptions(); auto wo = WriteOptions();
@ -1100,7 +1100,7 @@ static void IteratorNextWithPerfContext(benchmark::State& state) {
benchmark::Counter(static_cast<double>(iter_next_cpu_nanos), benchmark::Counter(static_cast<double>(iter_next_cpu_nanos),
benchmark::Counter::kAvgIterations); benchmark::Counter::kAvgIterations);
if (state.thread_index == 0) { if (state.thread_index() == 0) {
TeardownDB(state, db, options, kg); TeardownDB(state, db, options, kg);
} }
} }
@ -1119,10 +1119,10 @@ static void IteratorPrev(benchmark::State& state) {
Options options; Options options;
options.compaction_style = compaction_style; options.compaction_style = compaction_style;
auto rnd = Random(301 + state.thread_index); auto rnd = Random(301 + state.thread_index());
KeyGenerator kg(&rnd, key_num); KeyGenerator kg(&rnd, key_num);
if (state.thread_index == 0) { if (state.thread_index() == 0) {
SetupDB(state, options, &db, "IteratorPrev"); SetupDB(state, options, &db, "IteratorPrev");
// load db // load db
auto wo = WriteOptions(); auto wo = WriteOptions();
@ -1167,7 +1167,7 @@ static void IteratorPrev(benchmark::State& state) {
} }
} }
if (state.thread_index == 0) { if (state.thread_index() == 0) {
TeardownDB(state, db, options, kg); TeardownDB(state, db, options, kg);
} }
} }
@ -1212,10 +1212,10 @@ static void PrefixSeek(benchmark::State& state) {
options.table_factory.reset(NewBlockBasedTableFactory(table_options)); options.table_factory.reset(NewBlockBasedTableFactory(table_options));
} }
auto rnd = Random(301 + state.thread_index); auto rnd = Random(301 + state.thread_index());
KeyGenerator kg(&rnd, key_num, key_num / 100); KeyGenerator kg(&rnd, key_num, key_num / 100);
if (state.thread_index == 0) { if (state.thread_index() == 0) {
SetupDB(state, options, &db, "PrefixSeek"); SetupDB(state, options, &db, "PrefixSeek");
// load db // load db
@ -1259,7 +1259,7 @@ static void PrefixSeek(benchmark::State& state) {
} }
} }
if (state.thread_index == 0) { if (state.thread_index() == 0) {
TeardownDB(state, db, options, kg); TeardownDB(state, db, options, kg);
} }
} }

Write Preview
Loading…
Cancel
Save