UdsTest.java

package redis.clients.jedis;

import java.io.File;
import java.io.IOException;
import java.net.Socket;

import io.redis.test.annotations.ConditionalOnEnv;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.RegisterExtension;
import org.newsclub.net.unix.AFUNIXSocket;
import org.newsclub.net.unix.AFUNIXSocketAddress;
import redis.clients.jedis.exceptions.JedisConnectionException;
import redis.clients.jedis.providers.PooledConnectionProvider;
import redis.clients.jedis.util.EnvCondition;
import redis.clients.jedis.util.TestEnvUtil;

import static org.junit.jupiter.api.Assertions.assertEquals;

@ConditionalOnEnv(value = TestEnvUtil.ENV_OSS_SOURCE, enabled = true)
public class UdsTest {

  @RegisterExtension
  public static EnvCondition envCondition = new EnvCondition();

  @Test
  public void jedisConnectsToUds() {
    try (Jedis jedis = new Jedis(new UdsJedisSocketFactory())) {
      assertEquals("PONG", jedis.ping());
    }
  }

  @Test
  public void jedisConnectsToUdsResp3() {
    try (Jedis jedis = new Jedis(new UdsJedisSocketFactory(),
        DefaultJedisClientConfig.builder().resp3().build())) {
      assertEquals("PONG", jedis.ping());
    }
  }

  @Test
  public void redisClientConnectsToUds() {
    JedisSocketFactory socketFactory = new UdsJedisSocketFactory();
    ConnectionFactory connFactory = new ConnectionFactory(socketFactory,
        DefaultJedisClientConfig.builder().build());
    PooledConnectionProvider provider = new PooledConnectionProvider(connFactory);

    try (RedisClient client = RedisClient.builder().connectionProvider(provider).build()) {
      assertEquals("PONG", client.ping());
    }
  }

  @Test
  public void redisClientConnectsToUdsResp3() {
    JedisSocketFactory socketFactory = new UdsJedisSocketFactory();
    JedisClientConfig clientConfig = DefaultJedisClientConfig.builder().resp3().build();
    ConnectionFactory connFactory = new ConnectionFactory(socketFactory, clientConfig);
    PooledConnectionProvider provider = new PooledConnectionProvider(connFactory);

    try (RedisClient client = RedisClient.builder().connectionProvider(provider)
        .clientConfig(clientConfig).build()) {
      assertEquals("PONG", client.ping());
    }
  }

  private static class UdsJedisSocketFactory implements JedisSocketFactory {

    private static final File UDS_SOCKET = new File("/tmp/redis_uds.sock");

    @Override
    public Socket createSocket() throws JedisConnectionException {
      try {
        Socket socket = AFUNIXSocket.newStrictInstance();
        socket.connect(new AFUNIXSocketAddress(UDS_SOCKET), Protocol.DEFAULT_TIMEOUT);
        return socket;
      } catch (IOException ioe) {
        throw new JedisConnectionException("Failed to create UDS connection.", ioe);
      }
    }
  }
}