diff --git a/core/src/main/java/tech/ydb/core/impl/BaseGrpcTransport.java b/core/src/main/java/tech/ydb/core/impl/BaseGrpcTransport.java index ab41dfd78..08b747930 100644 --- a/core/src/main/java/tech/ydb/core/impl/BaseGrpcTransport.java +++ b/core/src/main/java/tech/ydb/core/impl/BaseGrpcTransport.java @@ -43,7 +43,7 @@ public abstract class BaseGrpcTransport implements GrpcTransport { protected volatile boolean shutdown = false; abstract AuthCallOptions getAuthCallOptions(); - abstract GrpcChannel getChannel(GrpcRequestSettings settings); + protected abstract GrpcChannel getChannel(GrpcRequestSettings settings); abstract void updateChannelStatus(GrpcChannel channel, io.grpc.Status status); @Override diff --git a/core/src/main/java/tech/ydb/core/impl/FixedCallOptionsTransport.java b/core/src/main/java/tech/ydb/core/impl/FixedCallOptionsTransport.java index b971be86a..43c832929 100644 --- a/core/src/main/java/tech/ydb/core/impl/FixedCallOptionsTransport.java +++ b/core/src/main/java/tech/ydb/core/impl/FixedCallOptionsTransport.java @@ -58,7 +58,7 @@ public AuthCallOptions getAuthCallOptions() { } @Override - GrpcChannel getChannel(GrpcRequestSettings settings) { + protected GrpcChannel getChannel(GrpcRequestSettings settings) { return channel; } diff --git a/core/src/main/java/tech/ydb/core/impl/MultiChannelTransport.java b/core/src/main/java/tech/ydb/core/impl/MultiChannelTransport.java index cc4579821..fd6192cc7 100644 --- a/core/src/main/java/tech/ydb/core/impl/MultiChannelTransport.java +++ b/core/src/main/java/tech/ydb/core/impl/MultiChannelTransport.java @@ -95,7 +95,7 @@ public AuthCallOptions getAuthCallOptions() { } @Override - GrpcChannel getChannel(GrpcRequestSettings settings) { + protected GrpcChannel getChannel(GrpcRequestSettings settings) { EndpointRecord endpoint = endpointPool.getEndpoint(null); return channelPool.getChannel(endpoint); } diff --git a/core/src/main/java/tech/ydb/core/impl/SingleChannelTransport.java b/core/src/main/java/tech/ydb/core/impl/SingleChannelTransport.java index eeaf55a59..f59be47e1 100644 --- a/core/src/main/java/tech/ydb/core/impl/SingleChannelTransport.java +++ b/core/src/main/java/tech/ydb/core/impl/SingleChannelTransport.java @@ -70,7 +70,7 @@ public AuthCallOptions getAuthCallOptions() { } @Override - GrpcChannel getChannel(GrpcRequestSettings settings) { + protected GrpcChannel getChannel(GrpcRequestSettings settings) { return channel; } diff --git a/core/src/main/java/tech/ydb/core/impl/YdbTransportImpl.java b/core/src/main/java/tech/ydb/core/impl/YdbTransportImpl.java index 7e4186c27..481209163 100644 --- a/core/src/main/java/tech/ydb/core/impl/YdbTransportImpl.java +++ b/core/src/main/java/tech/ydb/core/impl/YdbTransportImpl.java @@ -154,7 +154,7 @@ public AuthCallOptions getAuthCallOptions() { } @Override - GrpcChannel getChannel(GrpcRequestSettings settings) { + protected GrpcChannel getChannel(GrpcRequestSettings settings) { EndpointRecord endpoint = endpointPool.getEndpoint(settings.getPreferredNodeID()); return channelPool.getChannel(endpoint); }