pnowojski commented on a change in pull request #7911: [FLINK-11082][network] Fix the logic of getting backlog in sub partition URL: https://github.com/apache/flink/pull/7911#discussion_r264637060
########## File path: flink-runtime/src/main/java/org/apache/flink/runtime/io/network/partition/ResultSubpartition.java ########## @@ -116,52 +115,58 @@ protected Throwable getFailureCause() { public abstract boolean isReleased(); - /** - * Gets the number of non-event buffers in this subpartition. - * - * <p><strong>Beware:</strong> This method should only be used in tests in non-concurrent access - * scenarios since it does not make any concurrency guarantees. - */ - @VisibleForTesting - public int getBuffersInBacklog() { - return buffersInBacklog; - } - /** * Makes a best effort to get the current size of the queue. * This method must not acquire locks or interfere with the task and network threads in * any way. */ public abstract int unsynchronizedGetNumberOfQueuedBuffers(); + /** + * Gets the number of non-event buffers in this subpartition. + */ + public abstract int getBuffersInBacklog(); + + /** + * @param lastBufferAvailable whether the last buffer in this subpartition is available for consumption + * @return the number of non-event buffers in this subpartition + */ + protected int getBuffersInBacklog(boolean lastBufferAvailable) { Review comment: Ok, this gets a little bit too complicated for my taste as well, but if you don't have idea how to fix it, lets keep it as it is :( ---------------------------------------------------------------- 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: us...@infra.apache.org With regards, Apache Git Services