update

apache · Nov 20, 2023 · 96846d1 · 96846d1
1 parent 5d9b7dc
commit 96846d1
Showing 1 changed file with 19 additions and 17 deletions.
diff --git a/...engine/src/main/scala/org/apache/kyuubi/engine/spark/session/SparkSQLSessionManager.scala b/...engine/src/main/scala/org/apache/kyuubi/engine/spark/session/SparkSQLSessionManager.scala
@@ -67,25 +67,27 @@ class SparkSQLSessionManager private (name: String, spark: SparkSession)
     if (!userIsolatedSparkSession) {
       userIsolatedSparkSessionThread =
         Some(ThreadUtils.newDaemonSingleThreadScheduledExecutor("user-isolated-cache-checker"))
-      scheduleTolerableRunnableWithFixedDelay(
-        _,
-        () => {
-          userIsolatedCacheLock.synchronized {
-            val iter = userIsolatedCacheCount.entrySet().iterator()
-            while (iter.hasNext) {
-              val kv = iter.next()
-              if (kv.getValue._1 == 0 &&
-                kv.getValue._2 + userIsolatedIdleTimeout < System.currentTimeMillis()) {
-                userIsolatedCache.remove(kv.getKey)
-                iter.remove()
+      userIsolatedSparkSessionThread.foreach { thread =>
+        scheduleTolerableRunnableWithFixedDelay(
+          thread,
+          () => {
+            userIsolatedCacheLock.synchronized {
+              val iter = userIsolatedCacheCount.entrySet().iterator()
+              while (iter.hasNext) {
+                val kv = iter.next()
+                if (kv.getValue._1 == 0 &&
+                  kv.getValue._2 + userIsolatedIdleTimeout < System.currentTimeMillis()) {
+                  userIsolatedCache.remove(kv.getKey)
+                  iter.remove()
+                }
               }
             }
-          }
-        },
-        userIsolatedIdleInterval,
-        userIsolatedIdleInterval,
-        TimeUnit.MILLISECONDS,
-        "user-isolated-spark-session-cache-checker")
+          },
+          userIsolatedIdleInterval,
+          userIsolatedIdleInterval,
+          TimeUnit.MILLISECONDS,
+          "user-isolated-spark-session-cache-checker")
+      }
     }
   }