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

Show more of this group Show more artifacts with this name
Show all versions of sort-connector-redis-v1.15

There is a newer version: 1.13.0

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.util.SafeEncoder;

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

    public final byte[] raw;

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

  private List params = new ArrayList();

  /**
   * Set weights.
   * @param weights weights.
   * @deprecated Use {@link #weightsByDouble(double...)} instead
   */
  @Deprecated
  public ZParams weights(final int... weights) {
    params.add(WEIGHTS.raw);
    for (final int weight : weights) {
      params.add(Protocol.toByteArray(weight));
    }

    return this;
  }

  /**
   * Set weights.
   * @param weights weights.
   */
  public ZParams weightsByDouble(final double... weights) {
    params.add(WEIGHTS.raw);
    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.raw);
    params.add(aggregate.raw);
    return this;
  }
}