m-trieu commented on code in PR #29082:
URL: https://github.com/apache/beam/pull/29082#discussion_r1387287735
##########
runners/google-cloud-dataflow-java/worker/src/main/java/org/apache/beam/runners/dataflow/worker/streaming/ActiveWorkState.java:
##########
@@ -107,10 +140,26 @@ synchronized ActivateWorkResult
activateWorkForKey(ShardedKey shardedKey, Work w
return ActivateWorkResult.EXECUTE;
}
- // Ensure we don't already have this work token queued.
- for (Work queuedWork : workQueue) {
- if (queuedWork.getWorkItem().getWorkToken() ==
work.getWorkItem().getWorkToken()) {
+ // Check to see if we have this work token queued.
+ Iterator<Work> workIterator = workQueue.iterator();
+ while (workIterator.hasNext()) {
+ Work queuedWork = workIterator.next();
+ if (queuedWork.id().equals(work.id())) {
return ActivateWorkResult.DUPLICATE;
+ } else if (queuedWork.id().cacheToken() == work.id().cacheToken()) {
+ if (work.id().workToken() > queuedWork.id().workToken()) {
+ removeIfNotActive(queuedWork, workIterator, workQueue);
+ workQueue.addLast(work);
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]
