Add prefer_ipv6 parameter tp GetHostByNameActor.

GitOrigin-RevId: 727fc30d94657399f1e49efc208b6951a7677d2f

1 files changed, 5 insertions, 5 deletions

diff --git a/tdnet/td/net/GetHostByNameActor.cpp b/tdnet/td/net/GetHostByNameActor.cpp
index f2b9313f6..ed6d8ce7e 100644
--- a/tdnet/td/net/GetHostByNameActor.cpp
+++ b/tdnet/td/net/GetHostByNameActor.cpp
@@ -15,13 +15,13 @@ GetHostByNameActor::GetHostByNameActor(int32 ok_timeout, int32 error_timeout)
     : ok_timeout_(ok_timeout), error_timeout_(error_timeout) {
 }
 
-void GetHostByNameActor::run(std::string host, int port, td::Promise<td::IPAddress> promise) {
-  auto r_ip = load_ip(std::move(host), port);
+void GetHostByNameActor::run(std::string host, int port, bool prefer_ipv6, td::Promise<td::IPAddress> promise) {
+  auto r_ip = load_ip(std::move(host), port, prefer_ipv6);
   promise.set_result(std::move(r_ip));
 }
 
-Result<td::IPAddress> GetHostByNameActor::load_ip(string host, int port) {
-  auto &value = cache_.emplace(host, Value{{}, 0}).first->second;
+Result<td::IPAddress> GetHostByNameActor::load_ip(string host, int port, bool prefer_ipv6) {
+  auto &value = cache_[prefer_ipv6].emplace(host, Value{{}, 0}).first->second;
   auto begin_time = td::Time::now();
   if (value.expire_at > begin_time) {
     auto ip = value.ip.clone();
@@ -33,7 +33,7 @@ Result<td::IPAddress> GetHostByNameActor::load_ip(string host, int port) {
   }
 
   td::IPAddress ip;
-  auto status = ip.init_host_port(host, port);
+  auto status = ip.init_host_port(host, port, prefer_ipv6);
   auto end_time = td::Time::now();
   LOG(WARNING) << "Init host = " << host << ", port = " << port << " in " << end_time - begin_time << " seconds to "
                << ip;