[KYUUBI-2450][FOLLOWUP] Remove Operation handle from opHandleSet when…

… exception occurs
apache · Apr 28, 2022 · a4ef89a · a4ef89a
1 parent 93f13ef
commit a4ef89a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/kyuubi-common/src/main/scala/org/apache/kyuubi/session/AbstractSession.scala b/kyuubi-common/src/main/scala/org/apache/kyuubi/session/AbstractSession.scala
@@ -95,6 +95,7 @@ abstract class AbstractSession(
       opHandle
     } catch {
       case e: KyuubiSQLException =>
+        opHandleSet.remove(operation.getHandle)
         sessionManager.operationManager.closeOperation(operation.getHandle)
         throw e
     }