在Flink中，当对键控流应用进程函数时，我如何访问键？

Question

在Flink中，我有一个键控流，我正在对它应用一个进程函数。

myDataStream
  .keyBy(new MyKeySelector())
  .process(new FooBarProcessFunction())

我的键选择器看起来像这样...

public class MyKeySelector implements KeySelector<FooBar, FooKey>

public FooKey getKey (FooBar value) {
   return new FooKey (value);
}

FooBarProcessFunction看起来像这样...

public class FooBarProcessFunction extends ProcessFunction<FooBar, TransformedFooBar> {

    public void processElement(FooBar newFooBar, Context ctx, Collector<FooBar> out) {
        //do something with newFooBar
        // *****but I also want to know the Key (FooKey) here***** 
    }
}

在FooBarProcessFunction中，我希望获得由MyKeySelector的getKey方法创建的键。那可行吗？

目前，我正在使用一种变通方法，其中我基本上在processElement函数中重新创建键。但如果我能避免这样做，那就太理想了。

Answer 1

如果您打开键控流窗口并应用ProcessWindowFunction ,似乎可以获得键。

Apache flink文档中有一些这样的例子。请注意，ProcessWindowFunction效率不高，应与ReduceFunction、AggregateFunction或FoldFunction组合使用。

希望这有帮助！

Answer 2

为了从进程函数访问密钥，您应该使用keyedprocessfunction

你的例子变成：

public class FooBarProcessFunction extends KeyedProcessFunction<FooKey, FooBar, TransformedFooBar> {

public void processElement(FooBar newFooBar, Context ctx, Collector<FooBar> out) {
    //do something with newFooBar
    // *****but I also want to know the Key (FooKey) here***** 
   ctx.getCurrentKey
}

}