org.springframework.data.redis.connection.RedisClusterConnection Java Examples

@Override
public RedisConnection getConnection() {
  // support cluster connection
  RedisConnection connection = this.delegate.getConnection();
  if (connection instanceof RedisClusterConnection) {
    return new TracingRedisClusterConnection((RedisClusterConnection) connection,
        tracingConfiguration);
  }
  return new TracingRedisConnection(connection, tracingConfiguration);
}
 

@Override
public RedisClusterConnection getClusterConnection() {
    if (!redisson.getConfig().isClusterConfig()) {
        throw new InvalidDataAccessResourceUsageException("Redisson is not in Cluster mode");
    }
    return new RedissonClusterConnection(redisson);
}
 

@Override
public RedisClusterConnection getClusterConnection() {
    if (!redisson.getConfig().isClusterConfig()) {
        throw new InvalidDataAccessResourceUsageException("Redisson is not in Cluster mode");
    }
    return new RedissonClusterConnection(redisson);
}
 

@Override
public RedisClusterConnection getClusterConnection() {
    if (!redisson.getConfig().isClusterConfig()) {
        throw new InvalidDataAccessResourceUsageException("Redisson is not in Cluster mode");
    }
    return new RedissonClusterConnection(redisson);
}
 

@Override
public RedisClusterConnection getClusterConnection() {
    if (!redisson.getConfig().isClusterConfig()) {
        throw new InvalidDataAccessResourceUsageException("Redisson is not in Cluster mode");
    }
    return new RedissonClusterConnection(redisson);
}
 

@Bean
public RedisLimit build() {
    RedisClusterConnection clusterConnection = jedisConnectionFactory.getClusterConnection();
    JedisCluster jedisCluster = (JedisCluster) clusterConnection.getNativeConnection();
    RedisLimit redisLimit = new RedisLimit.Builder<>(jedisCluster)
            .limit(limit)
            .build();

    return redisLimit;
}
 

@Test
public void delegatesClusterConnectionIfCluster() {
  RedisConnectionFactory connectionFactory = mock(RedisConnectionFactory.class);
  when(connectionFactory.getConnection()).thenReturn(mock(RedisClusterConnection.class));
  TracingRedisConnectionFactory tracingConnectionFactory = new TracingRedisConnectionFactory(
      connectionFactory,
      new TracingConfiguration.Builder(tracer).traceWithActiveSpanOnly(false).build());
  assertTrue(tracingConnectionFactory.getConnection() instanceof RedisClusterConnection);
}
 

private Object getConnection() {
    Object connection;
    if (type == RedisToolsConstant.SINGLE) {
        RedisConnection redisConnection = jedisConnectionFactory.getConnection();
        connection = redisConnection.getNativeConnection();
    } else {
        RedisClusterConnection clusterConnection = jedisConnectionFactory.getClusterConnection();
        connection = clusterConnection.getNativeConnection();
    }
    return connection;
}
 

@Test
public void delegatesClusterConnectionIfCluster() {
  RedisConnectionFactory connectionFactory = mock(RedisConnectionFactory.class);
  when(connectionFactory.getConnection()).thenReturn(mock(RedisClusterConnection.class));
  TracingRedisConnectionFactory tracingConnectionFactory = new TracingRedisConnectionFactory(
      connectionFactory,
      new TracingConfiguration.Builder(tracer).traceWithActiveSpanOnly(false).build());
  assertTrue(tracingConnectionFactory.getConnection() instanceof RedisClusterConnection);
}
 

@Override
public RedisConnection getConnection() {
  // support cluster connection
  RedisConnection connection = this.delegate.getConnection();
  if (connection instanceof RedisClusterConnection) {
    return new TracingRedisClusterConnection((RedisClusterConnection) connection,
        tracingConfiguration);
  }
  return new TracingRedisConnection(connection, tracingConfiguration);
}
 

/**
 * Intercepts calls to {@link RedisConnectionFactory#getClusterConnection()} (and related),
 * wrapping the outcome in a {@link TracingRedisClusterConnection}
 *
 * @param pjp the intercepted join point
 * @return a new {@link TracingRedisClusterConnection} wrapping the result of the joint point
 */
@Around("getClusterConnection() && connectionFactory()")
public Object aroundGetClusterConnection(final ProceedingJoinPoint pjp) throws Throwable {
  final RedisClusterConnection clusterConnection = (RedisClusterConnection) pjp.proceed();

  final String prefixOperationName = this.properties.getPrefixOperationName();
  final TracingConfiguration tracingConfiguration = new TracingConfiguration.Builder(tracer)
      .withSpanNameProvider(RedisSpanNameProvider.PREFIX_OPERATION_NAME(prefixOperationName))
      .build();

  return new TracingRedisClusterConnection(clusterConnection, tracingConfiguration);
}
 

@Test
public void connectionIsTracingRedisClusterConnection() {
  RedisClusterConnection connection = connectionFactory.getClusterConnection();
  verify(delegate).getClusterConnection();
  assertTrue(connection instanceof TracingRedisClusterConnection);
}
 

@Override
protected RedisClusterConnection getConnection() {
  return connectionFactory.getClusterConnection();
}
 

public TracingRedisClusterConnection(RedisClusterConnection connection,
    TracingConfiguration tracingConfiguration) {
  super(connection, tracingConfiguration);
  this.connection = connection;
  this.helper = new TracingHelper(tracingConfiguration);
}
 

@Override
protected RedisClusterConnection mockRedisConnection() {
  return mockRedisClusterConnection;
}
 

@Override
public RedisClusterConnection getClusterConnection() {
  return new TracingRedisClusterConnection(delegate.getClusterConnection(), tracingConfiguration);
}
 

@Bean
public RedisClusterConnection mockRedisClusterConnection() {
  return mock(RedisClusterConnection.class);
}
 

@Override
protected RedisClusterConnection mockRedisConnection() {
  return mockRedisClusterConnection;
}
 

@Override
protected RedisClusterConnection getConnection() {
  return connectionFactory.getClusterConnection();
}
 

@Test
public void connectionIsTracingRedisClusterConnection() {
  RedisClusterConnection connection = connectionFactory.getClusterConnection();
  verify(delegate).getClusterConnection();
  assertTrue(connection instanceof TracingRedisClusterConnection);
}
 

@Bean
public RedisClusterConnection mockRedisClusterConnection() {
  return mock(RedisClusterConnection.class);
}
 

@Override
public RedisClusterConnection getClusterConnection() {
  return new TracingRedisClusterConnection(delegate.getClusterConnection(), tracingConfiguration);
}
 

public TracingRedisClusterConnection(RedisClusterConnection connection,
    TracingConfiguration tracingConfiguration) {
  super(connection, tracingConfiguration);
  this.connection = connection;
  this.helper = new TracingHelper(tracingConfiguration);
}
 

@Override
public RedisClusterConnection getClusterConnection() {
    System.out.println("getClusterConnection");
    return redisConnectionFactory.getClusterConnection();
}