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_r264219129
##########
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) {
+ assert Thread.holdsLock(buffers);
+
+ if (lastBufferAvailable) {
+ return buffersInBacklog;
+ } else {
+ return Math.max(buffersInBacklog - 1, 0);
+ }
+ }
+
/**
* Decreases the number of non-event buffers by one after fetching a
non-event
* buffer from this subpartition (for access by the subpartition views).
- *
- * @return backlog after the operation
*/
- public int decreaseBuffersInBacklog(Buffer buffer) {
+ public void decreaseBuffersInBacklogUnsafe(boolean isBuffer) {
Review comment:
Why did you reverted the `Unsafe` relation here?
`decreaseBuffersInBacklogUnsafe(boolean isBuffer)` was "unsafe" because it was
not synchronized and it was thus marked `protected` compared to the "safe"
version that did the synchronization.
To keep the same convention `getBuffersInBacklog(boolean
lastBufferAvailable)` should be renamed to `getBuffersInBacklogUnsafe(boolean
lastBufferAvailable)`
----------------------------------------------------------------
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]
With regards,
Apache Git Services
