config the interval for updating the same status for a query

vajaw · Dec 11, 2022 · 57d7319 · 57d7319
1 parent 7aa3445
commit 57d7319
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/kyuubi-common/src/main/scala/org/apache/kyuubi/config/KyuubiConf.scala b/kyuubi-common/src/main/scala/org/apache/kyuubi/config/KyuubiConf.scala
@@ -1453,6 +1453,14 @@ object KyuubiConf {
       .timeConf
       .createWithDefault(Duration.ofSeconds(5).toMillis)
 
+  val OPERATION_STATUS_UPDATE_INTERVAL: ConfigEntry[Long] =
+    buildConf("kyuubi.operation.status.update.interval")
+      .internal
+      .doc("Interval(ms) for updating the same status for a query.")
+      .version("1.7.0")
+      .timeConf
+      .createWithDefault(Duration.ofSeconds(5).toMillis)
+
   val OPERATION_FORCE_CANCEL: ConfigEntry[Boolean] =
     buildConf("kyuubi.operation.interrupt.on.cancel")
       .doc("When true, all running tasks will be interrupted if one cancels a query. " +

diff --git a/kyuubi-server/src/main/scala/org/apache/kyuubi/operation/ExecuteStatement.scala b/kyuubi-server/src/main/scala/org/apache/kyuubi/operation/ExecuteStatement.scala
@@ -23,8 +23,8 @@ import org.apache.hive.service.rpc.thrift.{TGetOperationStatusResp, TOperationSt
 import org.apache.hive.service.rpc.thrift.TOperationState._
 
 import org.apache.kyuubi.KyuubiSQLException
+import org.apache.kyuubi.config.KyuubiConf
 import org.apache.kyuubi.events.{EventBus, KyuubiOperationEvent}
-import org.apache.kyuubi.operation.ExecuteStatement.DEFAULT_SAME_STATE_UPDATE_INTERVAL
 import org.apache.kyuubi.operation.FetchOrientation.FETCH_NEXT
 import org.apache.kyuubi.operation.OperationState.OperationState
 import org.apache.kyuubi.operation.log.OperationLog
@@ -82,6 +82,8 @@ class ExecuteStatement(
       var isComplete = false
       var lastState: TOperationState = null
       var lastStateUpdateTime: Long = 0L
+      val stateUpdateInterval =
+        session.sessionManager.getConf.get(KyuubiConf.OPERATION_STATUS_UPDATE_INTERVAL)
       while (!isComplete) {
         fetchQueryLog()
         verifyTStatus(statusResp.getStatus)
@@ -90,7 +92,7 @@ class ExecuteStatement(
         }
         val remoteState = statusResp.getOperationState
         if (lastState != remoteState ||
-          System.currentTimeMillis() - lastStateUpdateTime > DEFAULT_SAME_STATE_UPDATE_INTERVAL) {
+          System.currentTimeMillis() - lastStateUpdateTime > stateUpdateInterval) {
           lastStateUpdateTime = System.currentTimeMillis()
           info(s"Query[$statementId] in ${remoteState.name()}")
         }
@@ -165,7 +167,3 @@ class ExecuteStatement(
     EventBus.post(KyuubiOperationEvent(this))
   }
 }
-
-object ExecuteStatement {
-  final val DEFAULT_SAME_STATE_UPDATE_INTERVAL = 5000L
-}