in AM create a new actorSystem

tmyklebu · Jan 9, 2015 · 7d33791 · 7d33791
1 parent 2164ea8
commit 7d33791
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/yarn/src/main/scala/org/apache/spark/deploy/yarn/ApplicationMaster.scala b/yarn/src/main/scala/org/apache/spark/deploy/yarn/ApplicationMaster.scala
@@ -245,12 +245,14 @@ private[spark] class ApplicationMaster(args: ApplicationMasterArguments,
         ApplicationMaster.EXIT_SC_NOT_INITED,
         "Timed out waiting for SparkContext.")
     } else {
+      actorSystem = AkkaUtils.createActorSystem("sparkYarnAM", Utils.localHostName, 0,
+        conf = sparkConf, securityManager = securityMgr)._1
       val driverUrl = "akka.tcp://%s@%s:%s/user/%s".format(
         SparkEnv.driverActorSystemName,
         sc.getConf.get("spark.driver.host"),
         sc.getConf.get("spark.driver.port"),
         YarnSchedulerBackend.ACTOR_NAME)
-      actor = sc.env.actorSystem.actorOf(Props(new AMActor(driverUrl, true)), name = "YarnAM")
+      actor = actorSystem.actorOf(Props(new AMActor(driverUrl, true)), name = "YarnAM")
       registerAM(sc.ui.map(_.appUIAddress).getOrElse(""), securityMgr)
       userClassThread.join()
     }