[ https://issues.apache.org/jira/browse/FLINK-7406?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=16184442#comment-16184442 ]
ASF GitHub Bot commented on FLINK-7406: --------------------------------------- Github user NicoK commented on a diff in the pull request: https://github.com/apache/flink/pull/4509#discussion_r141673902 --- Diff: flink-runtime/src/main/java/org/apache/flink/runtime/io/network/partition/consumer/RemoteInputChannel.java --- @@ -390,7 +390,63 @@ public BufferProvider getBufferProvider() throws IOException { return inputGate.getBufferProvider(); } - public void onBuffer(Buffer buffer, int sequenceNumber) { + /** + * Requests buffer from input channel directly for receiving network data. + * It should always return an available buffer in credit-based mode. + * + * @return The available buffer. + */ + public Buffer requestBuffer() { + synchronized (availableBuffers) { + return availableBuffers.poll(); + } + } + + /** + * Receives the backlog from producer's buffer response. If the number of available + * buffers is less than the backlog length, it will request floating buffers from buffer + * pool, and then notify unannounced credits to the producer. + * + * @param backlog The number of unsent buffers in the producer's sub partition. + */ + private void onSenderBacklog(int backlog) { + int numRequestedBuffers = 0; + + synchronized (availableBuffers) { + // Important: the isReleased check should be inside the synchronized block. + if (!isReleased.get()) { + senderBacklog.set(backlog); + + while (senderBacklog.get() > availableBuffers.size() && !isWaitingForFloatingBuffers.get()) { --- End diff -- By taking all buffers until none are available any more, we do not really ensure a fair distribution where large backlogs on one channel may actually "starve" another (except for the exclusive buffers). Floating buffers should be assigned round-robin among the channels. Also, actually, the total number of required buffers should be `backlog + initialCredit - currentCredit` with `currentCredit` being the credit that was guaranteed, i.e. `availableBuffers.size()`. Let me think about this a bit more to sort out the details - I'll come back to you. > Implement Netty receiver incoming pipeline for credit-based > ----------------------------------------------------------- > > Key: FLINK-7406 > URL: https://issues.apache.org/jira/browse/FLINK-7406 > Project: Flink > Issue Type: Sub-task > Components: Network > Reporter: zhijiang > Assignee: zhijiang > Fix For: 1.4.0 > > > This is a part of work for credit-based network flow control. > Currently {{PartitionRequestClientHandler}} receives and reads > {{BufferResponse}} from producer. It will request buffer from {{BufferPool}} > for holding the message. If not got, the message is staged temporarily and > {{autoread}} for channel is set false. > For credit-based mode, {{PartitionRequestClientHandler}} can always get > buffer from {{RemoteInputChannel}} for reading messages from producer. > The related works are: > * Add the backlog of producer in {{BufferResponse}} message structure > * {{PartitionRequestClientHandler}} requests buffer from > {{RemoteInputChannel}} directly > * {{PartitionRequestClientHandler}} updates backlog for > {{RemoteInputChannel}}, and it may trigger requests of floating buffers from > {{BufferPool}} -- This message was sent by Atlassian JIRA (v6.4.14#64029)