diff --git a/tools/db_bench_tool.cc b/tools/db_bench_tool.cc index 3fa45a693..c36596b95 100644 --- a/tools/db_bench_tool.cc +++ b/tools/db_bench_tool.cc @@ -4944,6 +4944,7 @@ void VerifyDBFromDB(std::string& truth_db_name) { if (FLAGS_perf_level > rocksdb::PerfLevel::kDisable) { thread->stats.AddMessage(get_perf_context()->ToString()); } + thread->stats.AddBytes(static_cast(inserter.GetBytesInserted())); } // Verifies consistency of data after RandomTransaction() has been run. diff --git a/util/transaction_test_util.cc b/util/transaction_test_util.cc index 0d6948b08..d439cf202 100644 --- a/util/transaction_test_util.cc +++ b/util/transaction_test_util.cc @@ -132,6 +132,7 @@ bool RandomTransactionInserter::DoInsert(DB* db, Transaction* txn, } else { batch.Put(key, sum); } + bytes_inserted_ += key.size() + sum.size(); } } diff --git a/util/transaction_test_util.h b/util/transaction_test_util.h index 8805490fd..768fc5283 100644 --- a/util/transaction_test_util.h +++ b/util/transaction_test_util.h @@ -83,6 +83,9 @@ class RandomTransactionInserter { // write any data. uint64_t GetFailureCount() { return failure_count_; } + // Returns the sum of user keys/values Put() to the DB. + size_t GetBytesInserted() { return bytes_inserted_; } + private: // Input options Random64* rand_; @@ -97,6 +100,8 @@ class RandomTransactionInserter { // Number of failed insert batches attempted uint64_t failure_count_ = 0; + size_t bytes_inserted_ = 0; + // Status returned by most recent insert operation Status last_status_;