Save autosave settings to database.
This commit is contained in:
parent
3d0140b3ef
commit
2260944afe
@ -7849,10 +7849,10 @@ setAutoDownloadSettings settings:autoDownloadSettings type:NetworkType = Ok;
|
||||
//@description Returns autosave settings for the current user
|
||||
getAutosaveSettings = AutosaveSettings;
|
||||
|
||||
//@description Sets autosave settings for the given scope @scope Autosave settings scope @settings New autosave settings for the scope; pass null to set autosave settings to default
|
||||
//@description Sets autosave settings for the given scope. The method is guaranteed to work only after at least one call to getAutosaveSettings @scope Autosave settings scope @settings New autosave settings for the scope; pass null to set autosave settings to default
|
||||
setAutosaveSettings scope:AutosaveSettingsScope settings:scopeAutosaveSettings = Ok;
|
||||
|
||||
//@description Clears the list of all autosave settings exceptions
|
||||
//@description Clears the list of all autosave settings exceptions. The method is guaranteed to work only after at least one call to getAutosaveSettings
|
||||
clearAutosaveSettingsExceptions = Ok;
|
||||
|
||||
|
||||
|
@ -7,9 +7,12 @@
|
||||
#include "td/telegram/AutosaveManager.h"
|
||||
|
||||
#include "td/telegram/ContactsManager.h"
|
||||
#include "td/telegram/Dependencies.h"
|
||||
#include "td/telegram/MessagesManager.h"
|
||||
#include "td/telegram/Td.h"
|
||||
|
||||
#include "td/db/SqliteKeyValueAsync.h"
|
||||
|
||||
#include "td/utils/algorithm.h"
|
||||
#include "td/utils/buffer.h"
|
||||
#include "td/utils/misc.h"
|
||||
@ -88,7 +91,7 @@ class SaveAutoSaveSettingsQuery final : public Td::ResultHandler {
|
||||
|
||||
void on_error(Status status) final {
|
||||
promise_.set_error(std::move(status));
|
||||
td_->autosave_manager_->reload_autosave_settings(Auto());
|
||||
td_->autosave_manager_->reload_autosave_settings();
|
||||
}
|
||||
};
|
||||
|
||||
@ -114,7 +117,7 @@ class DeleteAutoSaveExceptionsQuery final : public Td::ResultHandler {
|
||||
|
||||
void on_error(Status status) final {
|
||||
promise_.set_error(std::move(status));
|
||||
td_->autosave_manager_->reload_autosave_settings(Auto());
|
||||
td_->autosave_manager_->reload_autosave_settings();
|
||||
}
|
||||
};
|
||||
|
||||
@ -181,6 +184,26 @@ bool AutosaveManager::DialogAutosaveSettings::operator!=(const DialogAutosaveSet
|
||||
return !operator==(other);
|
||||
}
|
||||
|
||||
template <class StorerT>
|
||||
void AutosaveManager::DialogAutosaveSettings::store(StorerT &storer) const {
|
||||
CHECK(are_inited_);
|
||||
BEGIN_STORE_FLAGS();
|
||||
STORE_FLAG(autosave_photos_);
|
||||
STORE_FLAG(autosave_videos_);
|
||||
END_STORE_FLAGS();
|
||||
td::store(max_video_file_size_, storer);
|
||||
}
|
||||
|
||||
template <class ParserT>
|
||||
void AutosaveManager::DialogAutosaveSettings::parse(ParserT &parser) {
|
||||
are_inited_ = true;
|
||||
BEGIN_PARSE_FLAGS();
|
||||
PARSE_FLAG(autosave_photos_);
|
||||
PARSE_FLAG(autosave_videos_);
|
||||
END_PARSE_FLAGS();
|
||||
td::parse(max_video_file_size_, parser);
|
||||
}
|
||||
|
||||
td_api::object_ptr<td_api::autosaveSettings> AutosaveManager::AutosaveSettings::get_autosave_settings_object() const {
|
||||
CHECK(are_inited_);
|
||||
auto exceptions = transform(exceptions_, [](const auto &exception) {
|
||||
@ -191,20 +214,140 @@ td_api::object_ptr<td_api::autosaveSettings> AutosaveManager::AutosaveSettings::
|
||||
broadcast_settings_.get_scope_autosave_settings_object(), std::move(exceptions));
|
||||
}
|
||||
|
||||
template <class StorerT>
|
||||
void AutosaveManager::AutosaveSettings::store(StorerT &storer) const {
|
||||
CHECK(are_inited_);
|
||||
bool has_exceptions = !exceptions_.empty();
|
||||
BEGIN_STORE_FLAGS();
|
||||
STORE_FLAG(has_exceptions);
|
||||
END_STORE_FLAGS();
|
||||
td::store(user_settings_, storer);
|
||||
td::store(chat_settings_, storer);
|
||||
td::store(broadcast_settings_, storer);
|
||||
if (has_exceptions) {
|
||||
td::store(narrow_cast<uint32>(exceptions_.size()), storer);
|
||||
for (auto &exception : exceptions_) {
|
||||
td::store(exception.first, storer);
|
||||
td::store(exception.second, storer);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
template <class ParserT>
|
||||
void AutosaveManager::AutosaveSettings::parse(ParserT &parser) {
|
||||
are_inited_ = true;
|
||||
bool has_exceptions;
|
||||
BEGIN_PARSE_FLAGS();
|
||||
PARSE_FLAG(has_exceptions);
|
||||
END_PARSE_FLAGS();
|
||||
td::parse(user_settings_, parser);
|
||||
td::parse(chat_settings_, parser);
|
||||
td::parse(broadcast_settings_, parser);
|
||||
if (has_exceptions) {
|
||||
uint32 size;
|
||||
td::parse(size, parser);
|
||||
for (size_t i = 0; i < size; i++) {
|
||||
DialogId dialog_id;
|
||||
DialogAutosaveSettings settings;
|
||||
td::parse(dialog_id, parser);
|
||||
td::parse(settings, parser);
|
||||
CHECK(dialog_id.is_valid());
|
||||
exceptions_.emplace(dialog_id, std::move(settings));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void AutosaveManager::get_autosave_settings(Promise<td_api::object_ptr<td_api::autosaveSettings>> &&promise) {
|
||||
if (settings_.are_inited_) {
|
||||
return promise.set_value(settings_.get_autosave_settings_object());
|
||||
}
|
||||
|
||||
reload_autosave_settings(std::move(promise));
|
||||
load_autosave_settings(std::move(promise));
|
||||
}
|
||||
|
||||
void AutosaveManager::reload_autosave_settings(Promise<td_api::object_ptr<td_api::autosaveSettings>> &&promise) {
|
||||
string AutosaveManager::get_autosave_settings_database_key() {
|
||||
return "autosave_settings";
|
||||
}
|
||||
|
||||
void AutosaveManager::load_autosave_settings(Promise<td_api::object_ptr<td_api::autosaveSettings>> &&promise) {
|
||||
load_settings_queries_.push_back(std::move(promise));
|
||||
if (load_settings_queries_.size() != 1) {
|
||||
return;
|
||||
}
|
||||
|
||||
if (G()->parameters().use_message_db) {
|
||||
G()->td_db()->get_sqlite_pmc()->get(
|
||||
get_autosave_settings_database_key(),
|
||||
PromiseCreator::lambda(
|
||||
[actor_id = actor_id(this)](string value) mutable {
|
||||
send_closure(actor_id, &AutosaveManager::on_load_autosave_settings_from_database, std::move(value));
|
||||
}));
|
||||
return;
|
||||
}
|
||||
|
||||
reload_autosave_settings();
|
||||
}
|
||||
|
||||
void AutosaveManager::on_load_autosave_settings_from_database(string value) {
|
||||
if (settings_.are_inited_) {
|
||||
CHECK(load_settings_queries_.empty());
|
||||
return;
|
||||
}
|
||||
if (G()->close_flag()) {
|
||||
return fail_promises(load_settings_queries_, Global::request_aborted_error());
|
||||
}
|
||||
if (value.empty()) {
|
||||
LOG(INFO) << "Autosave settings aren't found in database";
|
||||
return reload_autosave_settings();
|
||||
}
|
||||
|
||||
LOG(INFO) << "Successfully loaded autosave settings from database";
|
||||
|
||||
auto status = log_event_parse(settings_, value);
|
||||
if (status.is_error()) {
|
||||
LOG(ERROR) << "Can't load autosave settings: " << status;
|
||||
settings_ = {};
|
||||
return reload_autosave_settings();
|
||||
}
|
||||
|
||||
Dependencies dependencies;
|
||||
for (auto &exception : settings_.exceptions_) {
|
||||
dependencies.add_dialog_and_dependencies(exception.first);
|
||||
}
|
||||
if (!dependencies.resolve_force(td_, "on_load_autosave_settings_from_database")) {
|
||||
G()->td_db()->get_binlog_pmc()->erase(get_autosave_settings_database_key());
|
||||
settings_ = {};
|
||||
return reload_autosave_settings();
|
||||
}
|
||||
|
||||
settings_.are_inited_ = true;
|
||||
send_update_autosave_settings(td_api::make_object<td_api::autosaveSettingsScopePrivateChats>(),
|
||||
settings_.user_settings_);
|
||||
send_update_autosave_settings(td_api::make_object<td_api::autosaveSettingsScopeGroupChats>(),
|
||||
settings_.chat_settings_);
|
||||
send_update_autosave_settings(td_api::make_object<td_api::autosaveSettingsScopeChannelChats>(),
|
||||
settings_.broadcast_settings_);
|
||||
for (auto &exception : settings_.exceptions_) {
|
||||
send_update_autosave_settings(td_api::make_object<td_api::autosaveSettingsScopeChat>(exception.first.get()),
|
||||
exception.second);
|
||||
}
|
||||
|
||||
auto promises = std::move(load_settings_queries_);
|
||||
for (auto &promise : promises) {
|
||||
promise.set_value(settings_.get_autosave_settings_object());
|
||||
}
|
||||
}
|
||||
|
||||
void AutosaveManager::reload_autosave_settings() {
|
||||
if (G()->close_flag()) {
|
||||
return fail_promises(load_settings_queries_, Global::request_aborted_error());
|
||||
}
|
||||
if (settings_.are_being_reloaded_) {
|
||||
settings_.need_reload_ = true;
|
||||
return;
|
||||
}
|
||||
settings_.are_being_reloaded_ = true;
|
||||
|
||||
auto query_promise = PromiseCreator::lambda(
|
||||
[actor_id = actor_id(this)](Result<telegram_api::object_ptr<telegram_api::account_autoSaveSettings>> r_settings) {
|
||||
send_closure(actor_id, &AutosaveManager::on_get_autosave_settings, std::move(r_settings));
|
||||
@ -214,6 +357,14 @@ void AutosaveManager::reload_autosave_settings(Promise<td_api::object_ptr<td_api
|
||||
|
||||
void AutosaveManager::on_get_autosave_settings(
|
||||
Result<telegram_api::object_ptr<telegram_api::account_autoSaveSettings>> r_settings) {
|
||||
CHECK(settings_.are_being_reloaded_);
|
||||
settings_.are_being_reloaded_ = false;
|
||||
SCOPE_EXIT {
|
||||
if (settings_.need_reload_) {
|
||||
settings_.need_reload_ = false;
|
||||
reload_autosave_settings();
|
||||
}
|
||||
};
|
||||
if (G()->close_flag() && r_settings.is_ok()) {
|
||||
r_settings = Global::request_aborted_error();
|
||||
}
|
||||
@ -270,12 +421,20 @@ void AutosaveManager::on_get_autosave_settings(
|
||||
DialogAutosaveSettings());
|
||||
}
|
||||
|
||||
save_autosave_settings();
|
||||
|
||||
auto promises = std::move(load_settings_queries_);
|
||||
for (auto &promise : promises) {
|
||||
if (promise) {
|
||||
promise.set_value(settings_.get_autosave_settings_object());
|
||||
}
|
||||
}
|
||||
|
||||
void AutosaveManager::save_autosave_settings() {
|
||||
if (G()->parameters().use_message_db) {
|
||||
LOG(INFO) << "Save autosave settings to database";
|
||||
G()->td_db()->get_sqlite_pmc()->set(get_autosave_settings_database_key(),
|
||||
log_event_store(settings_).as_slice().str(), Auto());
|
||||
}
|
||||
}
|
||||
|
||||
void AutosaveManager::set_autosave_settings(td_api::object_ptr<td_api::AutosaveSettingsScope> &&scope,
|
||||
@ -284,6 +443,9 @@ void AutosaveManager::set_autosave_settings(td_api::object_ptr<td_api::AutosaveS
|
||||
if (scope == nullptr) {
|
||||
return promise.set_error(Status::Error(400, "Scope must be non-empty"));
|
||||
}
|
||||
if (!settings_.are_inited_) {
|
||||
return promise.set_error(Status::Error(400, "Autosave settings must be loaded first"));
|
||||
}
|
||||
auto new_settings = DialogAutosaveSettings(settings.get());
|
||||
DialogAutosaveSettings *old_settings = nullptr;
|
||||
bool users = false;
|
||||
@ -320,7 +482,6 @@ void AutosaveManager::set_autosave_settings(td_api::object_ptr<td_api::AutosaveS
|
||||
if (*old_settings == new_settings) {
|
||||
return promise.set_value(Unit());
|
||||
}
|
||||
if (settings_.are_inited_) {
|
||||
if (new_settings.are_inited_) {
|
||||
*old_settings = std::move(new_settings);
|
||||
send_update_autosave_settings(std::move(scope), *old_settings);
|
||||
@ -329,17 +490,23 @@ void AutosaveManager::set_autosave_settings(td_api::object_ptr<td_api::AutosaveS
|
||||
settings_.exceptions_.erase(dialog_id);
|
||||
send_update_autosave_settings(std::move(scope), DialogAutosaveSettings());
|
||||
}
|
||||
}
|
||||
|
||||
save_autosave_settings();
|
||||
|
||||
td_->create_handler<SaveAutoSaveSettingsQuery>(std::move(promise))
|
||||
->send(users, chats, broadcasts, dialog_id, new_settings.get_input_auto_save_settings());
|
||||
}
|
||||
|
||||
void AutosaveManager::clear_autosave_settings_excpetions(Promise<Unit> &&promise) {
|
||||
if (!settings_.are_inited_) {
|
||||
return promise.set_error(Status::Error(400, "Autosave settings must be loaded first"));
|
||||
}
|
||||
for (const auto &exception : settings_.exceptions_) {
|
||||
send_update_autosave_settings(td_api::make_object<td_api::autosaveSettingsScopeChat>(exception.first.get()),
|
||||
DialogAutosaveSettings());
|
||||
}
|
||||
settings_.exceptions_.clear();
|
||||
save_autosave_settings();
|
||||
td_->create_handler<DeleteAutoSaveExceptionsQuery>(std::move(promise))->send();
|
||||
}
|
||||
|
||||
|
@ -25,7 +25,7 @@ class AutosaveManager final : public Actor {
|
||||
public:
|
||||
AutosaveManager(Td *td, ActorShared<> parent);
|
||||
|
||||
void reload_autosave_settings(Promise<td_api::object_ptr<td_api::autosaveSettings>> &&promise);
|
||||
void reload_autosave_settings();
|
||||
|
||||
void get_autosave_settings(Promise<td_api::object_ptr<td_api::autosaveSettings>> &&promise);
|
||||
|
||||
@ -63,22 +63,44 @@ class AutosaveManager final : public Actor {
|
||||
bool operator==(const DialogAutosaveSettings &other) const;
|
||||
|
||||
bool operator!=(const DialogAutosaveSettings &other) const;
|
||||
|
||||
template <class StorerT>
|
||||
void store(StorerT &storer) const;
|
||||
|
||||
template <class ParserT>
|
||||
void parse(ParserT &parser);
|
||||
};
|
||||
|
||||
struct AutosaveSettings {
|
||||
bool are_inited_ = false;
|
||||
bool are_being_reloaded_ = false;
|
||||
bool need_reload_ = false;
|
||||
DialogAutosaveSettings user_settings_;
|
||||
DialogAutosaveSettings chat_settings_;
|
||||
DialogAutosaveSettings broadcast_settings_;
|
||||
FlatHashMap<DialogId, DialogAutosaveSettings, DialogIdHash> exceptions_;
|
||||
|
||||
td_api::object_ptr<td_api::autosaveSettings> get_autosave_settings_object() const;
|
||||
|
||||
template <class StorerT>
|
||||
void store(StorerT &storer) const;
|
||||
|
||||
template <class ParserT>
|
||||
void parse(ParserT &parser);
|
||||
};
|
||||
|
||||
void tear_down() final;
|
||||
|
||||
string get_autosave_settings_database_key();
|
||||
|
||||
void load_autosave_settings(Promise<td_api::object_ptr<td_api::autosaveSettings>> &&promise);
|
||||
|
||||
void on_load_autosave_settings_from_database(string value);
|
||||
|
||||
void on_get_autosave_settings(Result<telegram_api::object_ptr<telegram_api::account_autoSaveSettings>> r_settings);
|
||||
|
||||
void save_autosave_settings();
|
||||
|
||||
static td_api::object_ptr<td_api::updateAutosaveSettings> get_update_autosave_settings(
|
||||
td_api::object_ptr<td_api::AutosaveSettingsScope> &&scope, const DialogAutosaveSettings &settings);
|
||||
|
||||
|
@ -3917,7 +3917,7 @@ void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateTranscribedAudi
|
||||
}
|
||||
|
||||
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateAutoSaveSettings> update, Promise<Unit> &&promise) {
|
||||
td_->autosave_manager_->reload_autosave_settings(Auto());
|
||||
td_->autosave_manager_->reload_autosave_settings();
|
||||
promise.set_value(Unit());
|
||||
}
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user