[FLINK-16057][task] Optimize TaskMailbox state retrieval

Don't lock in TaskMailboxImpl.getState for task thread. This makes ContinuousFileReaderOperator about 29% faster.
shuttie · Jun 2, 2020 · 31d661d · 31d661d
1 parent 1a69cb9
commit 31d661d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/...-java/src/main/java/org/apache/flink/streaming/runtime/tasks/mailbox/TaskMailboxImpl.java b/...-java/src/main/java/org/apache/flink/streaming/runtime/tasks/mailbox/TaskMailboxImpl.java
@@ -311,6 +311,9 @@ public List<Mail> close() {
 	@Nonnull
 	@Override
 	public State getState() {
+		if (isMailboxThread()) {
+			return state;
+		}
 		final ReentrantLock lock = this.lock;
 		lock.lock();
 		try {