using function

KaiXinXiaoLei · KaiXinXiaoLei · commit d7ed6dc98155 · 2015-09-14T10:48:21.000+08:00
diff --git a/core/src/main/scala/org/apache/spark/scheduler/cluster/YarnSchedulerBackend.scala b/core/src/main/scala/org/apache/spark/scheduler/cluster/YarnSchedulerBackend.scala
@@ -91,6 +91,7 @@ private[spark] abstract class YarnSchedulerBackend(
       filterParams.foreach { case (k, v) => conf.set(s"spark.$filterName.param.$k", v) }
       scheduler.sc.ui.foreach { ui => JettyUtils.addFilters(ui.getHandlers, conf) }
     }
+    scheduler.sc.executorAllocationManager.foreach(_.reSetNumExecutorsTarget())
   }
 
   override def createDriverEndpoint(properties: Seq[(String, String)]): DriverEndpoint = {

Original file line number	Diff line number	Diff line change
`@@ -91,6 +91,7 @@ private[spark] abstract class YarnSchedulerBackend(`
`91`	`91`	`filterParams.foreach { case (k, v) => conf.set(s"spark.$filterName.param.$k", v) }`
`92`	`92`	`scheduler.sc.ui.foreach { ui => JettyUtils.addFilters(ui.getHandlers, conf) }`
`93`	`93`	`}`
	`94`	`+ scheduler.sc.executorAllocationManager.foreach(_.reSetNumExecutorsTarget())`
`94`	`95`	`}`
`95`	`96`
`96`	`97`	`override def createDriverEndpoint(properties: Seq[(String, String)]): DriverEndpoint = {`