Load active story lists from database.

td/telegram/StoryManager.cpp

@@ -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() && stories->has_more_) {
-      if (stories->user_stories_.empty()) {
-        if (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;
     }

td/telegram/StoryManager.h

@@ -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);