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

Send updateChatOnlineMember count when a member goes online or offline.

Browse Source 
GitOrigin-RevId: e58d65fb71ce4b2a407101e2a494da05106b0b9f
This commit is contained in:
levlam 2019-03-05 15:49:10 +03:00
parent 986ce38998
commit 0a04a466ef
3 changed files with 48 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
td/telegram/ContactsManager.cpp
View File

@ -2402,6 +2402,10 @@ UserId ContactsManager::load_my_id() {
}
void ContactsManager::on_user_online_timeout_callback(void *contacts_manager_ptr, int64 user_id_long) {
if (G()->close_flag()) {
return;
}
auto contacts_manager = static_cast<ContactsManager *>(contacts_manager_ptr);
UserId user_id(narrow_cast<int32>(user_id_long));
auto u = contacts_manager->get_user(user_id);
@ -2411,6 +2415,8 @@ void ContactsManager::on_user_online_timeout_callback(void *contacts_manager_ptr
send_closure_later(
G()->td(), &Td::send_update,
make_tl_object<td_api::updateUserStatus>(user_id.get(), contacts_manager->get_user_status_object(user_id, u)));
contacts_manager->update_user_online_member_count(u);
}
void ContactsManager::on_channel_unban_timeout_callback(void *contacts_manager_ptr, int64 channel_id_long) {
@ -6215,6 +6221,10 @@ void ContactsManager::update_user(User *u, UserId user_id, bool from_binlog, boo
make_tl_object<td_api::updateUserStatus>(user_id.get(), get_user_status_object(user_id, u)));
u->is_status_changed = false;
}
if (u->is_online_status_changed) {
update_user_online_member_count(u);
u->is_online_status_changed = false;
}
if (!from_database) {
save_user(u, user_id, from_binlog);
@ -6975,6 +6985,8 @@ void ContactsManager::on_update_user_online(User *u, UserId user_id, tl_object_p
if (new_online != u->was_online) {
LOG(DEBUG) << "Update " << user_id << " online from " << u->was_online << " to " << new_online;
bool was_online = u->was_online > G()->unix_time_cached();
bool is_online = new_online > G()->unix_time_cached();
u->was_online = new_online;
u->is_status_changed = true;
@ -6983,6 +6995,8 @@ void ContactsManager::on_update_user_online(User *u, UserId user_id, tl_object_p
if (is_offline) {
td_->on_online_updated(false, false);
}
} else if (was_online != is_online) {
u->is_online_status_changed = true;
}
}
}
@ -7121,8 +7135,31 @@ void ContactsManager::invalidate_user_full(UserId user_id) {
update_user_full(user_full, user_id);
}
void ContactsManager::update_chat_online_member_count(const ChatFull *chat_full, ChatId chat_id,
bool is_from_server) const {
void ContactsManager::update_user_online_member_count(User *u) {
if (u->online_member_chats.empty()) {
return;
}
auto now = G()->unix_time_cached();
vector<ChatId> expired_chat_ids;
for (auto &it : u->online_member_chats) {
auto chat_id = it.first;
auto time = it.second;
if (time < now - MessagesManager::ONLINE_MEMBER_COUNT_CACHE_EXPIRE_TIME) {
expired_chat_ids.push_back(chat_id);
continue;
}
auto chat_full = get_chat_full(chat_id);
CHECK(chat_full != nullptr);
update_chat_online_member_count(chat_full, chat_id, false);
}
for (auto &chat_id : expired_chat_ids) {
u->online_member_chats.erase(chat_id);
}
}
void ContactsManager::update_chat_online_member_count(const ChatFull *chat_full, ChatId chat_id, bool is_from_server) {
if (td_->auth_manager_->is_bot()) {
return;
}
@ -7139,6 +7176,7 @@ void ContactsManager::update_chat_online_member_count(const ChatFull *chat_full,
if (was_online > time) {
online_member_count++;
}
u->online_member_chats[chat_id] = time;
}
}
td_->messages_manager_->on_update_dialog_online_member_count(DialogId(chat_id), online_member_count, is_from_server);

7
td/telegram/ContactsManager.h
View File

@ -471,6 +471,8 @@ class ContactsManager : public Actor {
std::unordered_set<int64> photo_ids;
std::unordered_map<ChatId, int32, ChatIdHash> online_member_chats; // id -> time
bool is_received = false;
bool is_verified = false;
bool is_support = false;
@ -490,6 +492,7 @@ class ContactsManager : public Actor {
bool is_changed = true; // have new changes not sent to the database except changes visible to the client
bool need_send_update = true; // have new changes not sent to the client
bool is_status_changed = true;
bool is_online_status_changed = true; // whether online/offline has changed
bool is_saved = false; // is current user version being saved/is saved to the database
bool is_being_saved = false; // is current user being saved to the database
@ -891,9 +894,11 @@ class ContactsManager : public Actor {
void speculative_add_channel_user(ChannelId channel_id, UserId user_id, DialogParticipantStatus status,
DialogParticipantStatus old_status);
void update_chat_online_member_count(const ChatFull *chat_full, ChatId chat_id, bool is_from_server) const;
void invalidate_chat_full(ChatId chat_id);
void update_user_online_member_count(User *u);
void update_chat_online_member_count(const ChatFull *chat_full, ChatId chat_id, bool is_from_server);
void on_chat_update(telegram_api::chatEmpty &chat, const char *source);
void on_chat_update(telegram_api::chat &chat, const char *source);
void on_chat_update(telegram_api::chatForbidden &chat, const char *source);

3
td/telegram/MessagesManager.h
View File

@ -159,6 +159,8 @@ class MessagesManager : public Actor {
static constexpr int32 SEND_MESSAGE_FLAG_GROUP_MEDIA = 1 << 9;
static constexpr int32 SEND_MESSAGE_FLAG_HAS_MESSAGE = 1 << 11;
static constexpr int32 ONLINE_MEMBER_COUNT_CACHE_EXPIRE_TIME = 30 * 60;
MessagesManager(Td *td, ActorShared<> parent);
MessagesManager(const MessagesManager &) = delete;
MessagesManager &operator=(const MessagesManager &) = delete;
@ -1242,7 +1244,6 @@ class MessagesManager : public Actor {
static constexpr int32 USERNAME_CACHE_EXPIRE_TIME = 3 * 86400;
static constexpr int32 USERNAME_CACHE_EXPIRE_TIME_SHORT = 900;
static constexpr int32 ONLINE_MEMBER_COUNT_CACHE_EXPIRE_TIME = 30 * 60;
static constexpr int32 ONLINE_MEMBER_COUNT_UPDATE_TIME = 5 * 60;
static constexpr int32 MAX_RESEND_DELAY = 86400; // seconds, some resonable limit