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

Combine reload_message_reactions queries.

Browse Source
This commit is contained in:
levlam 2022-02-16 17:09:26 +03:00
parent ddf1bdc5c4
commit 0646eb3f2e
2 changed files with 60 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
td/telegram/MessagesManager.cpp
View File

@ -6880,7 +6880,7 @@ void MessagesManager::on_get_message_reaction_list(FullMessageId full_message_id
if (it != pending_reactions_.end()) {
it->second.was_updated = true;
} else {
reload_message_reactions(td_, full_message_id.get_dialog_id(), {full_message_id.get_message_id()});
queue_message_reactions_reload(full_message_id);
}
}
@ -8574,6 +8574,53 @@ vector<string> MessagesManager::get_message_active_reactions(const Dialog *d, co
return get_dialog_active_reactions(d);
}
void MessagesManager::queue_message_reactions_reload(FullMessageId full_message_id) {
auto dialog_id = full_message_id.get_dialog_id();
CHECK(dialog_id.is_valid());
auto message_id = full_message_id.get_message_id();
CHECK(message_id.is_valid());
being_reloaded_reactions_[dialog_id].message_ids.insert(message_id);
try_reload_message_reactions(dialog_id, false);
}
void MessagesManager::try_reload_message_reactions(DialogId dialog_id, bool is_finished) {
if (G()->close_flag()) {
return;
}
auto it = being_reloaded_reactions_.find(dialog_id);
if (it == being_reloaded_reactions_.end()) {
return;
}
if (is_finished) {
CHECK(it->second.is_request_sent);
it->second.is_request_sent = false;
if (it->second.message_ids.empty()) {
being_reloaded_reactions_.erase(it);
return;
}
} else if (it->second.is_request_sent) {
return;
}
CHECK(!it->second.message_ids.empty());
CHECK(!it->second.is_request_sent);
it->second.is_request_sent = true;
static constexpr size_t MAX_MESSAGE_IDS = 100; // server-side limit
vector<MessageId> message_ids;
for (auto message_id_it = it->second.message_ids.begin();
message_id_it != it->second.message_ids.end() && message_ids.size() < MAX_MESSAGE_IDS; ++message_id_it) {
message_ids.push_back(*message_id_it);
}
for (auto message_id : message_ids) {
it->second.message_ids.erase(message_id);
}
reload_message_reactions(td_, dialog_id, std::move(message_ids));
}
bool MessagesManager::update_dialog_silent_send_message(Dialog *d, bool silent_send_message) {
if (td_->auth_manager_->is_bot()) {
// just in case
@ -23595,7 +23642,7 @@ unique_ptr<MessagesManager::Message> MessagesManager::parse_message(Dialog *d, M
}
return nullptr;
}
if (m->reactions != nullptr && d != nullptr) {
if (m->reactions != nullptr) {
if (m->available_reactions_generation < d->available_reactions_generation) {
m->reactions = nullptr;
m->available_reactions_generation = 0;
@ -24297,7 +24344,7 @@ void MessagesManager::on_set_message_reaction(FullMessageId full_message_id, Res
}
if (need_reload) {
reload_message_reactions(td_, full_message_id.get_dialog_id(), {full_message_id.get_message_id()});
queue_message_reactions_reload(full_message_id);
}
promise.set_result(std::move(result));

11
td/telegram/MessagesManager.h
View File

@ -2638,6 +2638,10 @@ class MessagesManager final : public Actor {
vector<string> get_message_active_reactions(const Dialog *d, const Message *m) const;
void queue_message_reactions_reload(FullMessageId full_message_id);
void try_reload_message_reactions(DialogId dialog_id, bool is_finished);
bool is_dialog_action_unneeded(DialogId dialog_id) const;
void on_send_dialog_action_timeout(DialogId dialog_id);
@ -3628,9 +3632,14 @@ class MessagesManager final : public Actor {
double updated_time = 0;
bool is_update_sent = false;
};
FlatHashMap<DialogId, OnlineMemberCountInfo, DialogIdHash> dialog_online_member_counts_;
struct ReactionsToReload {
std::unordered_set<MessageId, MessageIdHash> message_ids;
bool is_request_sent = false;
};
FlatHashMap<DialogId, ReactionsToReload, DialogIdHash> being_reloaded_reactions_;
FlatHashMap<DialogId, std::pair<bool, bool>, DialogIdHash> pending_dialog_group_call_updates_;
FlatHashMap<string, int32> auth_notification_id_date_;