[hotfix][task] Optimize and simplify Task#isBackPressured

deepfield · Jan 6, 2021 · 8d4be69 · 8d4be69
1 parent a8ec54c
commit 8d4be69
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/flink-runtime/src/main/java/org/apache/flink/runtime/taskmanager/Task.java b/flink-runtime/src/main/java/org/apache/flink/runtime/taskmanager/Task.java
@@ -511,12 +511,12 @@ public boolean isBackPressured() {
  if (invokable == null || consumableNotifyingPartitionWriters.length == 0 || !isRunning()) {
  return false;
  }
- final CompletableFuture<?>[] outputFutures =
-  new CompletableFuture[consumableNotifyingPartitionWriters.length];
- for (int i = 0; i < outputFutures.length; ++i) {
- outputFutures[i] = consumableNotifyingPartitionWriters[i].getAvailableFuture();
+ for (int i = 0; i < consumableNotifyingPartitionWriters.length; ++i) {
+ if (!consumableNotifyingPartitionWriters[i].isAvailable()) {
+  return true;
+ }
  }
- return !CompletableFuture.allOf(outputFutures).isDone();
+ return false;
  }
 
  // ------------------------------------------------------------------------