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

Add ContactsManager::finish_load_channel_recommendations_queries.

Browse Source
This commit is contained in:
levlam 2023-11-29 15:11:55 +03:00
parent fa91aeb065
commit 85b06f9614
2 changed files with 22 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
td/telegram/ContactsManager.cpp
View File

@ -9803,6 +9803,21 @@ void ContactsManager::fail_load_channel_recommendations_queries(ChannelId channe
fail_promises(promises, std::move(error)); fail_promises(promises, std::move(error));
} }
void ContactsManager::finish_load_channel_recommendations_queries(ChannelId channel_id, int32 total_count,
vector<DialogId> dialog_ids) {
auto it = get_channel_recommendations_queries_.find(channel_id);
CHECK(it != get_channel_recommendations_queries_.end());
auto promises = std::move(it->second);
CHECK(!promises.empty());
get_channel_recommendations_queries_.erase(it);
for (auto &promise : promises) {
if (promise) {
promise.set_value(td_->messages_manager_->get_chats_object(total_count, dialog_ids,
"finish_load_channel_recommendations_queries"));
}
}
}
void ContactsManager::on_load_channel_recommendations_from_database(ChannelId channel_id, string value) { void ContactsManager::on_load_channel_recommendations_from_database(ChannelId channel_id, string value) {
if (G()->close_flag()) { if (G()->close_flag()) {
return fail_load_channel_recommendations_queries(channel_id, G()->close_status()); return fail_load_channel_recommendations_queries(channel_id, G()->close_status());
@ -9819,20 +9834,10 @@ void ContactsManager::on_load_channel_recommendations_from_database(ChannelId ch
return reload_channel_recommendations(channel_id); return reload_channel_recommendations(channel_id);
} }
auto it = get_channel_recommendations_queries_.find(channel_id);
CHECK(it != get_channel_recommendations_queries_.end());
auto promises = std::move(it->second);
CHECK(!promises.empty());
get_channel_recommendations_queries_.erase(it);
auto total_count = recommended_dialogs.total_count_;
auto dialog_ids = recommended_dialogs.dialog_ids_;
auto next_reload_time = recommended_dialogs.next_reload_time_; auto next_reload_time = recommended_dialogs.next_reload_time_;
for (auto &promise : promises) { finish_load_channel_recommendations_queries(channel_id, recommended_dialogs.total_count_,
if (promise) { recommended_dialogs.dialog_ids_);
promise.set_value(td_->messages_manager_->get_chats_object(total_count, dialog_ids,
"on_load_channel_recommendations_from_database"));
}
}
if (next_reload_time <= Time::now()) { if (next_reload_time <= Time::now()) {
load_channel_recommendations(channel_id, false, Auto()); load_channel_recommendations(channel_id, false, Auto());
} }
@ -9883,17 +9888,7 @@ void ContactsManager::on_get_channel_recommendations(
log_event_store(recommended_dialogs).as_slice().str(), Promise<Unit>()); log_event_store(recommended_dialogs).as_slice().str(), Promise<Unit>());
} }
auto it = get_channel_recommendations_queries_.find(channel_id); finish_load_channel_recommendations_queries(channel_id, total_count, std::move(dialog_ids));
CHECK(it != get_channel_recommendations_queries_.end());
auto promises = std::move(it->second);
CHECK(!promises.empty());
get_channel_recommendations_queries_.erase(it);
for (auto &promise : promises) {
if (promise) {
promise.set_value(
td_->messages_manager_->get_chats_object(total_count, dialog_ids, "on_get_channel_recommendations"));
}
}
} }
void ContactsManager::return_created_public_dialogs(Promise<td_api::object_ptr<td_api::chats>> &&promise, void ContactsManager::return_created_public_dialogs(Promise<td_api::object_ptr<td_api::chats>> &&promise,

3
td/telegram/ContactsManager.h
View File

@ -1750,6 +1750,9 @@ class ContactsManager final : public Actor {
void fail_load_channel_recommendations_queries(ChannelId channel_id, Status &&error); void fail_load_channel_recommendations_queries(ChannelId channel_id, Status &&error);
void finish_load_channel_recommendations_queries(ChannelId channel_id, int32 total_count,
vector<DialogId> dialog_ids);
void on_load_channel_recommendations_from_database(ChannelId channel_id, string value); void on_load_channel_recommendations_from_database(ChannelId channel_id, string value);
void reload_channel_recommendations(ChannelId channel_id); void reload_channel_recommendations(ChannelId channel_id);