tdlight/tdnet/td/net/GetHostByNameActor.h

75 lines
1.9 KiB
C
Raw Normal View History

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2019
//
// 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/actor/PromiseFuture.h"
#include "td/utils/logging.h"
#include "td/utils/port/IPAddress.h"
#include "td/utils/Status.h"
#include <unordered_map>
namespace td {
extern int VERBOSITY_NAME(dns_resolver);
class GetHostByNameActor final : public Actor {
public:
enum class ResolverType { Native, Google };
struct Options {
static constexpr int32 DEFAULT_CACHE_TIME = 60 * 29; // 29 minutes
static constexpr int32 DEFAULT_ERROR_CACHE_TIME = 60 * 5; // 5 minutes
vector<ResolverType> resolver_types{ResolverType::Native};
int32 scheduler_id{-1};
int32 ok_timeout{DEFAULT_CACHE_TIME};
int32 error_timeout{DEFAULT_ERROR_CACHE_TIME};
};
explicit GetHostByNameActor(Options options);
void run(std::string host, int port, bool prefer_ipv6, Promise<IPAddress> promise);
private:
void on_query_result(std::string host, bool prefer_ipv6, Result<IPAddress> result);
struct Value {
Result<IPAddress> ip;
double expire_at;
Value(Result<IPAddress> ip, double expire_at) : ip(std::move(ip)), expire_at(expire_at) {
}
Result<IPAddress> get_ip_port(int port) const {
auto result = ip.clone();
if (result.is_ok()) {
result.ok_ref().set_port(port);
}
return result;
}
};
std::unordered_map<string, Value> cache_[2];
struct Query {
ActorOwn<> query;
size_t pos = 0;
string real_host;
double begin_time;
std::vector<std::pair<int, Promise<IPAddress>>> promises;
};
std::unordered_map<string, Query> active_queries_[2];
Options options_;
void run_query(std::string host, bool prefer_ipv6, Query &query);
};
} // namespace td