Periodically ping server by bots.

GitOrigin-RevId: 3e173e32d564f50c352b96a6ad2d04b7ec72af58
This commit is contained in:
levlam 2018-02-11 21:33:32 +03:00
parent be5d7a7059
commit f4069240e5
4 changed files with 60 additions and 5 deletions

View File

@ -3637,10 +3637,15 @@ void Td::on_alarm_timeout_callback(void *td_ptr, int64 alarm_id) {
} }
void Td::on_alarm_timeout(int64 alarm_id) { void Td::on_alarm_timeout(int64 alarm_id) {
if (alarm_id == 0) { if (alarm_id == ONLINE_ALARM_ID) {
on_online_updated(false, true); on_online_updated(false, true);
return; return;
} }
if (alarm_id == PING_SERVER_ALARM_ID && updates_manager_ != nullptr) {
updates_manager_->ping_server();
alarm_timeout_.set_timeout_in(PING_SERVER_ALARM_ID, PING_SERVER_TIMEOUT + Random::fast(0, PING_SERVER_TIMEOUT / 5));
return;
}
auto it = pending_alarms_.find(alarm_id); auto it = pending_alarms_.find(alarm_id);
CHECK(it != pending_alarms_.end()); CHECK(it != pending_alarms_.end());
uint64 request_id = it->second; uint64 request_id = it->second;
@ -3657,9 +3662,9 @@ void Td::on_online_updated(bool force, bool send_update) {
create_handler<UpdateStatusQuery>()->send(!is_online_); create_handler<UpdateStatusQuery>()->send(!is_online_);
} }
if (is_online_) { if (is_online_) {
alarm_timeout_.set_timeout_in(0, ONLINE_TIMEOUT); alarm_timeout_.set_timeout_in(ONLINE_ALARM_ID, ONLINE_TIMEOUT);
} else { } else {
alarm_timeout_.cancel_timeout(0); alarm_timeout_.cancel_timeout(ONLINE_ALARM_ID);
} }
} }
@ -3822,6 +3827,10 @@ void Td::on_result(NetQueryPtr query) {
updates_manager_->schedule_get_difference("failed to fetch update"); updates_manager_->schedule_get_difference("failed to fetch update");
} else { } else {
updates_manager_->on_get_updates(std::move(ptr)); updates_manager_->on_get_updates(std::move(ptr));
if (auth_manager_->is_bot()) {
alarm_timeout_.set_timeout_in(PING_SERVER_ALARM_ID,
PING_SERVER_TIMEOUT + Random::fast(0, PING_SERVER_TIMEOUT / 5));
}
} }
return; return;
} }
@ -4077,8 +4086,9 @@ void Td::clear() {
} }
if (is_online_) { if (is_online_) {
is_online_ = false; is_online_ = false;
alarm_timeout_.cancel_timeout(0); alarm_timeout_.cancel_timeout(ONLINE_ALARM_ID);
} }
alarm_timeout_.cancel_timeout(PING_SERVER_ALARM_ID);
LOG(DEBUG) << "Requests was answered " << timer; LOG(DEBUG) << "Requests was answered " << timer;
// close all pure actors // close all pure actors
@ -4416,7 +4426,11 @@ void Td::send_update(tl_object_ptr<td_api::Update> &&object) {
} }
void Td::send_result(uint64 id, tl_object_ptr<td_api::Object> object) { void Td::send_result(uint64 id, tl_object_ptr<td_api::Object> object) {
LOG_IF(ERROR, id == 0) << "Sending " << to_string(object) << " through send_result"; if (id == 0) {
LOG(ERROR) << "Sending " << to_string(object) << " through send_result";
return;
}
auto it = request_set_.find(id); auto it = request_set_.find(id);
if (it != request_set_.end()) { if (it != request_set_.end()) {
request_set_.erase(it); request_set_.erase(it);

View File

@ -193,7 +193,10 @@ class Td final : public NetQueryCallback {
private: private:
static constexpr const char *tdlib_version = "1.1.2"; static constexpr const char *tdlib_version = "1.1.2";
static constexpr int64 ONLINE_ALARM_ID = 0;
static constexpr int32 ONLINE_TIMEOUT = 240; static constexpr int32 ONLINE_TIMEOUT = 240;
static constexpr int64 PING_SERVER_ALARM_ID = -1;
static constexpr int32 PING_SERVER_TIMEOUT = 300;
void send_result(uint64 id, tl_object_ptr<td_api::Object> object); void send_result(uint64 id, tl_object_ptr<td_api::Object> object);
void send_error(uint64 id, Status error); void send_error(uint64 id, Status error);

View File

@ -92,6 +92,29 @@ class GetUpdatesStateQuery : public Td::ResultHandler {
} }
}; };
class PingServerQuery : public Td::ResultHandler {
public:
void send() {
send_query(G()->net_query_creator().create(create_storer(telegram_api::updates_getState())));
}
void on_result(uint64 id, BufferSlice packet) override {
auto result_ptr = fetch_result<telegram_api::updates_getState>(packet);
if (result_ptr.is_error()) {
return on_error(id, result_ptr.move_as_error());
}
auto state = result_ptr.move_as_ok();
CHECK(state->get_id() == telegram_api::updates_state::ID);
td->updates_manager_->on_server_pong(std::move(state));
}
void on_error(uint64 id, Status status) override {
status.ignore();
td->updates_manager_->on_server_pong(nullptr);
}
};
class GetDifferenceQuery : public Td::ResultHandler { class GetDifferenceQuery : public Td::ResultHandler {
public: public:
void send() { void send() {
@ -841,6 +864,17 @@ void UpdatesManager::init_state() {
get_difference("init_state"); get_difference("init_state");
} }
void UpdatesManager::ping_server() {
td_->create_handler<PingServerQuery>()->send();
}
void UpdatesManager::on_server_pong(tl_object_ptr<telegram_api::updates_state> &&state) {
LOG(INFO) << "Receive " << oneline(to_string(state));
if (state == nullptr || state->pts_ > get_pts() || state->seq_ > seq_) {
get_difference("on server pong");
}
}
void UpdatesManager::process_get_difference_updates( void UpdatesManager::process_get_difference_updates(
vector<tl_object_ptr<telegram_api::Message>> &&new_messages, vector<tl_object_ptr<telegram_api::Message>> &&new_messages,
vector<tl_object_ptr<telegram_api::EncryptedMessage>> &&new_encrypted_messages, int32 qts, vector<tl_object_ptr<telegram_api::EncryptedMessage>> &&new_encrypted_messages, int32 qts,

View File

@ -42,6 +42,10 @@ class UpdatesManager : public Actor {
void init_state(); void init_state();
void ping_server();
void on_server_pong(tl_object_ptr<telegram_api::updates_state> &&state);
int32 get_pts() const { int32 get_pts() const {
return pts_manager_.mem_pts(); return pts_manager_.mem_pts();
} }