Allow passing ProxyUrlAliasingRules to clients directly

commit_hash:5493d9ed32b24050fa5feb08fe13097302ca0a23

2 files changed, 33 insertions, 0 deletions

diff --git a/yt/cpp/mapreduce/interface/config.cpp b/yt/cpp/mapreduce/interface/config.cpp
index cdabeaf3240..901f59fc215 100644
--- a/yt/cpp/mapreduce/interface/config.cpp
+++ b/yt/cpp/mapreduce/interface/config.cpp
@@ -190,6 +190,30 @@ void TConfig::LoadTimings()
     HostListUpdateInterval = TDuration::Seconds(60);
 }
 
+void TConfig::LoadProxyUrlAliasingRules()
+{
+    TString strConfig = GetEnv("YT_PROXY_URL_ALIASING_CONFIG");
+    if (!strConfig) {
+        return;
+    }
+
+    NYT::TNode nodeConfig;
+
+    try {
+        nodeConfig = NodeFromYsonString(strConfig);
+        Y_ENSURE(nodeConfig.IsMap());
+    } catch (const yexception& exc) {
+        ythrow yexception()
+            << "Failed to parse YT_PROXY_URL_ALIASING_CONFIG (it must be yson map): "
+            << exc;
+    }
+
+    for (const auto& [key, value] : nodeConfig.AsMap()) {
+        Y_ENSURE(value.IsString(), "Proxy url is not string");
+        ProxyUrlAliasingRules.emplace(key, value.AsString());
+    }
+}
+
 void TConfig::Reset()
 {
     Hosts = GetEnv("YT_HOSTS", DefaultHosts);
@@ -219,6 +243,7 @@ void TConfig::Reset()
     LoadToken();
     LoadSpec();
     LoadTimings();
+    LoadProxyUrlAliasingRules();
 
     CacheUploadDeduplicationMode = GetUploadingDeduplicationMode("YT_UPLOAD_DEDUPLICATION", EUploadDeduplicationMode::Host);
     CacheUploadDeduplicationThreshold = 10_MB;
diff --git a/yt/cpp/mapreduce/interface/config.h b/yt/cpp/mapreduce/interface/config.h
index 6e6d7b191c2..7aba0ccd60f 100644
--- a/yt/cpp/mapreduce/interface/config.h
+++ b/yt/cpp/mapreduce/interface/config.h
@@ -9,6 +9,7 @@
 
 #include <util/generic/maybe.h>
 #include <util/generic/string.h>
+#include <util/generic/hash.h>
 #include <util/generic/hash_set.h>
 
 #include <util/datetime/base.h>
@@ -111,6 +112,12 @@ struct TConfig
     /// @brief Represents the role involved in RPC proxy configuration.
     TString RpcProxyRole;
 
+    /// @brief Proxy url aliasing rules to be used for connection.
+    ///
+    /// You can pass here "foo" => "fqdn:port" and afterwards use "foo" as handy alias,
+    /// while all connections will be made to "fqdn:port" address.
+    THashMap<TString, TString> ProxyUrlAliasingRules;
+
     ///
     /// For historical reasons mapreduce client uses its own logging system.
     ///
@@ -264,6 +271,7 @@ struct TConfig
     void LoadToken();
     void LoadSpec();
     void LoadTimings();
+    void LoadProxyUrlAliasingRules();
 
     void Reset();