redis.clients.jedis.ZParams Maven / Gradle / Ivy

package redis.clients.jedis;

import static redis.clients.jedis.Protocol.Keyword.AGGREGATE;
import static redis.clients.jedis.Protocol.Keyword.WEIGHTS;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.List;

import redis.clients.jedis.util.SafeEncoder;

public class ZParams {
  public enum Aggregate {
    SUM, MIN, MAX;

    /**
     * @deprecated This will be private in future. Use {@link #getRaw()}.
     */
    @Deprecated
    public final byte[] raw;

    Aggregate() {
      raw = SafeEncoder.encode(name());
    }

    public byte[] getRaw() {
      return raw;
    }
  }

  private final List params = new ArrayList<>();

  /**
   * Set weights.
   * @param weights weights.
   * @return
   */
  public ZParams weights(final double... weights) {
    params.add(WEIGHTS.getRaw());
    for (final double weight : weights) {
      params.add(Protocol.toByteArray(weight));
    }

    return this;
  }

  public Collection getParams() {
    return Collections.unmodifiableCollection(params);
  }

  public ZParams aggregate(final Aggregate aggregate) {
    params.add(AGGREGATE.getRaw());
    params.add(aggregate.raw);
    return this;
  }
}