com.google.sitebricks.persist.redis.JedisPersister Maven / Gradle / Ivy

Go to download
package com.google.sitebricks.persist.redis;

import com.google.sitebricks.persist.EntityStore;
import com.google.sitebricks.persist.Persister;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.JedisPoolConfig;
import redis.clients.jedis.Protocol;

import java.net.URI;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * @author [email protected] (Dhanji R. Prasanna)
 */
class JedisPersister extends Persister {
  private static final Pattern HOST_PORT_REGEX = Pattern.compile("(.*):(\\d+)");

  private final JedisPoolConfig config;
  private final String host;
  private final String password;
  private final int port;

  private volatile JedisPool pool;

  private static final EntityStore.EntityTransaction SINK = new EntityStore.EntityTransaction() {
    @Override
    public void commit() {}

    @Override
    public void rollback() {}
  };

  JedisPersister(JedisPoolConfig config, String hostOrUri) {
    this.config = config;
    String host = hostOrUri;
    int port = 6379;    // default redis port.
    String password = null;

    // Discover password if necessary.
    URI uri = URI.create(hostOrUri);
    // If using the redis protocol, try to discover password
    if ("redis".equals(uri.getScheme())) {
      host = uri.getHost();
      port = uri.getPort();

      if (uri.getUserInfo() != null) {
        String[] split = uri.getUserInfo().split(":", 2);
        if (split.length > 1)
          password = split[1];
      }
    } else {
      Matcher matcher = HOST_PORT_REGEX.matcher(hostOrUri);
      if (matcher.matches()) {
        host = matcher.group(1);
        port = Integer.parseInt(matcher.group(2));
      }
    }

    this.host = host;
    this.port = port;
    this.password = password;
  }

  @Override
  public synchronized void start() {
    if (password == null)
      pool = new JedisPool(config, host, port);
    else
      pool = new JedisPool(config, host, port, Protocol.DEFAULT_TIMEOUT, password);
  }

  @Override
  public synchronized void shutdown() {
    pool.destroy();
    pool = null;
  }

  JedisPool getPool() {
    return pool;
  }

  @Override
  protected EntityStore beginWork() {
    return new JedisEntityStore(pool.getResource());
  }

  @Override
  protected void endWork(EntityStore store, boolean commit /* ignored in Redis */) {
    pool.returnResource((Jedis) store.delegate());
  }

  @Override
  protected EntityStore.EntityTransaction beginTransaction() {
    return SINK;
  }
}