org.apache.flink.api.common.functions

Interface Partitioner<K>

Type Parameters:

K - The type of the key to be partitioned.

All Superinterfaces:

Serializable

All Known Implementing Classes:

ALS.BlockIDPartitioner, FlinkMLTools.ModuloKeyPartitioner$, IdPartitioner, SplitDataProperties.SourcePartitionerMarker

@Public
public interface Partitioner<K>
extends Serializable

Function to implement a custom partition assignment for keys.

Method Summary

Modifier and Type	Method and Description
int	partition(K key, int numPartitions) Computes the partition for the given key.

Method Detail

partition

int partition(K key,
              int numPartitions)

Computes the partition for the given key.

Parameters:

key - The key.

numPartitions - The number of partitions to partition into.

Returns:

The partition index.