arunpandianp commented on code in PR #38814:
URL: https://github.com/apache/beam/pull/38814#discussion_r3497998850
##########
runners/google-cloud-dataflow-java/worker/src/main/java/org/apache/beam/runners/dataflow/worker/StreamingModeExecutionContext.java:
##########
@@ -759,18 +741,50 @@ public List<Windmill.WorkItemCommitRequest>
getWorkItemCommits() {
return commits;
}
- public Map<Long, Pair<Instant, Runnable>> getAccumulatedCallbacks() {
- return accumulatedCallbacks;
+ // Returns list of Work that was executed in the bundle
+ public List<Work> getExecutedWorks() {
+ return executedWorks;
}
+ // Returns finalization callbacks recorded during the bundle execution
+ public Map<Long, Pair<Instant, Runnable>> getFinalizationCallbacks() {
+ return finalizationCallbacks;
+ }
+
+ // Returns the current key being processed or null if an unkeyed stage.
public @Nullable Object getKey() {
return key;
}
+ // Returns the current Work being processed.
public Work getWork() {
return checkStateNotNull(work);
}
+ // Returns the serialized windmill key for the current Work
+ public @Nullable ByteString getSerializedKey() {
+ return work == null ? null : work.getWorkItem().getKey();
+ }
+
+ // Returns the serialized windmill key for the current Work
+ public WindmillComputationKey getComputationKey() {
+ return checkStateNotNull(computationKey);
+ }
+
+ // Returns the windmill work token for the current Work
+ public long getWorkToken() {
+ return getWorkItem().getWorkToken();
+ }
+
+ // Returns the windmill WorkItem proto for the current Work
+ public Windmill.WorkItem getWorkItem() {
+ return checkStateNotNull(
Review Comment:
done.
--
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]
