session immediately

zhaohehuhu · Mar 5, 2024 · aad2bbb · aad2bbb
1 parent 8e957c7
commit aad2bbb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/...k-sql-engine/src/main/scala/org/apache/kyuubi/engine/spark/operation/SparkOperation.scala b/...k-sql-engine/src/main/scala/org/apache/kyuubi/engine/spark/operation/SparkOperation.scala
@@ -135,8 +135,9 @@ abstract class SparkOperation(session: Session)
           info(s"statementId=${statementId}, " +
             s"operationRunTime=${formatDuration(l.getOperationRunTime)}, " +
             s"operationCpuTime=${formatDuration(l.getOperationCpuTime / 1000000)}")
-          session.asInstanceOf[SparkSessionImpl].increaseRunTime(l.getOperationRunTime)
-          session.asInstanceOf[SparkSessionImpl].increaseCpuTime(l.getOperationCpuTime)
+          session.asInstanceOf[SparkSessionImpl].increaseRunAndCpuTime(
+            l.getOperationRunTime,
+            l.getOperationCpuTime)
         })
       }
     }

diff --git a/...k-sql-engine/src/main/scala/org/apache/kyuubi/engine/spark/session/SparkSessionImpl.scala b/...k-sql-engine/src/main/scala/org/apache/kyuubi/engine/spark/session/SparkSessionImpl.scala
@@ -129,11 +129,8 @@ class SparkSessionImpl(
       handle)
   }
 
-  def increaseRunTime(time: Long): Unit = {
-    sessionRunTime.getAndAdd(time)
-  }
-
-  def increaseCpuTime(time: Long): Unit = {
-    sessionCpuTime.getAndAdd(time)
+  def increaseRunAndCpuTime(runTime: Long, cpuTime: Long): Unit = {
+    sessionEvent.sessionRunTime = sessionRunTime.addAndGet(runTime)
+    sessionEvent.sessionCpuTime = sessionCpuTime.addAndGet(cpuTime)
   }
 }