Skip to content

Commit 5ccf4e6

Browse files
committed
move reconntion strategy to base
Signed-off-by: Shoham Elias <[email protected]>
1 parent ea51e46 commit 5ccf4e6

File tree

2 files changed

+27
-13
lines changed

2 files changed

+27
-13
lines changed

java/client/src/main/java/glide/managers/ConnectionManager.java

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -145,6 +145,19 @@ private ConnectionRequest.Builder setupConnectionRequestBuilderBaseConfiguration
145145
connectionRequestBuilder.setProtocolValue(configuration.getProtocol().ordinal());
146146
}
147147

148+
if (configuration.getReconnectStrategy() != null) {
149+
var reconnectionStrategyBuilder =
150+
ConnectionRequestOuterClass.ConnectionRetryStrategy.newBuilder()
151+
.setNumberOfRetries(configuration.getReconnectStrategy().getNumOfRetries())
152+
.setExponentBase(configuration.getReconnectStrategy().getExponentBase())
153+
.setFactor(configuration.getReconnectStrategy().getFactor());
154+
if (configuration.getReconnectStrategy().getJitterPercent() != null) {
155+
reconnectionStrategyBuilder.setJitterPercent(
156+
configuration.getReconnectStrategy().getJitterPercent());
157+
}
158+
connectionRequestBuilder.setConnectionRetryStrategy(reconnectionStrategyBuilder.build());
159+
}
160+
148161
return connectionRequestBuilder;
149162
}
150163

@@ -241,19 +254,6 @@ private ConnectionRequest.Builder setupConnectionRequestBuilderGlideClusterClien
241254
connectionRequestBuilder, configuration.getAdvancedConfiguration());
242255
}
243256

244-
if (configuration.getReconnectStrategy() != null) {
245-
var reconnectionStrategyBuilder =
246-
ConnectionRequestOuterClass.ConnectionRetryStrategy.newBuilder()
247-
.setNumberOfRetries(configuration.getReconnectStrategy().getNumOfRetries())
248-
.setExponentBase(configuration.getReconnectStrategy().getExponentBase())
249-
.setFactor(configuration.getReconnectStrategy().getFactor());
250-
if (configuration.getReconnectStrategy().getJitterPercent() != null) {
251-
reconnectionStrategyBuilder.setJitterPercent(
252-
configuration.getReconnectStrategy().getJitterPercent());
253-
}
254-
connectionRequestBuilder.setConnectionRetryStrategy(reconnectionStrategyBuilder.build());
255-
}
256-
257257
return connectionRequestBuilder;
258258
}
259259

java/client/src/test/java/glide/managers/ConnectionManagerTest.java

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -341,6 +341,7 @@ public void test_reconnect_strategy_to_protobuf() {
341341
GlideClientConfiguration glideClientConfiguration =
342342
GlideClientConfiguration.builder()
343343
.address(NodeAddress.builder().host(HOST).port(PORT).build())
344+
.useTLS(true)
344345
.reconnectStrategy(
345346
BackoffStrategy.builder()
346347
.numOfRetries(10)
@@ -351,6 +352,12 @@ public void test_reconnect_strategy_to_protobuf() {
351352
.build();
352353
ConnectionRequest expectedProtobufConnectionRequest =
353354
ConnectionRequest.newBuilder()
355+
.addAddresses(
356+
ConnectionRequestOuterClass.NodeAddress.newBuilder()
357+
.setHost(HOST)
358+
.setPort(PORT)
359+
.build())
360+
.setTlsMode(TlsMode.SecureTls)
354361
.setConnectionRetryStrategy(
355362
ConnectionRetryStrategy.newBuilder()
356363
.setNumberOfRetries(10)
@@ -378,6 +385,7 @@ public void test_reconnect_strategy_to_protobuf_cluster_client() {
378385
GlideClusterClientConfiguration glideClusterClientConfiguration =
379386
GlideClusterClientConfiguration.builder()
380387
.address(NodeAddress.builder().host(HOST).port(PORT).build())
388+
.useTLS(true)
381389
.reconnectStrategy(
382390
BackoffStrategy.builder()
383391
.numOfRetries(10)
@@ -388,6 +396,12 @@ public void test_reconnect_strategy_to_protobuf_cluster_client() {
388396
.build();
389397
ConnectionRequest expectedProtobufConnectionRequest =
390398
ConnectionRequest.newBuilder()
399+
.addAddresses(
400+
ConnectionRequestOuterClass.NodeAddress.newBuilder()
401+
.setHost(HOST)
402+
.setPort(PORT)
403+
.build())
404+
.setTlsMode(TlsMode.SecureTls)
391405
.setClusterModeEnabled(true)
392406
.setConnectionRetryStrategy(
393407
ConnectionRetryStrategy.newBuilder()

0 commit comments

Comments
 (0)