[KYUUBI #1806] Fix LaunchEngine may be closed multiple times

<!--
Thanks for sending a pull request!

Here are some tips for you:
  1. If this is your first time, please read our contributor guidelines: https://kyuubi.readthedocs.io/en/latest/community/contributions.html
  2. If the PR is related to an issue in https://github.com/apache/incubator-kyuubi/issues, add '[KYUUBI #XXXX]' in your PR title, e.g., '[KYUUBI #XXXX] Your PR title ...'.
  3. If the PR is unfinished, add '[WIP]' in your PR title, e.g., '[WIP][KYUUBI #XXXX] Your PR title ...'.
-->

### _Why are the changes needed?_
<!--
Please clarify why the changes are needed. For instance,
  1. If you add a feature, you can talk about the use case of it.
  2. If you fix a bug, you can clarify why it is a bug.
-->

close #1806

### _How was this patch tested?_
- [ ] Add some test cases that check the changes thoroughly including negative and positive cases if possible

- [ ] Add screenshots for manual tests if appropriate

- [ ] [Run test](https://kyuubi.readthedocs.io/en/latest/develop_tools/testing.html#running-tests) locally before make a pull request

Closes #1807 from cfmcgrady/kyuubi-1806.

Closes #1806

c70a594f [Fu Chen] fix

Authored-by: Fu Chen <cfmcgrady@gmail.com>
Signed-off-by: Kent Yao <yao@apache.org>
This commit is contained in:
Fu Chen 2022-01-20 12:29:04 +08:00 committed by Kent Yao
parent 555dfac5e3
commit 5bc7ea1159
No known key found for this signature in database
GPG Key ID: F7051850A0AF904D

View File

@ -126,7 +126,9 @@ class KyuubiSessionImpl(
}
override def close(): Unit = {
closeOperation(launchEngineOp.getHandle)
if (!launchEngineOp.isTimedOut) {
closeOperation(launchEngineOp.getHandle)
}
super.close()
sessionManager.credentialsManager.removeSessionCredentialsEpoch(handle.identifier.toString)
try {