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

Go to download

Show more of this group Show more artifacts with this name
Show all versions of streaming-api Show documentation

ray streaming api

There is a newer version: 1.10.0

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

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

/**
 * Partition the record by the key.
 *
 * @param  Type of the partition key.
 * @param  Type of the input record.
 */
public class KeyPartition implements Partition> {
  private int[] partitions = new int[1];

  @Override
  public int[] partition(KeyRecord keyRecord, int numPartition) {
    partitions[0] = Math.abs(keyRecord.getKey().hashCode() % numPartition);
    return partitions;
  }
}