pyflink.datastream.data_stream.CachedDataStream.partition_custom

CachedDataStream.partition_custom(partitioner: Callable | Partitioner, key_selector: Callable | KeySelector) → DataStream

Partitions a DataStream on the key returned by the selector, using a custom partitioner. This method takes the key selector to get the key to partition on, and a partitioner that accepts the key type.

Note that this method works only on single field keys, i.e. the selector cannot return tuples of fields.

Parameters:

• partitioner – The partitioner to assign partitions to keys.

• key_selector – The KeySelector with which the DataStream is partitioned.

Returns:

The partitioned DataStream.