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

Session: use MAX_INFLIGHT_QUERIES to limit count of simultaneously sent queries

Browse Source 
GitOrigin-RevId: c9c64de982c099b95c6c4bd31d28369f4315f81b
This commit is contained in:
Arseny Smirnov 2020-07-31 20:19:18 +03:00
parent bbea867145
commit 5a08290ae0
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
td/telegram/net/Session.cpp
View File

@ -1070,7 +1070,8 @@ void Session::connection_open_finish(ConnectionInfo *info,
info->state = ConnectionInfo::State::Ready;
info->created_at = Time::now_cached();
info->wakeup_at = Time::now_cached() + 10;
if (unknown_queries_.size() > 1024) {
if (unknown_queries_.size() > MAX_INFLIGHT_QUERIES) {
LOG(ERROR) << "With current limits `Too much queries with unknown state` error must be impossible";
on_session_failed(Status::Error("Too much queries with unknown state"));
return;
}
@ -1311,12 +1312,12 @@ void Session::loop() {
while (main_connection_.state == ConnectionInfo::State::Ready) {
if (auth_data_.is_ready(Time::now_cached())) {
if (need_send_query()) {
while (!pending_queries_.empty()) {
while (!pending_queries_.empty() && sent_queries_.size() < MAX_INFLIGHT_QUERIES) {
auto &query = pending_queries_.front();
connection_send_query(&main_connection_, std::move(query));
pending_queries_.pop_front();
need_flush = true;
}
need_flush = true;
}
if (need_send_bind_key()) {
// send auth.bindTempAuthKey

1
td/telegram/net/Session.h
View File

@ -157,6 +157,7 @@ class Session final
bool close_flag_ = false;
static constexpr double ACTIVITY_TIMEOUT = 60 * 5;
static constexpr size_t MAX_INFLIGHT_QUERIES = 1024;
struct ContainerInfo {
size_t ref_cnt;