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

28
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) {
return promise.set_value(td_api::make_object<td_api::chats>());
}
auto it = channel_recommended_dialog_ids_.find(channel_id);
if (it != channel_recommended_dialog_ids_.end()) {
auto it = channel_recommended_dialogs_.find(channel_id);
if (it != channel_recommended_dialogs_.end()) {
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)) {
is_valid = false;
break;
}
}
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_dialog_ids_.erase(it);
channel_recommended_dialogs_.erase(it);
}
}
reload_channel_recommendations(channel_id, std::move(promise));
}
@ -9663,19 +9668,24 @@ void ContactsManager::on_get_channel_recommendations(ChannelId channel_id,
auto channel_ids = get_channel_ids(r_chats.move_as_ok(), "on_get_channel_recommendations");
vector<DialogId> dialog_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)) {
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);
for (auto &promise : promises) {
if (promise) {
promise.set_value(td_->messages_manager_->get_chats_object(-1, dialog_ids, "on_get_channel_recommendations"));
}
}
}
void ContactsManager::return_created_public_dialogs(Promise<td_api::object_ptr<td_api::chats>> &&promise,
const vector<ChannelId> &channel_ids) {

7
td/telegram/ContactsManager.h
View File

@ -1218,6 +1218,7 @@ class ContactsManager final : public Actor {
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 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
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<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>
get_channel_recommendations_queries_;