Zakelly commented on code in PR #24633: URL: https://github.com/apache/flink/pull/24633#discussion_r1560429469
########## flink-runtime/src/main/java/org/apache/flink/runtime/asyncprocessing/AsyncExecutionController.java: ########## @@ -165,6 +215,31 @@ <IN, OUT> void insertBlockingBuffer(StateRequest<K, IN, OUT> request) { * @param force whether to trigger requests in force. */ void triggerIfNeeded(boolean force) { - // TODO: implement the trigger logic. + if (!force && stateRequestsBuffer.activeQueueSize() < batchSize) { Review Comment: Another thing comes to my mind. The timeout trigger is needed here. How about adding a TODO here? ########## flink-runtime/src/main/java/org/apache/flink/runtime/asyncprocessing/AsyncExecutionController.java: ########## @@ -0,0 +1,248 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.apache.flink.runtime.asyncprocessing; + +import org.apache.flink.api.common.operators.MailboxExecutor; +import org.apache.flink.api.common.state.v2.State; +import org.apache.flink.core.state.InternalStateFuture; +import org.apache.flink.util.FlinkRuntimeException; +import org.apache.flink.util.Preconditions; + +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +import javax.annotation.Nullable; + +import java.util.List; +import java.util.concurrent.atomic.AtomicInteger; + +/** + * The Async Execution Controller (AEC) receives processing requests from operators, and put them + * into execution according to some strategies. + * + * <p>It is responsible for: + * <li>Preserving the sequence of elements bearing the same key by delaying subsequent requests + * until the processing of preceding ones is finalized. + * <li>Tracking the in-flight data(records) and blocking the input if too much data in flight + * (back-pressure). It invokes {@link MailboxExecutor#yield()} to pause current operations, + * allowing for the execution of callbacks (mails in Mailbox). + * + * @param <R> the type of the record + * @param <K> the type of the key + */ +public class AsyncExecutionController<R, K> { + + private static final Logger LOG = LoggerFactory.getLogger(AsyncExecutionController.class); + + public static final int DEFAULT_BATCH_SIZE = 1000; + public static final int DEFAULT_MAX_IN_FLIGHT_RECORD_NUM = 6000; + + /** + * The batch size. When the number of state requests in the active buffer exceeds the batch + * size, a batched state execution would be triggered. + */ + private final int batchSize; + + /** The max allow number of in-flight records. */ + private final int maxInFlightRecordNum; + + /** + * The mailbox executor borrowed from {@code StreamTask}. Keeping the reference of + * mailboxExecutor here is to restrict the number of in-flight records, when the number of + * in-flight records > {@link #maxInFlightRecordNum}, the newly entering records would be + * blocked. + */ + private final MailboxExecutor mailboxExecutor; + + /** The key accounting unit which is used to detect the key conflict. */ + final KeyAccountingUnit<R, K> keyAccountingUnit; + + /** + * A factory to build {@link org.apache.flink.core.state.InternalStateFuture}, this will auto + * wire the created future with mailbox executor. Also conducting the context switch. + */ + private final StateFutureFactory<R, K> stateFutureFactory; + + /** The state executor where the {@link StateRequest} is actually executed. */ + final StateExecutor stateExecutor; + + /** The corresponding context that currently runs in task thread. */ + RecordContext<R, K> currentContext; + + /** The buffer to store the state requests to execute in batch. */ + StateRequestsBuffer<R, K> stateRequestsBuffer; + + /** The number of in-flight records. */ + final AtomicInteger inFlightRecordNum; + + public AsyncExecutionController(MailboxExecutor mailboxExecutor, StateExecutor stateExecutor) { + this(mailboxExecutor, stateExecutor, DEFAULT_BATCH_SIZE, DEFAULT_MAX_IN_FLIGHT_RECORD_NUM); + } + + public AsyncExecutionController( + MailboxExecutor mailboxExecutor, + StateExecutor stateExecutor, + int batchSize, + int maxInFlightRecords) { + this.keyAccountingUnit = new KeyAccountingUnit<>(); + this.mailboxExecutor = mailboxExecutor; + this.stateFutureFactory = new StateFutureFactory<>(this, mailboxExecutor); + this.stateExecutor = stateExecutor; + this.batchSize = batchSize; + this.maxInFlightRecordNum = maxInFlightRecords; + this.stateRequestsBuffer = new StateRequestsBuffer<>(); + this.inFlightRecordNum = new AtomicInteger(0); + LOG.info("Create AsyncExecutionController: maxInFlightRecordsNum {}", maxInFlightRecords); + } + + /** + * Build a new context based on record and key. Also wired with internal {@link + * KeyAccountingUnit}. + * + * @param record the given record. + * @param key the given key. + * @return the built record context. + */ + public RecordContext<R, K> buildContext(R record, K key) { + return new RecordContext<>(record, key, this::disposeContext); + } + + /** + * Each time before a code segment (callback) is about to run in mailbox (task thread), this + * method should be called to switch a context in AEC. + * + * @param switchingContext the context to switch. + */ + public void setCurrentContext(RecordContext<R, K> switchingContext) { + currentContext = switchingContext; + } + + /** + * Dispose a context. + * + * @param toDispose the context to dispose. + */ + public void disposeContext(RecordContext<R, K> toDispose) { + keyAccountingUnit.release(toDispose.getRecord(), toDispose.getKey()); + inFlightRecordNum.decrementAndGet(); + RecordContext<R, K> nextRecordCtx = + stateRequestsBuffer.tryMigrateSpecificOne(toDispose.getKey()); + if (nextRecordCtx != null) { + Preconditions.checkState( + tryOccupyKey(nextRecordCtx), + String.format("key(%s) is already occupied.", nextRecordCtx.getKey())); + } + } + + /** + * Try to occupy a key by a given context. + * + * @param recordContext the given context. + * @return true if occupy succeed or the key has already occupied by this context. + */ + boolean tryOccupyKey(RecordContext<R, K> recordContext) { + boolean occupied = recordContext.isKeyOccupied(); + if (!occupied + && keyAccountingUnit.occupy(recordContext.getRecord(), recordContext.getKey())) { + recordContext.setKeyOccupied(); + occupied = true; + } + return occupied; + } + + /** + * Submit a {@link StateRequest} to this AEC and trigger if needed. + * + * @param state the state to request. + * @param type the type of this request. + * @param payload the payload input for this request. + * @return the state future. + */ + public <IN, OUT> InternalStateFuture<OUT> handleRequest( + @Nullable State state, StateRequestType type, @Nullable IN payload) { + // Step 1: build state future & assign context. + InternalStateFuture<OUT> stateFuture = stateFutureFactory.build(currentContext); + + StateRequest<K, IN, OUT> request = + new StateRequest<>(state, type, payload, stateFuture, currentContext); + + // Step 2: try to seize the capacity, if the current in-flight records exceeds the limit, + // block the current state request from entering until some buffered requests are processed. + seizeCapacity(); + + // Step 3: try to occupy the key and place it into right buffer. + if (tryOccupyKey(currentContext)) { + insertActiveBuffer(request); + } else { + insertBlockingBuffer(request); + } + // Step 4: trigger the (active) buffer if needed. + triggerIfNeeded(false); + return stateFuture; + } + + <IN, OUT> void insertActiveBuffer(StateRequest<K, IN, OUT> request) { + stateRequestsBuffer.enqueueToActive(request); + } + + <IN, OUT> void insertBlockingBuffer(StateRequest<K, IN, OUT> request) { + stateRequestsBuffer.enqueueToBlocking(request); + } + + /** + * Trigger a batch of requests. + * + * @param force whether to trigger requests in force. + */ + void triggerIfNeeded(boolean force) { + if (!force && stateRequestsBuffer.activeQueueSize() < batchSize) { + return; + } + List<StateRequest<?, ?, ?>> toRun = stateRequestsBuffer.nextToRun(); + stateRequestsBuffer.clearActiveQueue(); + stateExecutor.executeBatchRequests(toRun); + } + + private void seizeCapacity() { + // 1. Check if the record is already in buffer. If yes, this indicates that it is a state + // request resulting from a callback statement, otherwise, it signifies the initial state + // request + // for a newly entered record. + if (currentContext.isKeyOccupied()) { + return; + } + RecordContext<R, K> storedContext = currentContext; + // 2. If the state request is for a newly entered record, the in-flight record number should + // be less than the max in-flight record number. + // Note: the currentContext may be updated by {@code StateFutureFactory#build}. + try { + while (inFlightRecordNum.get() > maxInFlightRecordNum) { + if (!mailboxExecutor.tryYield()) { + triggerIfNeeded(true); + Thread.sleep(50); Review Comment: Yet I thought this `sleep` is meaningful for preventing the CPU keep running with 100% usage. I think sleep for `1ms` is better. ########## flink-runtime/src/main/java/org/apache/flink/runtime/asyncprocessing/StateRequestBuffer.java: ########## @@ -0,0 +1,129 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.apache.flink.runtime.asyncprocessing; + +import org.apache.flink.annotation.VisibleForTesting; + +import oshi.annotation.concurrent.NotThreadSafe; + +import javax.annotation.Nullable; + +import java.util.Deque; +import java.util.HashMap; +import java.util.LinkedList; +import java.util.List; +import java.util.Map; + +/** + * A buffer to hold state requests to execute state requests in batch. + * + * @param <R> the type of the record + * @param <K> the type of the key + */ +@NotThreadSafe +public class StateRequestBuffer<R, K> { + /** + * The state requests in this buffer could be executed when the buffer is full or configured + * batch size is reached. All operations on this buffer must be invoked in task thread. + */ + final LinkedList<StateRequest<K, ?, ?>> activeQueue; + + /** + * The requests in that should wait until all preceding records with identical key finishing its + * execution. After which the queueing requests will move into the active buffer. All operations + * on this buffer must be invoked in task thread. + */ + final Map<K, Deque<StateRequest<K, ?, ?>>> blockingQueue; + + /** The number of state requests in blocking queue. */ + int blockingQueueSize; + + public StateRequestBuffer() { + this.activeQueue = new LinkedList<>(); + this.blockingQueue = new HashMap<>(); + this.blockingQueueSize = 0; + } + + void enqueueToActive(StateRequest<K, ?, ?> request) { + activeQueue.add(request); + } + + void enqueueToBlocking(StateRequest<K, ?, ?> request) { + blockingQueue + .computeIfAbsent(request.getRecordContext().getKey(), k -> new LinkedList<>()) + .add(request); + blockingQueueSize++; + } + + /** + * Try to migrate one state request with specific key from blocking queue to active queue. + * + * @param key The key to release, the other records with this key is no longer blocking. + * @return The first record context with the same key in blocking queue, null if no such record. + */ + @Nullable + RecordContext<R, K> tryMigrateSpecificOne(K key) { + if (!blockingQueue.containsKey(key)) { + return null; + } + + StateRequest<K, ?, ?> stateRequest = blockingQueue.get(key).getFirst(); + activeQueue.add(stateRequest); + blockingQueue.get(key).removeFirst(); + if (blockingQueue.get(key).isEmpty()) { + blockingQueue.remove(key); + } + blockingQueueSize--; + return (RecordContext<R, K>) stateRequest.getRecordContext(); + } + + /** + * Get the number of state requests of blocking queue in constant-time. + * + * @return the number of state requests of blocking queue. + */ + @VisibleForTesting + int blockingQueueSize() { + return blockingQueueSize; + } + + /** + * Get the number of state requests of active queue in constant-time. + * + * @return the number of state requests of active queue. + */ + @VisibleForTesting + int activeQueueSize() { + return activeQueue.size(); + } + + /** + * Try to pop N state requests from active queue, if the size of active queue is less than N, + * return all the requests in active queue. + * + * @param N the number of state requests to pop. + * @return A list of state requests. + */ + List<StateRequest<?, ?, ?>> popN(int N) { Review Comment: `popActive` ? ########## flink-runtime/src/main/java/org/apache/flink/runtime/asyncprocessing/StateRequestBuffer.java: ########## @@ -0,0 +1,129 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.apache.flink.runtime.asyncprocessing; + +import org.apache.flink.annotation.VisibleForTesting; + +import oshi.annotation.concurrent.NotThreadSafe; Review Comment: Should be `javax.annotation.concurrent.NotThreadSafe` ? ########## flink-runtime/src/main/java/org/apache/flink/runtime/asyncprocessing/StateRequestBuffer.java: ########## @@ -0,0 +1,129 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.apache.flink.runtime.asyncprocessing; + +import org.apache.flink.annotation.VisibleForTesting; + +import oshi.annotation.concurrent.NotThreadSafe; + +import javax.annotation.Nullable; + +import java.util.Deque; +import java.util.HashMap; +import java.util.LinkedList; +import java.util.List; +import java.util.Map; + +/** + * A buffer to hold state requests to execute state requests in batch. + * + * @param <R> the type of the record + * @param <K> the type of the key + */ +@NotThreadSafe Review Comment: Better add some description about only manipulating this class within `task thread` ########## flink-runtime/src/main/java/org/apache/flink/runtime/asyncprocessing/StateRequestBuffer.java: ########## @@ -0,0 +1,129 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.apache.flink.runtime.asyncprocessing; + +import org.apache.flink.annotation.VisibleForTesting; + +import oshi.annotation.concurrent.NotThreadSafe; + +import javax.annotation.Nullable; + +import java.util.Deque; +import java.util.HashMap; +import java.util.LinkedList; +import java.util.List; +import java.util.Map; + +/** + * A buffer to hold state requests to execute state requests in batch. + * + * @param <R> the type of the record + * @param <K> the type of the key + */ +@NotThreadSafe +public class StateRequestBuffer<R, K> { + /** + * The state requests in this buffer could be executed when the buffer is full or configured + * batch size is reached. All operations on this buffer must be invoked in task thread. + */ + final LinkedList<StateRequest<K, ?, ?>> activeQueue; + + /** + * The requests in that should wait until all preceding records with identical key finishing its + * execution. After which the queueing requests will move into the active buffer. All operations + * on this buffer must be invoked in task thread. + */ + final Map<K, Deque<StateRequest<K, ?, ?>>> blockingQueue; + + /** The number of state requests in blocking queue. */ + int blockingQueueSize; + + public StateRequestBuffer() { + this.activeQueue = new LinkedList<>(); + this.blockingQueue = new HashMap<>(); + this.blockingQueueSize = 0; + } + + void enqueueToActive(StateRequest<K, ?, ?> request) { + activeQueue.add(request); + } + + void enqueueToBlocking(StateRequest<K, ?, ?> request) { + blockingQueue + .computeIfAbsent(request.getRecordContext().getKey(), k -> new LinkedList<>()) + .add(request); + blockingQueueSize++; + } + + /** + * Try to migrate one state request with specific key from blocking queue to active queue. + * + * @param key The key to release, the other records with this key is no longer blocking. + * @return The first record context with the same key in blocking queue, null if no such record. + */ + @Nullable + RecordContext<R, K> tryMigrateSpecificOne(K key) { Review Comment: How about naming this `tryActivateOneByKey` or else? The 'migrate' seems confusing. -- 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: issues-unsubscr...@flink.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org