andreacavalli/tdlib-fork
Archived
1
0
Fork 0
Code Pull Requests Releases Activity

Persistent getDialogQuery.

Browse Source 
GitOrigin-RevId: e882cd049aac7aa86cf03750a78c3b92635f5379
This commit is contained in:
levlam 2018-06-01 19:58:23 +03:00
parent 9cd334cc4b
commit 98aba6304c
4 changed files with 66 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

65
td/telegram/MessagesManager.cpp
View File

@ -8832,10 +8832,10 @@ int32 MessagesManager::calc_new_unread_count(Dialog *d, MessageId max_message_id
void MessagesManager::repair_server_unread_count(DialogId dialog_id, int32 unread_count) {
LOG(INFO) << "Repair server unread count in " << dialog_id << " from " << unread_count;
create_actor<SleepActor>("RepairServerUnreadCountSleepActor", 0.5,
create_actor<SleepActor>("RepairServerUnreadCountSleepActor", 0.2,
PromiseCreator::lambda([actor_id = actor_id(this), dialog_id](Result<Unit> result) {
send_closure(actor_id, &MessagesManager::send_get_dialog_query, dialog_id,
Promise<Unit>());
send_closure(actor_id, &MessagesManager::send_get_dialog_query, dialog_id, Promise<Unit>(),
0);
}))
.release();
}
@ -19981,7 +19981,22 @@ void MessagesManager::on_get_dialog_notification_settings_query_finished(DialogI
}
}
void MessagesManager::send_get_dialog_query(DialogId dialog_id, Promise<Unit> &&promise) {
class MessagesManager::GetDialogFromServerLogEvent {
public:
DialogId dialog_id_;
template <class StorerT>
void store(StorerT &storer) const {
td::store(dialog_id_, storer);
}
template <class ParserT>
void parse(ParserT &parser) {
td::parse(dialog_id_, parser);
}
};
void MessagesManager::send_get_dialog_query(DialogId dialog_id, Promise<Unit> &&promise, uint64 logevent_id) {
if (td_->auth_manager_->is_bot() || dialog_id.get_type() == DialogType::SecretChat) {
return promise.set_error(Status::Error(500, "Wrong getDialog query"));
}
@ -19996,6 +20011,18 @@ void MessagesManager::send_get_dialog_query(DialogId dialog_id, Promise<Unit> &&
return;
}
if (logevent_id == 0 && G()->parameters().use_message_db) {
GetDialogFromServerLogEvent logevent;
logevent.dialog_id_ = dialog_id;
auto storer = LogEventStorerImpl<GetDialogFromServerLogEvent>(logevent);
logevent_id = BinlogHelper::add(G()->td_db()->get_binlog(), LogEvent::HandlerType::GetDialogFromServer, storer);
}
if (logevent_id != 0) {
get_dialog_query_logevent_id_[dialog_id] = logevent_id;
}
LOG(INFO) << "Send get " << dialog_id << " query";
td_->create_handler<GetDialogQuery>()->send(dialog_id);
}
@ -20006,6 +20033,12 @@ void MessagesManager::on_get_dialog_query_finished(DialogId dialog_id, Status &&
auto promises = std::move(it->second);
get_dialog_queries_.erase(it);
auto logevent_it = get_dialog_query_logevent_id_.find(dialog_id);
if (logevent_it != get_dialog_query_logevent_id_.end()) {
BinlogHelper::erase(G()->td_db()->get_binlog(), logevent_it->second);
get_dialog_query_logevent_id_.erase(logevent_it);
}
for (auto &promise : promises) {
if (status.is_ok()) {
promise.set_value(Unit());
@ -25563,6 +25596,30 @@ void MessagesManager::on_binlog_events(vector<BinlogEvent> &&events) {
change_dialog_report_spam_state_on_server(dialog_id, log_event.is_spam_dialog_, event.id_, Promise<Unit>());
break;
}
case LogEvent::HandlerType::GetDialogFromServer: {
if (!G()->parameters().use_message_db) {
BinlogHelper::erase(G()->td_db()->get_binlog(), event.id_);
break;
}
GetDialogFromServerLogEvent log_event;
log_event_parse(log_event, event.data_).ensure();
auto dialog_id = log_event.dialog_id_;
Dependencies dependencies;
add_dialog_dependencies(dependencies, dialog_id);
resolve_dependencies_force(dependencies);
get_dialog_force(dialog_id); // load it if exists
if (!have_input_peer(dialog_id, AccessRights::Read)) {
BinlogHelper::erase(G()->td_db()->get_binlog(), event.id_);
break;
}
send_get_dialog_query(dialog_id, Promise<Unit>(), event.id_);
break;
}
case LogEvent::HandlerType::GetChannelDifference: {
GetChannelDifferenceLogEvent log_event;
log_event_parse(log_event, event.data_).ensure();

4
td/telegram/MessagesManager.h
View File

@ -1857,6 +1857,7 @@ class MessagesManager : public Actor {
class SendMessageLogEvent;
class SendScreenshotTakenNotificationMessageLogEvent;
class ToggleDialogIsPinnedOnServerLogEvent;
class GetDialogFromServerLogEvent;
static constexpr size_t MAX_GROUPED_MESSAGES = 10; // server side limit
static constexpr int32 MAX_GET_DIALOGS = 100; // server side limit
@ -2334,7 +2335,7 @@ class MessagesManager : public Actor {
void on_get_dialogs_from_database(vector<BufferSlice> &&dialogs, Promise<Unit> &&promise);
void send_get_dialog_query(DialogId dialog_id, Promise<Unit> &&promise);
void send_get_dialog_query(DialogId dialog_id, Promise<Unit> &&promise, uint64 logevent_id = 0);
void send_search_public_dialogs_query(const string &query, Promise<Unit> &&promise);
@ -2801,6 +2802,7 @@ class MessagesManager : public Actor {
std::unordered_map<DialogId, vector<Promise<Unit>>, DialogIdHash> get_dialog_notification_settings_queries_;
std::unordered_map<DialogId, vector<Promise<Unit>>, DialogIdHash> get_dialog_queries_;
std::unordered_map<DialogId, uint64, DialogIdHash> get_dialog_query_logevent_id_;
std::unordered_map<FullMessageId, int32, FullMessageIdHash> replied_by_yet_unsent_messages_;

1
td/telegram/TdDb.cpp
View File

@ -86,6 +86,7 @@ Status init_binlog(Binlog &binlog, string path, BinlogKeyValue<Binlog> &binlog_p
case LogEvent::HandlerType::UpdateScopeNotificationSettingsOnServer:
case LogEvent::HandlerType::ResetAllNotificationSettingsOnServer:
case LogEvent::HandlerType::ChangeDialogReportSpamStateOnServer:
case LogEvent::HandlerType::GetDialogFromServer:
case LogEvent::HandlerType::GetChannelDifference:
events.to_messages_manager.push_back(event.clone());
break;

1
td/telegram/logevent/LogEvent.h
View File

@ -87,6 +87,7 @@ class LogEvent {
UpdateScopeNotificationSettingsOnServer = 0x110,
ResetAllNotificationSettingsOnServer = 0x111,
ChangeDialogReportSpamStateOnServer = 0x112,
GetDialogFromServer = 0x113,
GetChannelDifference = 0x140,
ConfigPmcMagic = 0x1f18,
BinlogPmcMagic = 0x4327