andreacavalli/tdlib-fork
Archived
1
0
Fork 0
Code Pull Requests Releases Activity

Speculatively change cached supergroup member list.

Browse Source 
GitOrigin-RevId: 7d908f39f10b8480310960c54ec2610febd0b018
This commit is contained in:
levlam 2019-03-12 17:52:10 +03:00
parent a893653ba6
commit 9ff02e7435
3 changed files with 76 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

78
td/telegram/ContactsManager.cpp
View File

@ -7490,9 +7490,11 @@ void ContactsManager::on_get_channel_participants_success(
bot_user_ids = std::move(user_ids);
}
if (get_channel_type(channel_id) == ChannelType::Megagroup) {
cached_channel_participants_[channel_id] = result;
update_channel_online_member_count(channel_id, true);
}
}
if (filter.is_administrators() || filter.is_recent()) {
on_update_dialog_administrators(DialogId(channel_id), std::move(administrator_user_ids), true);
}
@ -7545,14 +7547,35 @@ bool ContactsManager::speculative_add_count(int32 &count, int32 new_count) {
}
void ContactsManager::speculative_add_channel_participants(ChannelId channel_id, const vector<UserId> &added_user_ids,
bool by_me) {
UserId inviter_user_id, int32 date, bool by_me) {
auto it = cached_channel_participants_.find(channel_id);
bool is_participants_cache_changed = false;
int32 new_participant_count = 0;
for (auto &user_id : added_user_ids) {
for (auto user_id : added_user_ids) {
if (!user_id.is_valid()) {
continue;
}
new_participant_count++;
if (it != cached_channel_participants_.end()) {
auto &participants = it->second;
bool is_found = false;
for (auto &participant : participants) {
if (participant.user_id == user_id) {
is_found = true;
break;
}
}
if (!is_found) {
is_participants_cache_changed = true;
participants.emplace_back(user_id, inviter_user_id, date, DialogParticipantStatus::Member());
}
}
}
if (is_participants_cache_changed) {
update_channel_online_member_count(channel_id, false);
}
if (new_participant_count == 0) {
return;
@ -7566,6 +7589,18 @@ void ContactsManager::speculative_delete_channel_participant(ChannelId channel_i
return;
}
auto it = cached_channel_participants_.find(channel_id);
if (it != cached_channel_participants_.end()) {
auto &participants = it->second;
for (size_t i = 0; i < participants.size(); i++) {
if (participants[i].user_id == deleted_user_id) {
participants.erase(participants.begin() + i);
update_channel_online_member_count(channel_id, false);
break;
}
}
}
speculative_add_channel_participants(channel_id, -1, by_me);
}
@ -7593,16 +7628,17 @@ void ContactsManager::speculative_add_channel_participants(ChannelId channel_id,
update_channel_full(channel_full, channel_id);
}
void ContactsManager::speculative_add_channel_user(ChannelId channel_id, UserId user_id, DialogParticipantStatus status,
void ContactsManager::speculative_add_channel_user(ChannelId channel_id, UserId user_id,
DialogParticipantStatus new_status,
DialogParticipantStatus old_status) {
auto c = get_channel(channel_id);
if (c != nullptr && c->participant_count != 0 &&
speculative_add_count(c->participant_count, status.is_member() - old_status.is_member())) {
speculative_add_count(c->participant_count, new_status.is_member() - old_status.is_member())) {
c->need_send_update = true;
update_channel(c, channel_id);
}
if (status.is_administrator() != old_status.is_administrator()) {
if (new_status.is_administrator() != old_status.is_administrator()) {
DialogId dialog_id(channel_id);
auto administrators_it = dialog_administrators_.find(dialog_id);
if (administrators_it != dialog_administrators_.end()) {
@ -7610,7 +7646,7 @@ void ContactsManager::speculative_add_channel_user(ChannelId channel_id, UserId
auto it = std::find(user_ids.begin(), user_ids.end(), user_id);
bool is_found = it != user_ids.end();
if (status.is_administrator() != is_found) {
if (new_status.is_administrator() != is_found) {
if (is_found) {
user_ids.erase(it);
} else {
@ -7621,19 +7657,41 @@ void ContactsManager::speculative_add_channel_user(ChannelId channel_id, UserId
}
}
auto it = cached_channel_participants_.find(channel_id);
if (it != cached_channel_participants_.end()) {
auto &participants = it->second;
bool is_found = false;
for (size_t i = 0; i < participants.size(); i++) {
if (participants[i].user_id == user_id) {
if (!new_status.is_member()) {
participants.erase(participants.begin() + i);
update_channel_online_member_count(channel_id, false);
} else {
participants[i].status = new_status;
}
is_found = true;
break;
}
}
if (!is_found && new_status.is_member()) {
participants.emplace_back(user_id, get_my_id(), G()->unix_time(), new_status);
update_channel_online_member_count(channel_id, false);
}
}
auto channel_full = get_channel_full(channel_id);
if (channel_full == nullptr) {
return;
}
channel_full->is_changed |=
speculative_add_count(channel_full->participant_count, status.is_member() - old_status.is_member());
speculative_add_count(channel_full->participant_count, new_status.is_member() - old_status.is_member());
channel_full->is_changed |= speculative_add_count(channel_full->administrator_count,
status.is_administrator() - old_status.is_administrator());
new_status.is_administrator() - old_status.is_administrator());
channel_full->is_changed |=
speculative_add_count(channel_full->restricted_count, status.is_restricted() - old_status.is_restricted());
speculative_add_count(channel_full->restricted_count, new_status.is_restricted() - old_status.is_restricted());
channel_full->is_changed |=
speculative_add_count(channel_full->banned_count, status.is_banned() - old_status.is_banned());
speculative_add_count(channel_full->banned_count, new_status.is_banned() - old_status.is_banned());
update_channel_full(channel_full, channel_id);
}

5
td/telegram/ContactsManager.h
View File

@ -172,7 +172,8 @@ class ContactsManager : public Actor {
void on_update_dialog_administrators(DialogId dialog_id, vector<UserId> administrator_user_ids, bool have_access);
void speculative_add_channel_participants(ChannelId channel_id, const vector<UserId> &added_user_ids, bool by_me);
void speculative_add_channel_participants(ChannelId channel_id, const vector<UserId> &added_user_ids,
UserId inviter_user_id, int32 date, bool by_me);
void speculative_delete_channel_participant(ChannelId channel_id, UserId deleted_user_id, bool by_me);
@ -895,7 +896,7 @@ class ContactsManager : public Actor {
void speculative_add_channel_participants(ChannelId channel_id, int32 new_participant_count, bool by_me);
void speculative_add_channel_user(ChannelId channel_id, UserId user_id, DialogParticipantStatus status,
void speculative_add_channel_user(ChannelId channel_id, UserId user_id, DialogParticipantStatus new_status,
DialogParticipantStatus old_status);
void invalidate_chat_full(ChatId chat_id);

5
td/telegram/MessagesManager.cpp
View File

@ -22108,11 +22108,12 @@ MessagesManager::Message *MessagesManager::add_message_to_dialog(Dialog *d, uniq
switch (message_content_type) {
case MessageContentType::ChatAddUsers:
td_->contacts_manager_->speculative_add_channel_participants(
dialog_id.get_channel_id(), get_message_content_added_user_ids(m->content.get()),
m->sender_user_id == my_user_id);
dialog_id.get_channel_id(), get_message_content_added_user_ids(m->content.get()), m->sender_user_id,
m->date, m->sender_user_id == my_user_id);
break;
case MessageContentType::ChatJoinedByLink:
td_->contacts_manager_->speculative_add_channel_participants(dialog_id.get_channel_id(), {m->sender_user_id},
m->sender_user_id, m->date,
m->sender_user_id == my_user_id);
break;
case MessageContentType::ChatDeleteUser: