mxm commented on a change in pull request #12551:
URL: https://github.com/apache/beam/pull/12551#discussion_r470623744
##########
File path:
runners/flink/src/main/java/org/apache/beam/runners/flink/translation/wrappers/streaming/DoFnOperator.java
##########
@@ -1226,7 +1227,7 @@ public TimerInternals timerInternals() {
* fire time of the timer. Used for calculating the output watermark hold.
This avoids fetching
* timer data from the state backend which is expensive if done for each
timer.
*/
- private final PriorityQueue<Long> outputTimestampQueue;
+ private final TreeMultiset<Long> outputTimestamps = TreeMultiset.create();
Review comment:
We could also consider making this even more efficient by using a
`TreeMap<Long, Integer>` where the key is the output timestamp and the value
the number of timers which have set it, similar to how it's done in
`FlinkStateInternals` for the watermark holds.
Further, we could remove `outputTimestamps` entirely and simply use
`stateInternals.addWatermarkHoldUsage(output_timestamp)` and
`stateInternals.removeWatermarkHoldUsage(output_timestamp)`.
----------------------------------------------------------------
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.
For queries about this service, please contact Infrastructure at:
[email protected]
