tdlight-team/tdlight
tdlight-team/tdlight
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix benchmark/bench_db.cpp

Browse Source
This commit is contained in:
Arseny Smirnov 2023-02-08 10:46:37 +01:00
parent 6ba394fc41
commit fb854c93a8
1 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
benchmark/bench_db.cpp
View File

@ -29,7 +29,7 @@
template <class KeyValueT> template <class KeyValueT>
class TdKvBench final : public td::Benchmark { class TdKvBench final : public td::Benchmark {
td::ConcurrentScheduler sched{1, 0}; td::unique_ptr<td::ConcurrentScheduler> scheduler_;
td::string name_; td::string name_;
public: public:
@ -72,18 +72,20 @@ class TdKvBench final : public td::Benchmark {
}; };
void start_up_n(int n) final { void start_up_n(int n) final {
sched.create_actor_unsafe<Main>(1, "Main", n).release(); scheduler_ = td::make_unique<td::ConcurrentScheduler>(1, 0);
scheduler_->create_actor_unsafe<Main>(1, "Main", n).release();
} }
void run(int n) final { void run(int n) final {
sched.start(); scheduler_->start();
while (sched.run_main(10)) { while (scheduler_->run_main(10)) {
// empty // empty
} }
sched.finish(); scheduler_->finish();
} }
void tear_down() final { void tear_down() final {
scheduler_.reset();
} }
}; };
@ -184,6 +186,7 @@ class SqliteKeyValueAsyncBench final : public td::Benchmark {
td::string sql_db_name = "testdb.sqlite"; td::string sql_db_name = "testdb.sqlite";
td::SqliteDb::destroy(sql_db_name).ignore(); 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<td::SqliteConnectionSafe>(sql_db_name, td::DbKey::empty()); sql_connection_ = std::make_shared<td::SqliteConnectionSafe>(sql_db_name, td::DbKey::empty());
auto &db = sql_connection_->get(); auto &db = sql_connection_->get();
@ -229,12 +232,13 @@ class BinlogKeyValueBench final : public td::Benchmark {
int main() { int main() {
SET_VERBOSITY_LEVEL(VERBOSITY_NAME(WARNING)); SET_VERBOSITY_LEVEL(VERBOSITY_NAME(WARNING));
bench(TdKvBench<td::BinlogKeyValue<td::Binlog>>("BinlogKeyValue<Binlog>"));
bench(TdKvBench<td::BinlogKeyValue<td::ConcurrentBinlog>>("BinlogKeyValue<ConcurrentBinlog>"));
bench(BinlogKeyValueBench<true>()); bench(BinlogKeyValueBench<true>());
bench(BinlogKeyValueBench<false>()); bench(BinlogKeyValueBench<false>());
bench(SqliteKVBench<false>()); bench(SqliteKVBench<false>());
bench(SqliteKVBench<true>()); bench(SqliteKVBench<true>());
bench(SqliteKeyValueAsyncBench()); bench(SqliteKeyValueAsyncBench());
bench(TdKvBench<td::BinlogKeyValue<td::Binlog>>("BinlogKeyValue<Binlog>"));
bench(TdKvBench<td::BinlogKeyValue<td::ConcurrentBinlog>>("BinlogKeyValue<ConcurrentBinlog>"));
bench(SeqKvBench()); bench(SeqKvBench());
} }