diff --git a/yt/yt/core/rpc/dynamic_channel_pool.cpp b/yt/yt/core/rpc/dynamic_channel_pool.cpp index 3bb45c0548f1..f5d3918853cd 100644 --- a/yt/yt/core/rpc/dynamic_channel_pool.cpp +++ b/yt/yt/core/rpc/dynamic_channel_pool.cpp @@ -308,6 +308,7 @@ class TDynamicChannelPool::TImpl auto channel = owner->ChannelFactory_->CreateChannel(address); auto request = owner->PeerDiscovery_->Discover( channel, + address, owner->Config_->DiscoverTimeout, /*replyDelay*/ TDuration::Zero(), owner->ServiceName_); @@ -533,7 +534,12 @@ class TDynamicChannelPool::TImpl auto channel = owner->ChannelFactory_->CreateChannel(PeerAddress_); auto requestTimeout = peerPollingPeriod + owner->Config_->PeerPollingRequestTimeout; - auto req = owner->PeerDiscovery_->Discover(channel, requestTimeout, /*replyDelay*/ peerPollingPeriod, owner->ServiceName_); + auto req = owner->PeerDiscovery_->Discover( + channel, + PeerAddress_, + requestTimeout, + /*replyDelay*/ peerPollingPeriod, + owner->ServiceName_); YT_LOG_DEBUG("Polling peer (PollingPeriod: %v, RequestTimeout: %v)", peerPollingPeriod, requestTimeout); diff --git a/yt/yt/core/rpc/peer_discovery.cpp b/yt/yt/core/rpc/peer_discovery.cpp index 5b3d031cff2d..696051c419a8 100644 --- a/yt/yt/core/rpc/peer_discovery.cpp +++ b/yt/yt/core/rpc/peer_discovery.cpp @@ -20,6 +20,7 @@ class TDefaultPeerDiscovery TFuture Discover( IChannelPtr channel, + const TString& /*address*/, TDuration timeout, TDuration replyDelay, const std::string& serviceName) override diff --git a/yt/yt/core/rpc/peer_discovery.h b/yt/yt/core/rpc/peer_discovery.h index a7120e2b3391..6157196ad6a9 100644 --- a/yt/yt/core/rpc/peer_discovery.h +++ b/yt/yt/core/rpc/peer_discovery.h @@ -17,6 +17,7 @@ struct IPeerDiscovery { virtual TFuture Discover( IChannelPtr channel, + const TString& address, TDuration timeout, TDuration replyDelay, const std::string& serviceName) = 0;