[flink] 02/03: [FLINK-24706][coordination] Strip completion exception in HeartbeatManager

[chesnay]

This is an automated email from the ASF dual-hosted git repository.

chesnay pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/flink.git

commit 250cc81adbcf1264defa00ef97fe0f26241a6808
Author: Chesnay Schepler <ches...@apache.org>
AuthorDate: Tue Nov 2 14:05:39 2021 +0100

    [FLINK-24706][coordination] Strip completion exception in HeartbeatManager
---
 .../java/org/apache/flink/runtime/heartbeat/HeartbeatManagerImpl.java | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git 
a/flink-runtime/src/main/java/org/apache/flink/runtime/heartbeat/HeartbeatManagerImpl.java
 
b/flink-runtime/src/main/java/org/apache/flink/runtime/heartbeat/HeartbeatManagerImpl.java
index 3063581..71e7c2e 100644
--- 
a/flink-runtime/src/main/java/org/apache/flink/runtime/heartbeat/HeartbeatManagerImpl.java
+++ 
b/flink-runtime/src/main/java/org/apache/flink/runtime/heartbeat/HeartbeatManagerImpl.java
@@ -20,6 +20,7 @@ package org.apache.flink.runtime.heartbeat;
 
 import org.apache.flink.runtime.clusterframework.types.ResourceID;
 import org.apache.flink.runtime.rpc.exceptions.RecipientUnreachableException;
+import org.apache.flink.util.ExceptionUtils;
 import org.apache.flink.util.Preconditions;
 import org.apache.flink.util.concurrent.FutureUtils;
 import org.apache.flink.util.concurrent.ScheduledExecutor;
@@ -244,7 +245,8 @@ public class HeartbeatManagerImpl<I, O> implements 
HeartbeatManager<I, O> {
     protected BiConsumer<Void, Throwable> handleHeartbeatRpc(ResourceID 
heartbeatTarget) {
         return (unused, failure) -> {
             if (failure != null) {
-                handleHeartbeatRpcFailure(heartbeatTarget, failure);
+                handleHeartbeatRpcFailure(
+                        heartbeatTarget, 
ExceptionUtils.stripCompletionException(failure));
             } else {
                 handleHeartbeatRpcSuccess(heartbeatTarget);
             }