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) {
89
89
if (session == null ) {
90
90
throw KyuubiSQLException (s " Invalid $sessionHandle" )
91
91
}
92
+ deleteOperationLogSessionDir(sessionHandle)
92
93
info(s " $sessionHandle is closed, current opening sessions $getOpenSessionCount" )
93
94
session.close()
94
95
}
95
96
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
+
96
109
def getSession (sessionHandle : SessionHandle ): Session = {
97
110
val session = handleToSession.get(sessionHandle)
98
111
if (session == null ) {
Original file line number Diff line number Diff line change @@ -82,6 +82,9 @@ class OperationLogSuite extends KyuubiFunSuite {
82
82
83
83
operationLog.close()
84
84
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)))
85
88
}
86
89
87
90
test(" log divert appender" ) {
You can’t perform that action at this time.
0 commit comments