diff --git a/service/src/main/java/org/whispersystems/textsecuregcm/redis/FaultTolerantPubSubConnection.java b/service/src/main/java/org/whispersystems/textsecuregcm/redis/FaultTolerantPubSubConnection.java index 9f4122a88..aa5c45710 100644 --- a/service/src/main/java/org/whispersystems/textsecuregcm/redis/FaultTolerantPubSubConnection.java +++ b/service/src/main/java/org/whispersystems/textsecuregcm/redis/FaultTolerantPubSubConnection.java @@ -5,7 +5,6 @@ import com.codahale.metrics.SharedMetricRegistries; import com.codahale.metrics.Timer; import io.github.resilience4j.circuitbreaker.CircuitBreaker; import io.lettuce.core.cluster.pubsub.StatefulRedisClusterPubSubConnection; -import org.whispersystems.textsecuregcm.configuration.CircuitBreakerConfiguration; import org.whispersystems.textsecuregcm.util.CircuitBreakerUtil; import org.whispersystems.textsecuregcm.util.Constants; @@ -21,9 +20,9 @@ public class FaultTolerantPubSubConnection { private final Timer executeTimer; - public FaultTolerantPubSubConnection(final String name, final StatefulRedisClusterPubSubConnection pubSubConnection, final CircuitBreakerConfiguration circuitBreakerConfiguration) { + public FaultTolerantPubSubConnection(final String name, final StatefulRedisClusterPubSubConnection pubSubConnection, final CircuitBreaker circuitBreaker) { this.pubSubConnection = pubSubConnection; - this.circuitBreaker = CircuitBreaker.of(name + "-pubsub", circuitBreakerConfiguration.toCircuitBreakerConfig()); + this.circuitBreaker = circuitBreaker; CircuitBreakerUtil.registerMetrics(SharedMetricRegistries.getOrCreate(Constants.METRICS_NAME), this.circuitBreaker, diff --git a/service/src/main/java/org/whispersystems/textsecuregcm/redis/FaultTolerantRedisCluster.java b/service/src/main/java/org/whispersystems/textsecuregcm/redis/FaultTolerantRedisCluster.java index a0db63aaf..56e578050 100644 --- a/service/src/main/java/org/whispersystems/textsecuregcm/redis/FaultTolerantRedisCluster.java +++ b/service/src/main/java/org/whispersystems/textsecuregcm/redis/FaultTolerantRedisCluster.java @@ -41,7 +41,6 @@ public class FaultTolerantRedisCluster { private final List> pubSubConnections = new ArrayList<>(); - private final CircuitBreakerConfiguration circuitBreakerConfiguration; private final CircuitBreaker circuitBreaker; private final Timer executeTimer; @@ -65,8 +64,7 @@ public class FaultTolerantRedisCluster { this.stringConnection = clusterClient.connect(); this.binaryConnection = clusterClient.connect(ByteArrayCodec.INSTANCE); - this.circuitBreakerConfiguration = circuitBreakerConfiguration; - this.circuitBreaker = CircuitBreaker.of(name, circuitBreakerConfiguration.toCircuitBreakerConfig()); + this.circuitBreaker = CircuitBreaker.of(name, circuitBreakerConfiguration.toCircuitBreakerConfig()); CircuitBreakerUtil.registerMetrics(SharedMetricRegistries.getOrCreate(Constants.METRICS_NAME), circuitBreaker, @@ -144,6 +142,6 @@ public class FaultTolerantRedisCluster { final StatefulRedisClusterPubSubConnection pubSubConnection = clusterClient.connectPubSub(); pubSubConnections.add(pubSubConnection); - return new FaultTolerantPubSubConnection<>(name, pubSubConnection, circuitBreakerConfiguration); + return new FaultTolerantPubSubConnection<>(name, pubSubConnection, circuitBreaker); } }