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

Immediately forget local polls after sending.

Browse Source
This commit is contained in:
levlam 2022-06-20 16:53:28 +03:00
parent 71573fd320
commit 49bd383b35
2 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
td/telegram/PollManager.cpp
View File

@ -674,6 +674,10 @@ void PollManager::unregister_poll(PollId poll_id, FullMessageId full_message_id,
auto &message_ids = other_poll_messages_[poll_id];
auto is_deleted = message_ids.erase(full_message_id) > 0;
LOG_CHECK(is_deleted) << source << ' ' << poll_id << ' ' << full_message_id;
if (is_local_poll_id(poll_id)) {
CHECK(message_ids.empty());
forget_local_poll(poll_id);
}
if (message_ids.empty()) {
other_poll_messages_.erase(poll_id);
@ -685,6 +689,10 @@ void PollManager::unregister_poll(PollId poll_id, FullMessageId full_message_id,
auto &message_ids = server_poll_messages_[poll_id];
auto is_deleted = message_ids.erase(full_message_id) > 0;
LOG_CHECK(is_deleted) << source << ' ' << poll_id << ' ' << full_message_id;
if (is_local_poll_id(poll_id)) {
CHECK(message_ids.empty());
forget_local_poll(poll_id);
}
if (message_ids.empty()) {
server_poll_messages_.erase(poll_id);
update_poll_timeout_.cancel_timeout(poll_id.get());
@ -1308,6 +1316,18 @@ void PollManager::on_unload_poll_timeout(PollId poll_id) {
unload_poll_timeout_.cancel_timeout(poll_id.get());
}
void PollManager::forget_local_poll(PollId poll_id) {
CHECK(is_local_poll_id(poll_id));
LOG(INFO) << "Forget " << poll_id;
auto is_deleted = polls_.erase(poll_id) > 0;
CHECK(is_deleted);
CHECK(poll_voters_.count(poll_id) == 0);
CHECK(loaded_from_database_polls_.count(poll_id) == 0);
}
void PollManager::on_get_poll_results(PollId poll_id, uint64 generation,
Result<tl_object_ptr<telegram_api::Updates>> result) {
auto poll = get_poll(poll_id);

2
td/telegram/PollManager.h
View File

@ -213,6 +213,8 @@ class PollManager final : public Actor {
void do_stop_poll(PollId poll_id, FullMessageId full_message_id, unique_ptr<ReplyMarkup> &&reply_markup,
uint64 log_event_id, Promise<Unit> &&promise);
void forget_local_poll(PollId poll_id);
MultiTimeout update_poll_timeout_{"UpdatePollTimeout"};
MultiTimeout close_poll_timeout_{"ClosePollTimeout"};
MultiTimeout unload_poll_timeout_{"UnloadPollTimeout"};