tdlight-telegram-bot-api/telegram-bot-api/ClientParameters.h
2024-09-09 00:20:53 +02:00

139 lines
3.4 KiB
C++

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2024
//
// 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/db/KeyValueSyncInterface.h"
#include "td/db/TQueue.h"
#include "td/net/GetHostByNameActor.h"
#include "td/actor/actor.h"
#include "td/utils/common.h"
#include "td/utils/List.h"
#include "td/utils/port/IPAddress.h"
#include <atomic>
#include <limits>
#include <memory>
namespace td {
class NetQueryStats;
} // namespace td
namespace telegram_bot_api {
struct SharedData {
std::atomic<td::uint64> query_count_{0};
std::atomic<size_t> query_list_size_{0};
std::atomic<int> next_verbosity_level_{-1};
// not thread-safe, must be used from a single thread
td::ListNode query_list_;
td::unique_ptr<td::KeyValueSyncInterface> webhook_db_;
td::unique_ptr<td::KeyValueSyncInterface> user_db_;
td::unique_ptr<td::TQueue> tqueue_;
double unix_time_difference_{-1e100};
static constexpr size_t TQUEUE_EVENT_BUFFER_SIZE = 1000;
td::TQueue::Event event_buffer_[TQUEUE_EVENT_BUFFER_SIZE];
td::int32 get_unix_time(double now) const {
auto result = unix_time_difference_ + now;
if (result <= 0) {
return 0;
}
if (result >= std::numeric_limits<td::int32>::max()) {
return std::numeric_limits<td::int32>::max();
}
return static_cast<td::int32>(result);
}
static td::int32 get_file_gc_scheduler_id() {
// the same scheduler as for file GC in Td
return 2;
}
static td::int32 get_client_scheduler_id() {
// the thread for ClientManager and all Clients
return 4;
}
static td::int32 get_watchdog_scheduler_id() {
// the thread for watchdogs
return 5;
}
static td::int32 get_slow_incoming_http_scheduler_id() {
// the thread for slow incoming HTTP connections
return 6;
}
static td::int32 get_slow_outgoing_http_scheduler_id() {
// the thread for slow outgoing HTTP connections
return 7;
}
static td::int32 get_dns_resolver_scheduler_id() {
// the thread for DNS resolving
return 8;
}
static td::int32 get_binlog_scheduler_id() {
// the thread for TQueue and webhook binlogs
return 9;
}
static td::int32 get_webhook_certificate_scheduler_id() {
// the thread for webhook certificate processing
return 10;
}
static td::int32 get_statistics_thread_id() {
// the thread for CPU usage updating
return 11;
}
static td::int32 get_thread_count() {
return 12;
}
};
struct ClientParameters {
td::string working_directory_;
bool allow_colon_in_filenames_ = true;
bool local_mode_ = false;
bool allow_http_ = false;
bool use_relative_path_ = false;
bool no_file_limit_ = false;
bool allow_users_ = false;
bool allow_users_registration_ = false;
bool stats_hide_sensible_data_ = false;
td::int32 api_id_ = 0;
td::string api_hash_;
td::string version_;
td::int32 default_max_webhook_connections_ = 0;
td::IPAddress webhook_proxy_ip_address_;
td::uint32 max_batch_operations = 10000;
double start_time_ = 0;
td::int32 file_expiration_timeout_seconds_ = 3600;
td::ActorId<td::GetHostByNameActor> get_host_by_name_actor_id_;
std::shared_ptr<SharedData> shared_data_;
std::shared_ptr<td::NetQueryStats> net_query_stats_;
};
} // namespace telegram_bot_api