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

Add get_database_scheduler_id().

Browse Source
This commit is contained in:
levlam 2022-06-20 22:02:45 +03:00
parent 854b67d1e3
commit 1528cfb9da
2 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
td/telegram/Td.cpp
View File

@ -3647,6 +3647,12 @@ void Td::complete_pending_preauthentication_requests(const T &func) {
} }
} }
int32 Td::get_database_scheduler_id() {
auto current_scheduler_id = Scheduler::instance()->sched_id();
auto scheduler_count = Scheduler::instance()->sched_count();
return min(current_scheduler_id + 1, scheduler_count - 1);
}
void Td::start_init(uint64 id, string &&key) { void Td::start_init(uint64 id, string &&key) {
VLOG(td_init) << "Begin to init database"; VLOG(td_init) << "Begin to init database";
init_request_id_ = id; init_request_id_ = id;
@ -3654,10 +3660,7 @@ void Td::start_init(uint64 id, string &&key) {
auto promise = PromiseCreator::lambda([actor_id = actor_id(this)](Result<TdDb::OpenedDatabase> r_opened_database) { auto promise = PromiseCreator::lambda([actor_id = actor_id(this)](Result<TdDb::OpenedDatabase> r_opened_database) {
send_closure(actor_id, &Td::init, std::move(r_opened_database)); send_closure(actor_id, &Td::init, std::move(r_opened_database));
}); });
auto current_scheduler_id = Scheduler::instance()->sched_id(); TdDb::open(get_database_scheduler_id(), parameters_, as_db_key(std::move(key)), std::move(promise));
auto scheduler_count = Scheduler::instance()->sched_count();
TdDb::open(min(current_scheduler_id + 1, scheduler_count - 1), parameters_, as_db_key(std::move(key)),
std::move(promise));
} }
void Td::init(Result<TdDb::OpenedDatabase> r_opened_database) { void Td::init(Result<TdDb::OpenedDatabase> r_opened_database) {

2
td/telegram/Td.h
View File

@ -1426,6 +1426,8 @@ class Td final : public Actor {
static DbKey as_db_key(string key); static DbKey as_db_key(string key);
static int32 get_database_scheduler_id();
void start_init(uint64 id, string &&key); void start_init(uint64 id, string &&key);
void init(Result<TdDb::OpenedDatabase> r_opened_database); void init(Result<TdDb::OpenedDatabase> r_opened_database);