diff --git a/spark-client/src/main/java/org/apache/hive/spark/client/rpc/Rpc.java b/spark-client/src/main/java/org/apache/hive/spark/client/rpc/Rpc.java index 680e6b8..cbbfb1c 100644 --- a/spark-client/src/main/java/org/apache/hive/spark/client/rpc/Rpc.java +++ b/spark-client/src/main/java/org/apache/hive/spark/client/rpc/Rpc.java @@ -491,7 +491,12 @@ public void handle(Callback[] callbacks) { void sendHello(Channel c) throws Exception { byte[] hello = client.hasInitialResponse() ? client.evaluateChallenge(new byte[0]) : new byte[0]; - c.writeAndFlush(new SaslMessage(clientId, hello)); + c.writeAndFlush(new SaslMessage(clientId, hello)).addListener(future -> { + if (!future.isSuccess()) { + LOG.error("Failed to send hello to server", future.cause()); + onError(future.cause()); + } + }); } } diff --git a/spark-client/src/main/java/org/apache/hive/spark/client/rpc/SaslHandler.java b/spark-client/src/main/java/org/apache/hive/spark/client/rpc/SaslHandler.java index 26edc63..8a42773 100644 --- a/spark-client/src/main/java/org/apache/hive/spark/client/rpc/SaslHandler.java +++ b/spark-client/src/main/java/org/apache/hive/spark/client/rpc/SaslHandler.java @@ -36,7 +36,7 @@ // LOG is not static to make debugging easier (being able to identify which sub-class // generated the log message). - private final Logger LOG; + protected final Logger LOG; private final boolean requiresEncryption; private KryoMessageCodec kryo; private boolean hasAuthResponse = false;