// // Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2023 // // Distributed under the Boost Software License, Version 1.0. (See accompanying // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) // #pragma once #include "td/actor/actor.h" #include "td/utils/common.h" #include "td/utils/port/Stat.h" #include "td/utils/Time.h" #include "td/utils/TimedStat.h" #include namespace telegram_bot_api { struct StatItem { td::string key_; td::string value_; }; class CpuStat { public: void on_event(const td::CpuStat &event) { if (cnt_ == 0) { first_ = event; cnt_ = 1; } else { cnt_ = 2; last_ = event; } } td::vector as_vector() const; private: int cnt_ = 0; td::CpuStat first_; td::CpuStat last_; }; class ServerCpuStat { public: static ServerCpuStat &instance() { static ServerCpuStat stat; return stat; } static void update(double now); static td::string get_description(); td::vector as_vector(double now); private: static constexpr std::size_t SIZE = 4; static constexpr const char *DESCR[SIZE] = {"inf", "5sec", "1min", "1hour"}; static constexpr int DURATIONS[SIZE] = {0, 5, 60, 60 * 60}; std::mutex mutex_; td::TimedStat stat_[SIZE]; ServerCpuStat(); }; class ServerBotInfo { public: td::string id_; td::string token_; td::string username_; td::string webhook_; bool has_webhook_certificate_ = false; td::int32 head_update_id_ = 0; td::int32 tail_update_id_ = 0; td::int32 webhook_max_connections_ = 0; std::size_t pending_update_count_ = 0; double start_time_ = 0; }; struct ServerBotStat { double request_count_ = 0; double request_bytes_ = 0; double request_file_count_ = 0; double request_files_bytes_ = 0; td::int64 request_files_max_bytes_ = 0; double response_count_ = 0; double response_count_ok_ = 0; double response_count_error_ = 0; double response_bytes_ = 0; double update_count_ = 0; void normalize(double duration); void add(const ServerBotStat &stat); struct Update {}; void on_event(const Update &update) { update_count_ += 1; } struct Response { bool ok_; size_t size_; td::int64 file_count_; td::int64 files_size_; }; void on_event(const Response &response) { response_count_++; if (response.ok_) { response_count_ok_++; } else { response_count_error_++; } response_bytes_ += static_cast(response.size_); } struct Request { td::int64 size_; td::int64 file_count_; td::int64 files_size_; td::int64 files_max_size_; }; void on_event(const Request &request) { request_count_++; request_bytes_ += static_cast(request.size_); request_file_count_ += static_cast(request.file_count_); request_files_bytes_ += static_cast(request.files_size_); request_files_max_bytes_ = td::max(request_files_max_bytes_, request.files_max_size_); } td::vector as_vector() const; }; class BotStatActor final : public td::Actor { public: BotStatActor() = default; explicit BotStatActor(td::ActorId parent) : parent_(parent) { for (std::size_t i = 0; i < SIZE; i++) { stat_[i] = td::TimedStat(DURATIONS[i], td::Time::now()); } register_actor("ServerBotStat", this).release(); } BotStatActor(const BotStatActor &) = delete; BotStatActor &operator=(const BotStatActor &) = delete; BotStatActor(BotStatActor &&) = default; BotStatActor &operator=(BotStatActor &&other) noexcept { if (!empty()) { do_stop(); } this->Actor::operator=(std::move(other)); std::move(other.stat_, other.stat_ + SIZE, stat_); parent_ = other.parent_; return *this; } ~BotStatActor() final = default; template void add_event(const EventT &event, double now) { last_activity_timestamp_ = now; for (auto &stat : stat_) { stat.add_event(event, now); } on_event(event); if (!parent_.empty()) { send_closure(parent_, &BotStatActor::add_event, event, now); } } td::vector as_vector(double now); static td::string get_description(); double get_score(double now); double get_minute_update_count(double now); td::int64 get_active_request_count() const; td::int64 get_active_file_upload_bytes() const; td::int64 get_active_file_upload_count() const; bool is_active(double now) const; private: static constexpr std::size_t SIZE = 4; static constexpr const char *DESCR[SIZE] = {"inf", "5sec", "1min", "1hour"}; static constexpr int DURATIONS[SIZE] = {0, 5, 60, 60 * 60}; td::TimedStat stat_[SIZE]; td::ActorId parent_; double last_activity_timestamp_ = -1e9; td::int64 active_request_count_ = 0; td::int64 active_file_upload_bytes_ = 0; td::int64 active_file_upload_count_ = 0; void on_event(const ServerBotStat::Update &update) { } void on_event(const ServerBotStat::Response &response) { active_request_count_--; active_file_upload_count_ -= response.file_count_; active_file_upload_bytes_ -= response.files_size_; CHECK(active_request_count_ >= 0); CHECK(active_file_upload_bytes_ >= 0); } void on_event(const ServerBotStat::Request &request) { active_request_count_++; active_file_upload_count_ += request.file_count_; active_file_upload_bytes_ += request.files_size_; } }; } // namespace telegram_bot_api