io.ray.streaming.api.partition.impl.RoundRobinPartition Maven / Gradle / Ivy

package io.ray.streaming.api.partition.impl;

import io.ray.streaming.api.partition.Partition;

/**
 * Partition record to downstream tasks in a round-robin matter.
 *
 * @param  Type of the input record.
 */
public class RoundRobinPartition implements Partition {

  private int seq;
  private int[] partitions = new int[1];

  public RoundRobinPartition() {
    this.seq = 0;
  }

  @Override
  public int[] partition(T value, int numPartition) {
    seq = (seq + 1) % numPartition;
    partitions[0] = seq;
    return partitions;
  }
}