[LIVY-987] NPE when waiting for thrift session to start timeout.

apache · Aug 25, 2023 · 9413296 · 9413296
1 parent 5dccc47
commit 9413296
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...tserver/server/src/main/scala/org/apache/livy/thriftserver/LivyThriftSessionManager.scala b/...tserver/server/src/main/scala/org/apache/livy/thriftserver/LivyThriftSessionManager.scala
@@ -95,7 +95,7 @@ class LivyThriftSessionManager(val server: LivyThriftServer, val livyConf: LivyC
     if (!future.isCompleted) {
       Try(Await.result(future, maxSessionWait)) match {
         case Success(session) => session
-        case Failure(e) => throw e.getCause
+        case Failure(e) => throw Option(e.getCause).getOrElse(e)
       }
     } else {
       future.value match {