redis.clients.jedis.params.FailoverParams Maven / Gradle / Ivy

package redis.clients.jedis.params;

import java.util.ArrayList;
import java.util.List;
import redis.clients.jedis.HostAndPort;
import redis.clients.jedis.Protocol;
import redis.clients.jedis.Protocol.Keyword;
import redis.clients.jedis.util.SafeEncoder;

public class FailoverParams {

  private HostAndPort to;

  private boolean force;

  private Long timeout;

  public static FailoverParams failoverParams() {
    return new FailoverParams();
  }

  public FailoverParams to(String host, int port) {
    return to(new HostAndPort(host, port));
  }

  public FailoverParams to(HostAndPort to) {
    this.to = to;
    return this;
  }

  /**
   * WARNING: FORCE option can be used only if both TO and TIMEOUT options are specified.
   */
  public FailoverParams force() {
    this.force = true;
    return this;
  }

  public FailoverParams timeout(long timeout) {
    this.timeout = timeout;
    return this;
  }

  public byte[][] getByteParams() {
    List params = new ArrayList<>();

    if (to != null) {
      params.add(Keyword.TO.getRaw());
      params.add(SafeEncoder.encode(to.getHost()));
      params.add(Protocol.toByteArray(to.getPort()));

    }

    if (force) {
      if (to == null || timeout == null) {
        throw new IllegalStateException("ERR FAILOVER with force option requires both a timeout and target HOST and IP.");
      }
      params.add(Keyword.FORCE.getRaw());
    }

    if (timeout != null) {
      params.add(Keyword.TIMEOUT.getRaw());
      params.add(Protocol.toByteArray(timeout));
    }

    return params.toArray(new byte[params.size()][]);
  }
}