acrites commented on code in PR #40951:
URL: https://github.com/apache/beam/pull/40951#discussion_r4448963544
##########
runners/google-cloud-dataflow-java/worker/src/main/java/org/apache/beam/runners/dataflow/worker/SimpleParDoFn.java:
##########
@@ -209,14 +211,20 @@ public void abort() throws Exception {
helpers.abort();
}
+ /** Returns the {@link DoFnInfo} used by this {@link SimpleParDoFn}. */
+ @Override
+ public DoFnInfo<?, ?> getDoFnInfo() throws Exception {
+ return helpers.doFnInstanceManager.peek();
Review Comment:
With this new helper method, I think it makes sense to just remove the
getActiveDoFnInfo() method entirely. It's only used in tests anyway and those
tests will fail if it doesn't return the activeFnInfo.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
