Improve limit for active queries.

This commit is contained in:
levlam 2022-11-30 17:30:28 +03:00
parent ed0532bcf7
commit 48a5609de2
3 changed files with 18 additions and 4 deletions

View File

@ -4070,9 +4070,11 @@ void Client::start_up() {
void Client::send(PromisedQueryPtr query) { void Client::send(PromisedQueryPtr query) {
if (!query->is_internal()) { if (!query->is_internal()) {
query->set_stat_actor(stat_actor_); query->set_stat_actor(stat_actor_);
if (!parameters_->local_mode_ && !is_local_method(query->method())) { if (!parameters_->local_mode_ && !is_local_method(query->method()) &&
const BotStatActor *stat = stat_actor_.get_actor_unsafe(); td::Time::now() > parameters_->start_time_ + 10 * 60) {
if (stat->get_active_request_count() > 5000) { BotStatActor *stat = stat_actor_.get_actor_unsafe();
auto update_per_minute = static_cast<int64>(stat->get_minute_update_count(td::Time::now()) * 60);
if (stat->get_active_request_count() > 500 + update_per_minute) {
LOG(INFO) << "Fail a query, because there are too many active queries: " << *query; LOG(INFO) << "Fail a query, because there are too many active queries: " << *query;
return query->set_retry_after_error(60); return query->set_retry_after_error(60);
} }
@ -4080,7 +4082,7 @@ void Client::send(PromisedQueryPtr query) {
LOG(INFO) << "Fail a query, because the total size of active file uploads is too big: " << *query; LOG(INFO) << "Fail a query, because the total size of active file uploads is too big: " << *query;
return query->set_retry_after_error(60); return query->set_retry_after_error(60);
} }
if (stat->get_active_file_upload_count() > 500 && !query->files().empty()) { if (stat->get_active_file_upload_count() > 100 + update_per_minute / 5 && !query->files().empty()) {
LOG(INFO) << "Fail a query, because there are too many active file uploads: " << *query; LOG(INFO) << "Fail a query, because there are too many active file uploads: " << *query;
return query->set_retry_after_error(60); return query->set_retry_after_error(60);
} }

View File

@ -160,6 +160,15 @@ double BotStatActor::get_score(double now) {
return result; return result;
} }
double BotStatActor::get_minute_update_count(double now) {
auto minute_stat = stat_[2].stat_duration(now);
double result = minute_stat.first.update_count_;
if (minute_stat.second != 0) {
result /= minute_stat.second;
}
return result;
}
td::int64 BotStatActor::get_active_request_count() const { td::int64 BotStatActor::get_active_request_count() const {
return active_request_count_; return active_request_count_;
} }

View File

@ -182,10 +182,13 @@ class BotStatActor final : public td::Actor {
} }
td::vector<StatItem> as_vector(double now); td::vector<StatItem> as_vector(double now);
td::string get_description() const; td::string get_description() const;
double get_score(double now); double get_score(double now);
double get_minute_update_count(double now);
td::int64 get_active_request_count() const; td::int64 get_active_request_count() const;
td::int64 get_active_file_upload_bytes() const; td::int64 get_active_file_upload_bytes() const;