diff --git a/externals/kyuubi-spark-sql-engine/src/main/scala/org/apache/kyuubi/engine/spark/session/SparkSessionImpl.scala b/externals/kyuubi-spark-sql-engine/src/main/scala/org/apache/kyuubi/engine/spark/session/SparkSessionImpl.scala index 85c918eb6..9498483be 100644 --- a/externals/kyuubi-spark-sql-engine/src/main/scala/org/apache/kyuubi/engine/spark/session/SparkSessionImpl.scala +++ b/externals/kyuubi-spark-sql-engine/src/main/scala/org/apache/kyuubi/engine/spark/session/SparkSessionImpl.scala @@ -24,6 +24,7 @@ import org.apache.spark.sql.{AnalysisException, SparkSession} import org.apache.spark.ui.SparkUIUtils.formatDuration import org.apache.kyuubi.KyuubiSQLException +import org.apache.kyuubi.config.KyuubiConf.SESSION_IDLE_TIMEOUT import org.apache.kyuubi.config.KyuubiReservedKeys.KYUUBI_SESSION_HANDLE_KEY import org.apache.kyuubi.engine.spark.events.SessionEvent import org.apache.kyuubi.engine.spark.operation.SparkSQLOperationManager @@ -57,6 +58,13 @@ class SparkSessionImpl( } } + override val sessionIdleTimeoutThreshold: Long = { + conf.get(SESSION_IDLE_TIMEOUT.key) + .map(_.toLong) + .getOrElse( + sessionManager.getConf.get(SESSION_IDLE_TIMEOUT)) + } + private val sessionEvent = SessionEvent(this) override def open(): Unit = {