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

Reload recommended chats from time to time.

Browse Source
This commit is contained in:
levlam 2023-11-20 23:12:42 +03:00
parent b6bb02baad
commit 21222604d0
2 changed files with 29 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
td/telegram/ContactsManager.cpp
View File

@ -9614,21 +9614,26 @@ void ContactsManager::get_channel_recommendations(DialogId dialog_id,
if (!is_broadcast_channel(channel_id) || get_input_channel(channel_id) == nullptr) { if (!is_broadcast_channel(channel_id) || get_input_channel(channel_id) == nullptr) {
return promise.set_value(td_api::make_object<td_api::chats>()); return promise.set_value(td_api::make_object<td_api::chats>());
} }
auto it = channel_recommended_dialog_ids_.find(channel_id); auto it = channel_recommended_dialogs_.find(channel_id);
if (it != channel_recommended_dialog_ids_.end()) { if (it != channel_recommended_dialogs_.end()) {
bool is_valid = true; bool is_valid = true;
for (auto recommended_dialog_id : it->second) { for (auto recommended_dialog_id : it->second.dialog_ids_) {
if (!is_suitable_recommended_channel(recommended_dialog_id)) { if (!is_suitable_recommended_channel(recommended_dialog_id)) {
is_valid = false; is_valid = false;
break; break;
} }
} }
if (is_valid) { if (is_valid) {
return promise.set_value(td_->messages_manager_->get_chats_object(-1, it->second, "get_channel_recommendations")); promise.set_value(
td_->messages_manager_->get_chats_object(-1, it->second.dialog_ids_, "get_channel_recommendations"));
if (it->second.next_reload_time_ > Time::now()) {
return;
}
promise = {};
} else {
LOG(INFO) << "Drop cache for similar chats of " << dialog_id;
channel_recommended_dialogs_.erase(it);
} }
LOG(INFO) << "Drop cache for similar chats of " << dialog_id;
channel_recommended_dialog_ids_.erase(it);
} }
reload_channel_recommendations(channel_id, std::move(promise)); reload_channel_recommendations(channel_id, std::move(promise));
} }
@ -9663,17 +9668,22 @@ void ContactsManager::on_get_channel_recommendations(ChannelId channel_id,
auto channel_ids = get_channel_ids(r_chats.move_as_ok(), "on_get_channel_recommendations"); auto channel_ids = get_channel_ids(r_chats.move_as_ok(), "on_get_channel_recommendations");
vector<DialogId> dialog_ids; vector<DialogId> dialog_ids;
for (auto recommended_channel_id : channel_ids) { for (auto recommended_channel_id : channel_ids) {
td_->messages_manager_->force_create_dialog(DialogId(recommended_channel_id), "on_get_channel_recommendations"); auto recommended_dialog_id = DialogId(recommended_channel_id);
td_->messages_manager_->force_create_dialog(recommended_dialog_id, "on_get_channel_recommendations");
if (is_suitable_recommended_channel(recommended_channel_id)) { if (is_suitable_recommended_channel(recommended_channel_id)) {
dialog_ids.push_back(DialogId(recommended_channel_id)); dialog_ids.push_back(recommended_dialog_id);
} }
} }
channel_recommended_dialog_ids_[channel_id] = dialog_ids; auto &recommended_dialogs = channel_recommended_dialogs_[channel_id];
recommended_dialogs.dialog_ids_ = dialog_ids;
recommended_dialogs.next_reload_time_ = Time::now() + CHANNEL_RECOMMENDATIONS_CACHE_TIME;
// save_channel_recommendations(channel_id); // save_channel_recommendations(channel_id);
for (auto &promise : promises) { for (auto &promise : promises) {
promise.set_value(td_->messages_manager_->get_chats_object(-1, dialog_ids, "on_get_channel_recommendations")); if (promise) {
promise.set_value(td_->messages_manager_->get_chats_object(-1, dialog_ids, "on_get_channel_recommendations"));
}
} }
} }

11
td/telegram/ContactsManager.h
View File

@ -1216,8 +1216,9 @@ class ContactsManager final : public Actor {
static constexpr size_t MAX_INVITE_LINK_TITLE_LENGTH = 32; // server side limit static constexpr size_t MAX_INVITE_LINK_TITLE_LENGTH = 32; // server side limit
static constexpr int32 MAX_GET_CHANNEL_PARTICIPANTS = 200; // server side limit static constexpr int32 MAX_GET_CHANNEL_PARTICIPANTS = 200; // server side limit
static constexpr int32 CHANNEL_PARTICIPANT_CACHE_TIME = 1800; // some reasonable limit static constexpr int32 CHANNEL_PARTICIPANT_CACHE_TIME = 1800; // some reasonable limit
static constexpr int32 MAX_ACTIVE_STORY_ID_RELOAD_TIME = 3600; // some reasonable limit static constexpr int32 MAX_ACTIVE_STORY_ID_RELOAD_TIME = 3600; // some reasonable limit
static constexpr int32 CHANNEL_RECOMMENDATIONS_CACHE_TIME = 86400; // some reasonable limit
// the True fields aren't set for manually created telegram_api::user objects, therefore the flags must be used // the True fields aren't set for manually created telegram_api::user objects, therefore the flags must be used
static constexpr int32 USER_FLAG_HAS_ACCESS_HASH = 1 << 0; static constexpr int32 USER_FLAG_HAS_ACCESS_HASH = 1 << 0;
@ -1999,7 +2000,11 @@ class ContactsManager final : public Actor {
FlatHashMap<string, unique_ptr<InviteLinkInfo>> invite_link_infos_; FlatHashMap<string, unique_ptr<InviteLinkInfo>> invite_link_infos_;
FlatHashMap<DialogId, DialogAccessByInviteLink, DialogIdHash> dialog_access_by_invite_link_; FlatHashMap<DialogId, DialogAccessByInviteLink, DialogIdHash> dialog_access_by_invite_link_;
FlatHashMap<ChannelId, vector<DialogId>, ChannelIdHash> channel_recommended_dialog_ids_; struct RecommendedDialogs {
vector<DialogId> dialog_ids_;
double next_reload_time_ = 0.0;
};
FlatHashMap<ChannelId, RecommendedDialogs, ChannelIdHash> channel_recommended_dialogs_;
FlatHashMap<ChannelId, vector<Promise<td_api::object_ptr<td_api::chats>>>, ChannelIdHash> FlatHashMap<ChannelId, vector<Promise<td_api::object_ptr<td_api::chats>>>, ChannelIdHash>
get_channel_recommendations_queries_; get_channel_recommendations_queries_;