WithKeys

Javadoc

Takes a PCollection<V> and produces a PCollection<KV<K, V>> by associating each input element with a key.

There are two versions of WithKeys, depending on how the key should be determined:

• WithKeys.of(SerializableFunction<V, K> fn) takes a function to compute the key from each value.
• WithKeys.of(K key) associates each value with the specified key.

Examples

Example

PCollection<String> words = Create.of("Hello", "World", "Beam", "is", "fun");
PCollection<KV<Integer, String>> lengthAndWord =
  words.apply(WithKeys.of(new SerializableFunction<String, Integer>() {
    @Override
    public Integer apply(String s) {
      return s.length();
    }
  });

Related transforms

• Keys for extracting the key of each component.
• Values for extracting the value of each element.
• KvSwap swaps key-value pair values.