Class GridRedisClient

java/src/org/openqa/selenium/redis/GridRedisClient.java:44–241

import org.redisson.config.SingleServerConfig;

public class GridRedisClient implements Closeable {
  private final RedisClient client;
  private final RedissonClient redissonClient;
  private final StatefulRedisConnection<String, String> connection;
  private final RLiveObjectService service;

  public GridRedisClient(URI serverUri) {
    client = RedisClient.create(RedisURI.create(serverUri));
    connection = client.connect();

    Config redissonConfig = new Config();
    redissonConfig.setNettyThreads(0);
    SingleServerConfig singleServerConfig = redissonConfig.useSingleServer();
    singleServerConfig.setAddress(serverUri.toString());
    singleServerConfig.setConnectionMinimumIdleSize(5);
    singleServerConfig.setConnectionPoolSize(5);
    redissonClient = Redisson.create(redissonConfig);
    service = redissonClient.getLiveObjectService();
  }

  public StatefulRedisConnection<String, String> getConnection() {
    return connection;
  }

  public void mset(Map<String, String> map) {
    connection.sync().mset(map);
  }

  public List<KeyValue<String, String>> mget(String... keys) {
    return connection.sync().mget(keys);
  }

  @Nullable
  public String get(String key) {
    return connection.sync().get(key);
  }

  public List<String> getKeysByPattern(String pattern) {
    return connection.sync().keys(pattern);
  }

  public void set(String key, String value) {
    connection.sync().set(key, value);
  }

  public void setWithTtl(String key, String value, long ttlMillis) {
    connection.sync().set(key, value, SetArgs.Builder.px(ttlMillis));
  }

  /** Returns true if the key was set (caller won the race), false if already present. */
  public boolean setIfAbsent(String key, String value, long ttlMillis) {
    String result = connection.sync().set(key, value, SetArgs.Builder.nx().px(ttlMillis));
    return "OK".equals(result);
  }

  /** Returns true if the key was set (no TTL variant). */
  public boolean setIfAbsent(String key, String value) {
    String result = connection.sync().set(key, value, SetArgs.Builder.nx());