update

Author: AngersZhuuuu

core/src/main/scala/org/apache/spark/deploy/history/HistoryServer.scala

@@ -308,6 +308,7 @@ object HistoryServer extends Logging {
 
     val server = new HistoryServer(conf, provider, securityManager, port)
     server.bind()
+    server.attachAllHandler()
     provider.start()
 
     ShutdownHookManager.addShutdownHook { () => server.stop() }

core/src/main/scala/org/apache/spark/deploy/master/Master.scala

@@ -143,6 +143,7 @@ private[deploy] class Master(
     logInfo(s"Running Spark version ${org.apache.spark.SPARK_VERSION}")
     webUi = new MasterWebUI(this, webUiPort)
     webUi.bind()
+    webUi.attachAllHandler()
     masterWebUiUrl = webUi.webUrl
     if (reverseProxy) {
       val uiReverseProxyUrl = conf.get(UI_REVERSE_PROXY_URL).map(_.stripSuffix("/"))

core/src/main/scala/org/apache/spark/deploy/worker/Worker.scala

@@ -234,6 +234,7 @@ private[deploy] class Worker(
     setupWorkerResources()
     webUi = new WorkerWebUI(this, workDir, webUiPort)
     webUi.bind()
+    webUi.attachAllHandler()
 
     workerWebUiUrl = s"${webUi.scheme}$publicAddress:${webUi.boundPort}"
     registerWithMaster()

core/src/test/scala/org/apache/spark/deploy/history/HistoryServerSuite.scala

@@ -93,6 +93,7 @@ class HistoryServerSuite extends SparkFunSuite with BeforeAndAfter with Matchers
     server = new HistoryServer(conf, provider, securityManager, 18080)
     server.initialize()
     server.bind()
+    server.attachAllHandler()
     provider.start()
     port = server.boundPort
   }
@@ -412,6 +413,7 @@ class HistoryServerSuite extends SparkFunSuite with BeforeAndAfter with Matchers
     server = new HistoryServer(myConf, provider, securityManager, 0)
     server.initialize()
     server.bind()
+    server.attachAllHandler()
     provider.start()
     val port = server.boundPort
     val metrics = server.cacheMetrics

core/src/test/scala/org/apache/spark/deploy/history/RealBrowserUIHistoryServerSuite.scala

@@ -89,6 +89,7 @@ abstract class RealBrowserUIHistoryServerSuite(val driverProp: String)
     server = new HistoryServer(conf, provider, securityManager, 18080)
     server.initialize()
     server.bind()
+    server.attachAllHandler()
     provider.start()
     port = server.boundPort
   }

core/src/test/scala/org/apache/spark/deploy/master/ui/MasterWebUISuite.scala

@@ -50,6 +50,7 @@ class MasterWebUISuite extends SparkFunSuite with BeforeAndAfterAll {
   override def beforeAll(): Unit = {
     super.beforeAll()
     masterWebUI.bind()
+    masterWebUI.attachAllHandler()
   }
 
   override def afterAll(): Unit = {