YT: Use FormatNetworkAddress to build address from IP

commit_hash:56e50f44274025f97877254c8ec020c4a425be68

1 files changed, 5 insertions, 2 deletions

diff --git a/yt/yt/core/http/helpers.cpp b/yt/yt/core/http/helpers.cpp
index e5bead9511b..5ed8355c6d4 100644
--- a/yt/yt/core/http/helpers.cpp
+++ b/yt/yt/core/http/helpers.cpp
@@ -12,6 +12,8 @@
 #include <yt/yt/core/json/json_parser.h>
 #include <yt/yt/core/json/config.h>
 
+#include <yt/yt/core/net/address.h>
+
 #include <yt/yt/core/ytree/fluent.h>
 
 #include <util/stream/buffer.h>
@@ -308,8 +310,9 @@ std::optional<TString> FindBalancerRealIP(const IRequestPtr& req)
     auto forwardedFor = headers->Find(XForwardedForYHeaderName);
     auto sourcePort = headers->Find(XSourcePortYHeaderName);
 
-    if (forwardedFor && sourcePort) {
-        return Format("[%v]:%v", *forwardedFor, *sourcePort);
+    int port = 0;
+    if (forwardedFor && sourcePort && TryIntFromString<10>(*sourcePort, port)) {
+        return NNet::FormatNetworkAddress(*forwardedFor, port);
     }
 
     return {};