File tree Expand file tree Collapse file tree 2 files changed +16
-0
lines changed
main/scala/org/apache/kyuubi/session
test/scala/org/apache/kyuubi/operation/log Expand file tree Collapse file tree 2 files changed +16
-0
lines changed Original file line number Diff line number Diff line change @@ -89,10 +89,23 @@ abstract class SessionManager(name: String) extends CompositeService(name) {
8989 if (session == null ) {
9090 throw KyuubiSQLException (s " Invalid $sessionHandle" )
9191 }
92+ deleteOperationLogSessionDir(sessionHandle)
9293 info(s " $sessionHandle is closed, current opening sessions $getOpenSessionCount" )
9394 session.close()
9495 }
9596
97+ private def deleteOperationLogSessionDir (sessionHandle : SessionHandle ): Unit = {
98+ _operationLogRoot.foreach(logRoot => {
99+ val rootPath = Paths .get(logRoot, sessionHandle.identifier.toString)
100+ try {
101+ Files .deleteIfExists(rootPath)
102+ } catch {
103+ case e : IOException =>
104+ error(s " Failed to delete session operation log directory ${rootPath.toString}" , e)
105+ }
106+ })
107+ }
108+
96109 def getSession (sessionHandle : SessionHandle ): Session = {
97110 val session = handleToSession.get(sessionHandle)
98111 if (session == null ) {
Original file line number Diff line number Diff line change @@ -82,6 +82,9 @@ class OperationLogSuite extends KyuubiFunSuite {
8282
8383 operationLog.close()
8484 assert(! Files .exists(logFile))
85+ assert(Files .exists(Paths .get(operationLogRoot, sHandle.identifier.toString)))
86+ sessionManager.closeSession(sHandle)
87+ assert(! Files .exists(Paths .get(operationLogRoot, sHandle.identifier.toString)))
8588 }
8689
8790 test(" log divert appender" ) {
You can’t perform that action at this time.
0 commit comments