io.ray.streaming.api.partition.impl.BroadcastPartition 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 java.util.stream.IntStream;

/** Broadcast the record to all downstream partitions. */
public class BroadcastPartition implements Partition {

  private int[] partitions = new int[0];

  public BroadcastPartition() {}

  @Override
  public int[] partition(T value, int numPartition) {
    if (partitions.length != numPartition) {
      partitions = IntStream.rangeClosed(0, numPartition - 1).toArray();
    }
    return partitions;
  }
}