tdlight-telegram-bot-api/telegram-bot-api/ClientManager.h

104 lines
2.9 KiB
C
Raw Normal View History

//
2022-12-31 22:31:16 +01:00
// 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 "telegram-bot-api/Client.h"
#include "telegram-bot-api/Query.h"
#include "telegram-bot-api/Stats.h"
#include "telegram-bot-api/Watchdog.h"
#include "td/actor/actor.h"
#include "td/utils/buffer.h"
#include "td/utils/common.h"
#include "td/utils/Container.h"
#include "td/utils/FlatHashMap.h"
#include "td/utils/FloodControlFast.h"
2022-06-30 19:27:08 +02:00
#include "td/utils/Promise.h"
#include "td/utils/Slice.h"
#include <memory>
#include <utility>
namespace telegram_bot_api {
struct ClientParameters;
struct SharedData;
class ClientManager final : public td::Actor {
public:
struct TokenRange {
td::uint64 rem;
td::uint64 mod;
bool operator()(td::uint64 x) {
return x % mod == rem;
}
};
ClientManager(std::shared_ptr<const ClientParameters> parameters, TokenRange token_range)
: parameters_(std::move(parameters)), token_range_(token_range) {
}
2022-10-06 21:18:36 +02:00
void dump_statistics();
void send(PromisedQueryPtr query);
void user_login(PromisedQueryPtr query);
bool check_flood_limits(PromisedQueryPtr &query, bool is_user_login=false);
2022-08-21 20:16:05 +02:00
void get_stats(td::Promise<td::BufferSlice> promise, td::vector<std::pair<td::string, td::string>> args, bool as_json);
void close(td::Promise<td::Unit> &&promise);
private:
class ClientInfo {
public:
BotStatActor stat_;
td::string token_;
td::int64 tqueue_id_;
td::ActorOwn<Client> client_;
};
td::Container<ClientInfo> clients_;
BotStatActor stat_{td::ActorId<BotStatActor>()};
std::shared_ptr<const ClientParameters> parameters_;
TokenRange token_range_;
td::FlatHashMap<td::string, td::uint64> token_to_id_;
td::FlatHashMap<td::string, td::FloodControlFast> flood_controls_;
td::FlatHashMap<td::int64, td::uint64> active_client_count_;
bool close_flag_ = false;
td::vector<td::Promise<td::Unit>> close_promises_;
td::ActorOwn<Watchdog> watchdog_id_;
2022-10-13 23:46:54 +02:00
double next_tqueue_gc_time_ = 0.0;
2022-10-06 20:42:33 +02:00
td::int64 tqueue_deleted_events_ = 0;
td::int64 last_tqueue_deleted_events_ = 0;
2023-01-06 14:30:12 +01:00
static constexpr double WATCHDOG_TIMEOUT = 0.25;
static td::int64 get_tqueue_id(td::int64 user_id, bool is_test_dc);
static PromisedQueryPtr get_webhook_restore_query(td::Slice token, bool is_user, td::Slice webhook_info,
std::shared_ptr<SharedData> shared_data);
2023-01-06 13:52:53 +01:00
struct TopClients {
td::int32 active_count = 0;
td::vector<td::uint64> top_client_ids;
};
TopClients get_top_clients(std::size_t max_count, td::Slice token_filter);
2022-01-25 16:18:44 +01:00
void start_up() final;
void raw_event(const td::Event::Raw &event) final;
void timeout_expired() final;
2022-01-25 16:18:44 +01:00
void hangup_shared() final;
void close_db();
void finish_close();
};
} // namespace telegram_bot_api