Signal/Signal-Server
1
0
Fork 0
mirror of https://github.com/signalapp/Signal-Server synced 2026-04-21 04:58:06 +01:00
Code Issues Packages Projects Releases Wiki Activity

Introduce a Lettuce-based fault-tolerant Redis cluster accessor.

Browse Source
This commit is contained in:
Jon Chambers
2020-06-06 11:24:11 -04:00
committed by Jon Chambers
parent ba6ac778fc
commit fe1054d58a
6 changed files with 189 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

86
service/src/test/java/org/whispersystems/textsecuregcm/redis/FaultTolerantRedisClusterTest.java Normal file
View File

@@ -0,0 +1,86 @@
package org.whispersystems.textsecuregcm.redis;
import io.github.resilience4j.circuitbreaker.CircuitBreakerOpenException;
import io.lettuce.core.RedisException;
import io.lettuce.core.cluster.RedisClusterClient;
import io.lettuce.core.cluster.api.StatefulRedisClusterConnection;
import io.lettuce.core.cluster.api.sync.RedisAdvancedClusterCommands;
import org.junit.Before;
import org.junit.Test;
import org.whispersystems.textsecuregcm.configuration.CircuitBreakerConfiguration;
import static org.junit.Assert.*;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
public class FaultTolerantRedisClusterTest {
private RedisAdvancedClusterCommands<String, String> clusterCommands;
private FaultTolerantRedisCluster faultTolerantCluster;
@SuppressWarnings("unchecked")
@Before
public void setUp() {
final RedisClusterClient clusterClient = mock(RedisClusterClient.class);
final StatefulRedisClusterConnection<String, String> clusterConnection = mock(StatefulRedisClusterConnection.class);
clusterCommands = mock(RedisAdvancedClusterCommands.class);
when(clusterClient.connect()).thenReturn(clusterConnection);
when(clusterConnection.sync()).thenReturn(clusterCommands);
final CircuitBreakerConfiguration breakerConfiguration = new CircuitBreakerConfiguration();
breakerConfiguration.setFailureRateThreshold(100);
breakerConfiguration.setRingBufferSizeInClosedState(1);
breakerConfiguration.setWaitDurationInOpenStateInSeconds(Integer.MAX_VALUE);
faultTolerantCluster = new FaultTolerantRedisCluster("test", clusterClient, breakerConfiguration);
}
@Test
public void testReadBreaker() {
when(clusterCommands.get(anyString()))
.thenReturn("value")
.thenThrow(new RedisException("Badness has ensued."));
assertEquals("value", faultTolerantCluster.withReadCluster(connection -> connection.sync().get("key")));
assertThrows(RedisException.class,
() -> faultTolerantCluster.withReadCluster(connection -> connection.sync().get("OH NO")));
assertThrows(CircuitBreakerOpenException.class,
() -> faultTolerantCluster.withReadCluster(connection -> connection.sync().get("OH NO")));
}
@Test
public void testReadsContinueWhileWriteBreakerOpen() {
when(clusterCommands.set(anyString(), anyString())).thenThrow(new RedisException("Badness has ensued."));
assertThrows(RedisException.class,
() -> faultTolerantCluster.useWriteCluster(connection -> connection.sync().set("OH", "NO")));
assertThrows(CircuitBreakerOpenException.class,
() -> faultTolerantCluster.useWriteCluster(connection -> connection.sync().set("OH", "NO")));
when(clusterCommands.get("key")).thenReturn("value");
assertEquals("value", faultTolerantCluster.withReadCluster(connection -> connection.sync().get("key")));
}
@Test
public void testWriteBreaker() {
when(clusterCommands.get(anyString()))
.thenReturn("value")
.thenThrow(new RedisException("Badness has ensued."));
assertEquals("value", faultTolerantCluster.withWriteCluster(connection -> connection.sync().get("key")));
assertThrows(RedisException.class,
() -> faultTolerantCluster.withWriteCluster(connection -> connection.sync().get("OH NO")));
assertThrows(CircuitBreakerOpenException.class,
() -> faultTolerantCluster.withWriteCluster(connection -> connection.sync().get("OH NO")));
}
}