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

Load active story lists from database.

Browse Source
This commit is contained in:
levlam 2023-07-17 15:50:27 +03:00
parent 308918f737
commit 6c39eabb09
2 changed files with 132 additions and 49 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

104
td/telegram/StoryManager.cpp
View File

@ -24,7 +24,6 @@
#include "td/telegram/ReportReason.h"
#include "td/telegram/StoryContent.h"
#include "td/telegram/StoryContentType.h"
#include "td/telegram/StoryDb.h"
#include "td/telegram/StoryInteractionInfo.hpp"
#include "td/telegram/Td.h"
#include "td/telegram/TdDb.h"
@ -931,6 +930,7 @@ StoryManager::StoryManager(Td *td, ActorShared<> parent) : td_(td), parent_(std:
story_list.state_ = std::move(saved_story_list.state_);
story_list.server_total_count_ = td::max(saved_story_list.total_count_, 0);
story_list.server_has_more_ = saved_story_list.has_more_;
story_list.database_has_more_ = true;
}
}
}
@ -1323,9 +1323,93 @@ void StoryManager::load_active_stories(StoryListId story_list_id, Promise<Unit>
return promise.set_error(Status::Error(404, "Not found"));
}
if (story_list.database_has_more_) {
CHECK(G()->use_message_database());
story_list.load_list_from_database_queries_.push_back(std::move(promise));
if (story_list.load_list_from_database_queries_.size() == 1u) {
G()->td_db()->get_story_db_async()->get_active_story_list(
story_list_id, story_list.last_loaded_database_dialog_date_.get_order(),
story_list.last_loaded_database_dialog_date_.get_dialog_id(), 10,
PromiseCreator::lambda(
[actor_id = actor_id(this), story_list_id](Result<StoryDbGetActiveStoryListResult> &&result) {
send_closure(actor_id, &StoryManager::on_load_active_stories_from_database, story_list_id,
std::move(result));
}));
}
return;
}
if (!story_list.server_has_more_) {
auto min_story_date = story_list.list_last_story_date_;
story_list.list_last_story_date_ = MAX_DIALOG_DATE;
for (auto it = story_list.ordered_stories_.upper_bound(min_story_date); it != story_list.ordered_stories_.end();
++it) {
on_dialog_active_stories_order_updated(it->get_dialog_id(), "load_active_stories");
}
return promise.set_error(Status::Error(404, "Not found"));
}
load_active_stories_from_server(story_list_id, story_list, !story_list.state_.empty(), std::move(promise));
}
void StoryManager::on_load_active_stories_from_database(StoryListId story_list_id,
Result<StoryDbGetActiveStoryListResult> result) {
G()->ignore_result_if_closing(result);
auto &story_list = get_story_list(story_list_id);
auto promises = std::move(story_list.load_list_from_database_queries_);
CHECK(!promises.empty());
if (result.is_error()) {
return fail_promises(promises, result.move_as_error());
}
auto active_story_list = result.move_as_ok();
LOG(INFO) << "Load " << active_story_list.active_stories_.size() << " chats with active stories from database";
Dependencies dependencies;
for (auto &active_stories_it : active_story_list.active_stories_) {
dependencies.add_dialog_and_dependencies(active_stories_it.first);
}
if (!dependencies.resolve_force(td_, "on_load_active_stories_from_database")) {
active_story_list.active_stories_.clear();
story_list.state_.clear();
story_list.server_has_more_ = true;
}
if (active_story_list.active_stories_.empty()) {
story_list.last_loaded_database_dialog_date_ = MAX_DIALOG_DATE;
story_list.database_has_more_ = false;
} else {
for (auto &active_stories_it : active_story_list.active_stories_) {
on_get_active_stories_from_database(active_stories_it.first, active_stories_it.second,
"on_load_active_stories_from_database");
}
DialogDate max_story_date(active_story_list.next_order_, active_story_list.next_dialog_id_);
if (story_list.last_loaded_database_dialog_date_ < max_story_date) {
story_list.last_loaded_database_dialog_date_ = max_story_date;
auto min_story_date = story_list.list_last_story_date_;
story_list.list_last_story_date_ = max_story_date;
auto &owner_dialog_ids = dependencies.get_dialog_ids();
for (auto it = story_list.ordered_stories_.upper_bound(min_story_date);
it != story_list.ordered_stories_.end() && *it <= max_story_date; ++it) {
auto dialog_id = it->get_dialog_id();
if (owner_dialog_ids.count(dialog_id) == 0) {
on_dialog_active_stories_order_updated(dialog_id, "on_load_active_stories_from_database 1");
}
}
for (auto owner_dialog_id : owner_dialog_ids) {
on_dialog_active_stories_order_updated(owner_dialog_id, "on_load_active_stories_from_database 2");
}
} else {
LOG(ERROR) << "Last database story date didn't increase";
}
update_story_list_sent_total_count(story_list_id, story_list);
}
set_promises(promises);
}
void StoryManager::load_active_stories_from_server(StoryListId story_list_id, StoryList &story_list, bool is_next,
Promise<Unit> &&promise) {
story_list.load_list_from_server_queries_.push_back(std::move(promise));
@ -1392,19 +1476,11 @@ void StoryManager::on_load_active_stories_from_server(
}));
auto lock = mpas.get_promise();
vector<DialogId> delete_dialog_ids;
if (stories->user_stories_.empty()) {
if (stories->has_more_) {
if (stories->user_stories_.empty() && stories->has_more_) {
LOG(ERROR) << "Receive no stories, but expected more";
stories->has_more_ = false;
}
if (!is_next) {
// there should be no more active stories; reload all of them
for (const auto &story_date : story_list.ordered_stories_) {
delete_dialog_ids.push_back(story_date.get_dialog_id());
}
}
story_list.list_last_story_date_ = MAX_DIALOG_DATE;
} else {
auto max_story_date = MIN_DIALOG_DATE;
vector<DialogId> owner_dialog_ids;
for (auto &user_stories : stories->user_stories_) {
@ -1427,6 +1503,8 @@ void StoryManager::on_load_active_stories_from_server(
if (!stories->has_more_) {
max_story_date = MAX_DIALOG_DATE;
}
vector<DialogId> delete_dialog_ids;
auto min_story_date = is_next ? story_list.list_last_story_date_ : MIN_DIALOG_DATE;
for (auto it = story_list.ordered_stories_.upper_bound(min_story_date);
it != story_list.ordered_stories_.end() && *it <= max_story_date; ++it) {
@ -1443,7 +1521,6 @@ void StoryManager::on_load_active_stories_from_server(
} else if (is_next) {
LOG(ERROR) << "Last story date didn't increase";
}
}
if (!delete_dialog_ids.empty()) {
LOG(INFO) << "Delete active stories in " << delete_dialog_ids;
}
@ -1452,6 +1529,7 @@ void StoryManager::on_load_active_stories_from_server(
load_dialog_expiring_stories(dialog_id, 0, "on_load_active_stories_from_server 1");
}
update_story_list_sent_total_count(story_list_id, story_list);
lock.set_value(Unit());
break;
}

7
td/telegram/StoryManager.h
View File

@ -13,6 +13,7 @@
#include "td/telegram/FullMessageId.h"
#include "td/telegram/MessageEntity.h"
#include "td/telegram/MessageViewer.h"
#include "td/telegram/StoryDb.h"
#include "td/telegram/StoryFullId.h"
#include "td/telegram/StoryId.h"
#include "td/telegram/StoryInteractionInfo.h"
@ -158,12 +159,14 @@ class StoryManager final : public Actor {
string state_;
bool server_has_more_ = true;
bool database_has_more_ = true;
bool database_has_more_ = false;
vector<Promise<Unit>> load_list_from_server_queries_;
vector<Promise<Unit>> load_list_from_database_queries_;
std::set<DialogDate> ordered_stories_; // all known active stories from the story list
DialogDate last_loaded_database_dialog_date_ = MIN_DIALOG_DATE; // in memory
DialogDate list_last_story_date_ = MIN_DIALOG_DATE; // in memory
};
@ -404,6 +407,8 @@ class StoryManager final : public Actor {
void on_load_dialog_expiring_stories(DialogId owner_dialog_id);
void on_load_active_stories_from_database(StoryListId story_list_id, Result<StoryDbGetActiveStoryListResult> result);
void load_active_stories_from_server(StoryListId story_list_id, StoryList &story_list, bool is_next,
Promise<Unit> &&promise);