[SPARK-24506][UI] Add UI filters also to thriftserver tab

Author: mgaido91

core/src/main/scala/org/apache/spark/ui/WebUI.scala

@@ -59,6 +59,8 @@ private[spark] abstract class WebUI(
   def getTabs: Seq[WebUITab] = tabs
   def getHandlers: Seq[ServletContextHandler] = handlers
   def getSecurityManager: SecurityManager = securityManager
+  def handlersForPage(page: WebUIPage): Seq[ServletContextHandler] =
+    pageToHandlers.getOrElse(page, Seq.empty)
 
   /** Attach a tab to this UI, along with all of its attached pages. */
   def attachTab(tab: WebUITab) {

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

@@ -21,7 +21,7 @@ import org.apache.spark.{SparkContext, SparkException}
 import org.apache.spark.internal.Logging
 import org.apache.spark.sql.hive.thriftserver.HiveThriftServer2
 import org.apache.spark.sql.hive.thriftserver.ui.ThriftServerTab._
-import org.apache.spark.ui.{SparkUI, SparkUITab}
+import org.apache.spark.ui.{JettyUtils, SparkUI, SparkUITab}
 
 /**
  * Spark Web UI tab that shows statistics of jobs running in the thrift server.
@@ -39,6 +39,10 @@ private[thriftserver] class ThriftServerTab(sparkContext: SparkContext)
   attachPage(new ThriftServerSessionPage(this))
   parent.attachTab(this)
 
+  // We need to add the filters to the handlers generated here since the SparkUI has been already
+  // started.
+  JettyUtils.addFilters(this.pages.flatMap(parent.handlersForPage), parent.conf)
+
   def detach() {
     getSparkUI(sparkContext).detachTab(this)
   }