diff --git a/benchmark/bench_db.cpp b/benchmark/bench_db.cpp index 0425b018c..b47fa98bb 100644 --- a/benchmark/bench_db.cpp +++ b/benchmark/bench_db.cpp @@ -29,7 +29,7 @@ template class TdKvBench final : public td::Benchmark { - td::ConcurrentScheduler sched{1, 0}; + td::unique_ptr scheduler_; td::string name_; public: @@ -72,18 +72,20 @@ class TdKvBench final : public td::Benchmark { }; void start_up_n(int n) final { - sched.create_actor_unsafe
(1, "Main", n).release(); + scheduler_ = td::make_unique(1, 0); + scheduler_->create_actor_unsafe
(1, "Main", n).release(); } void run(int n) final { - sched.start(); - while (sched.run_main(10)) { + scheduler_->start(); + while (scheduler_->run_main(10)) { // empty } - sched.finish(); + scheduler_->finish(); } void tear_down() final { + scheduler_.reset(); } }; @@ -184,6 +186,7 @@ class SqliteKeyValueAsyncBench final : public td::Benchmark { td::string sql_db_name = "testdb.sqlite"; td::SqliteDb::destroy(sql_db_name).ignore(); + td::SqliteDb::open_with_key(sql_db_name, true, td::DbKey::empty()).move_as_ok(); sql_connection_ = std::make_shared(sql_db_name, td::DbKey::empty()); auto &db = sql_connection_->get(); @@ -229,12 +232,13 @@ class BinlogKeyValueBench final : public td::Benchmark { int main() { SET_VERBOSITY_LEVEL(VERBOSITY_NAME(WARNING)); + bench(TdKvBench>("BinlogKeyValue")); + bench(TdKvBench>("BinlogKeyValue")); + bench(BinlogKeyValueBench()); bench(BinlogKeyValueBench()); bench(SqliteKVBench()); bench(SqliteKVBench()); bench(SqliteKeyValueAsyncBench()); - bench(TdKvBench>("BinlogKeyValue")); - bench(TdKvBench>("BinlogKeyValue")); bench(SeqKvBench()); }