Send update after slow mode delay expires.

GitOrigin-RevId: 90b18c2a75f1e39ac967ebc3bbce84145769b2d8
This commit is contained in:
levlam 2019-11-21 00:40:34 +03:00
parent fb90abcccc
commit 2415d28b4e
2 changed files with 57 additions and 3 deletions

View File

@ -2645,6 +2645,9 @@ ContactsManager::ContactsManager(Td *td, ActorShared<> parent) : td_(td), parent
user_nearby_timeout_.set_callback(on_user_nearby_timeout_callback);
user_nearby_timeout_.set_callback_data(static_cast<void *>(this));
slow_mode_delay_timeout_.set_callback(on_slow_mode_delay_timeout_callback);
slow_mode_delay_timeout_.set_callback_data(static_cast<void *>(this));
}
void ContactsManager::tear_down() {
@ -2748,6 +2751,24 @@ void ContactsManager::on_user_nearby_timeout(UserId user_id) {
}
}
void ContactsManager::on_slow_mode_delay_timeout_callback(void *contacts_manager_ptr, int64 channel_id_long) {
if (G()->close_flag()) {
return;
}
auto contacts_manager = static_cast<ContactsManager *>(contacts_manager_ptr);
send_closure_later(contacts_manager->actor_id(contacts_manager), &ContactsManager::on_slow_mode_delay_timeout,
ChannelId(narrow_cast<int32>(channel_id_long)));
}
void ContactsManager::on_slow_mode_delay_timeout(ChannelId channel_id) {
if (G()->close_flag()) {
return;
}
on_update_channel_slow_mode_next_send_date(channel_id, 0);
}
template <class StorerT>
void ContactsManager::BotInfo::store(StorerT &storer) const {
using td::store;
@ -8020,6 +8041,22 @@ void ContactsManager::update_channel_full(ChannelFull *channel_full, ChannelId c
channel_full->administrator_count = channel_full->participant_count;
}
if (channel_full->is_slow_mode_next_send_date_changed) {
auto now = G()->server_time();
if (channel_full->slow_mode_next_send_date > now + 3601) {
channel_full->slow_mode_next_send_date = static_cast<int32>(now) + 3601;
}
if (channel_full->slow_mode_next_send_date <= now) {
channel_full->slow_mode_next_send_date = 0;
}
if (channel_full->slow_mode_next_send_date == 0) {
slow_mode_delay_timeout_.cancel_timeout(channel_id.get());
} else {
slow_mode_delay_timeout_.set_timeout_in(channel_id.get(), channel_full->slow_mode_next_send_date - now + 0.002);
}
channel_full->is_slow_mode_next_send_date_changed = false;
}
channel_full->need_send_update |= channel_full->is_changed;
channel_full->need_save_to_database |= channel_full->is_changed;
channel_full->is_changed = false;
@ -9577,6 +9614,7 @@ void ContactsManager::invalidate_channel_full(ChannelId channel_id, bool drop_in
if (drop_slow_mode_delay && channel_full->slow_mode_delay != 0) {
channel_full->slow_mode_delay = 0;
channel_full->slow_mode_next_send_date = 0;
channel_full->is_slow_mode_next_send_date_changed = true;
channel_full->is_changed = true;
}
update_channel_full(channel_full, channel_id);
@ -9733,6 +9771,7 @@ void ContactsManager::on_update_channel_full_slow_mode_next_send_date(ChannelFul
}
if (channel_full->slow_mode_next_send_date != slow_mode_next_send_date) {
channel_full->slow_mode_next_send_date = slow_mode_next_send_date;
channel_full->is_slow_mode_next_send_date_changed = true;
channel_full->is_changed = true;
}
}
@ -10527,6 +10566,14 @@ void ContactsManager::on_update_channel_slow_mode_delay(ChannelId channel_id, in
}
}
void ContactsManager::on_update_channel_slow_mode_next_send_date(ChannelId channel_id, int32 next_send_date) {
auto channel_full = get_channel_full_force(channel_id);
if (channel_full != nullptr) {
on_update_channel_full_slow_mode_next_send_date(channel_full, next_send_date);
update_channel_full(channel_full, channel_id);
}
}
void ContactsManager::on_update_channel_bot_user_ids(ChannelId channel_id, vector<UserId> &&bot_user_ids) {
CHECK(channel_id.is_valid());
if (!have_channel(channel_id)) {

View File

@ -186,6 +186,7 @@ class ContactsManager : public Actor {
void on_update_channel_linked_channel_id(ChannelId channel_id, ChannelId group_channel_id);
void on_update_channel_location(ChannelId channel_id, const DialogLocation &location);
void on_update_channel_slow_mode_delay(ChannelId channel_id, int32 slow_mode_delay);
void on_update_channel_slow_mode_next_send_date(ChannelId channel_id, int32 next_send_date);
void on_update_channel_is_all_history_available(ChannelId channel_id, bool is_all_history_available);
void on_update_channel_default_permissions(ChannelId channel_id, RestrictedRights default_permissions);
void on_update_channel_administrator_count(ChannelId channel_id, int32 administrator_count);
@ -246,8 +247,6 @@ class ContactsManager : public Actor {
void on_channel_unban_timeout(ChannelId channel_id);
void on_user_nearby_timeout(UserId user_id);
void check_dialog_username(DialogId dialog_id, const string &username, Promise<CheckDialogUsernameResult> &&promise);
static td_api::object_ptr<td_api::CheckChatUsernameResult> get_check_chat_username_result_object(
@ -764,6 +763,7 @@ class ContactsManager : public Actor {
bool can_view_statistics = false;
bool is_all_history_available = true;
bool is_slow_mode_next_send_date_changed = true;
bool is_changed = true; // have new changes that need to be sent to the client and database
bool need_send_update = true; // have new changes that need only to be sent to the client
bool need_save_to_database = true; // have new changes that need only to be saved to the database
@ -1286,8 +1286,14 @@ class ContactsManager : public Actor {
static void on_user_nearby_timeout_callback(void *contacts_manager_ptr, int64 user_id_long);
static void on_slow_mode_delay_timeout_callback(void *contacts_manager_ptr, int64 channel_id_long);
void on_user_online_timeout(UserId user_id);
void on_user_nearby_timeout(UserId user_id);
void on_slow_mode_delay_timeout(ChannelId channel_id);
void tear_down() override;
Td *td_;
@ -1401,7 +1407,8 @@ class ContactsManager : public Actor {
MultiTimeout user_online_timeout_{"UserOnlineTimeout"};
MultiTimeout channel_unban_timeout_{"ChannelUnbanTimeout"};
MultiTimeout user_nearby_timeout_{"UserOnlineTimeout"};
MultiTimeout user_nearby_timeout_{"UserNearbyTimeout"};
MultiTimeout slow_mode_delay_timeout_{"SlowModeDelayTimeout"};
};
} // namespace td