org.apache.beam.runners.flink.translation.wrappers.streaming

Class WorkItemKeySelector<K,V>

java.lang.Object

org.apache.beam.runners.flink.translation.wrappers.streaming.WorkItemKeySelector<K,V>

All Implemented Interfaces:

java.io.Serializable, org.apache.flink.api.common.functions.Function, org.apache.flink.api.java.functions.KeySelector<org.apache.beam.sdk.util.WindowedValue<org.apache.beam.runners.core.KeyedWorkItem<K,V>>,java.nio.ByteBuffer>, org.apache.flink.api.java.typeutils.ResultTypeQueryable<java.nio.ByteBuffer>

public class WorkItemKeySelector<K,V>
extends java.lang.Object
implements org.apache.flink.api.java.functions.KeySelector<org.apache.beam.sdk.util.WindowedValue<org.apache.beam.runners.core.KeyedWorkItem<K,V>>,java.nio.ByteBuffer>, org.apache.flink.api.java.typeutils.ResultTypeQueryable<java.nio.ByteBuffer>

KeySelector that retrieves a key from a KeyedWorkItem. This will return the key as encoded by the provided Coder in a ByteBuffer. This ensures that all key comparisons/hashing happen on the encoded form.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
WorkItemKeySelector(org.apache.beam.sdk.coders.Coder<K> keyCoder, org.apache.beam.runners.core.construction.SerializablePipelineOptions pipelineOptions)

Method Summary

Modifier and Type	Method and Description
java.nio.ByteBuffer	getKey(org.apache.beam.sdk.util.WindowedValue<org.apache.beam.runners.core.KeyedWorkItem<K,V>> value)
org.apache.flink.api.common.typeinfo.TypeInformation<java.nio.ByteBuffer>	getProducedType()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

WorkItemKeySelector

public WorkItemKeySelector(org.apache.beam.sdk.coders.Coder<K> keyCoder,
                           org.apache.beam.runners.core.construction.SerializablePipelineOptions pipelineOptions)

Method Detail

getKey

public java.nio.ByteBuffer getKey(org.apache.beam.sdk.util.WindowedValue<org.apache.beam.runners.core.KeyedWorkItem<K,V>> value)
                           throws java.lang.Exception

Specified by:

getKey in interface org.apache.flink.api.java.functions.KeySelector<org.apache.beam.sdk.util.WindowedValue<org.apache.beam.runners.core.KeyedWorkItem<K,V>>,java.nio.ByteBuffer>

Throws:

java.lang.Exception

getProducedType

public org.apache.flink.api.common.typeinfo.TypeInformation<java.nio.ByteBuffer> getProducedType()

Specified by:

getProducedType in interface org.apache.flink.api.java.typeutils.ResultTypeQueryable<java.nio.ByteBuffer>