update

Author: shahidki31

sql/hive-thriftserver/src/main/resources/META-INF/services/org.apache.spark.status.AppHistoryServerPlugin

@@ -1 +1 @@
-org.apache.spark.sql.hive.thriftserver.ThriftServerHistoryServerPlugin
+org.apache.spark.sql.hive.thriftserver.HiveThriftServer2HistoryServerPlugin

sql/hive-thriftserver/src/main/scala/org/apache/spark/sql/hive/thriftserver/ThriftServerHistoryServerPlugin.scala renamed to sql/hive-thriftserver/src/main/scala/org/apache/spark/sql/hive/thriftserver/HiveThriftServer2HistoryServerPlugin.scala

@@ -18,21 +18,20 @@
 package org.apache.spark.sql.hive.thriftserver
 
 import org.apache.spark.SparkConf
-import org.apache.spark.internal.Logging
 import org.apache.spark.scheduler.SparkListener
 import org.apache.spark.sql.hive.thriftserver.ui.{HiveThriftServer2AppStatusStore, ThriftServerTab}
 import org.apache.spark.status.{AppHistoryServerPlugin, ElementTrackingStore}
 import org.apache.spark.ui.SparkUI
 
-class ThriftServerHistoryServerPlugin extends AppHistoryServerPlugin with Logging {
+class HiveThriftServer2HistoryServerPlugin extends AppHistoryServerPlugin {
 
   override def createListeners(conf: SparkConf, store: ElementTrackingStore): Seq[SparkListener] = {
     Seq(new HiveThriftServer2Listener(store, None, None, Some(conf), false))
   }
 
   override def setupUI(ui: SparkUI): Unit = {
     val store = new HiveThriftServer2AppStatusStore(ui.store.store)
-    if (store.getSessionList.nonEmpty) {
+    if (store.getExecutionCount() > 0) {
       new ThriftServerTab(store, ui)
     }
   }

sql/hive-thriftserver/src/main/scala/org/apache/spark/sql/hive/thriftserver/ui/HiveThriftServer2AppStatusStore.scala

@@ -66,6 +66,14 @@ class HiveThriftServer2AppStatusStore(
       execInfo.state == ExecutionState.CANCELED ||
       execInfo.state == ExecutionState.CLOSED)
   }
+
+  def getSessionCount(): Long = {
+    store.count(classOf[SessionInfo])
+  }
+
+  def getExecutionCount(): Long = {
+    store.count(classOf[ExecutionInfo])
+  }
 }
 
 private[thriftserver] class SessionInfo(