Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[jvm-packages] RabitTracker for Scala: allow specifying host ip from the xgboost-tracker.properties file (#3833)

Browse Source
This commit is contained in:
Matthew Tovbin 2018-10-26 22:01:36 -07:00 committed by Nan Zhu
parent 5fbe230636
commit d81fedb955
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
jvm-packages/xgboost4j/src/main/scala/ml/dmlc/xgboost4j/scala/rabit/RabitTracker.scala
View File

@ -20,7 +20,7 @@ import java.net.{InetAddress, InetSocketAddress}
import akka.actor.ActorSystem import akka.actor.ActorSystem
import akka.pattern.ask import akka.pattern.ask
import ml.dmlc.xgboost4j.java.IRabitTracker import ml.dmlc.xgboost4j.java.{IRabitTracker, TrackerProperties}
import ml.dmlc.xgboost4j.scala.rabit.handler.RabitTrackerHandler import ml.dmlc.xgboost4j.scala.rabit.handler.RabitTrackerHandler
import scala.concurrent.duration._ import scala.concurrent.duration._
@ -93,8 +93,11 @@ private[scala] class RabitTracker(numWorkers: Int, port: Option[Int] = None,
* @return Boolean flag indicating if the Rabit tracker starts successfully. * @return Boolean flag indicating if the Rabit tracker starts successfully.
*/ */
private def start(timeout: Duration): Boolean = { private def start(timeout: Duration): Boolean = {
val hostAddress = Option(TrackerProperties.getInstance().getHostIp)
.map(InetAddress.getByName).getOrElse(InetAddress.getLocalHost)
handler ? RabitTrackerHandler.StartTracker( handler ? RabitTrackerHandler.StartTracker(
new InetSocketAddress(InetAddress.getLocalHost, port.getOrElse(0)), maxPortTrials, timeout) new InetSocketAddress(hostAddress, port.getOrElse(0)), maxPortTrials, timeout)
// block by waiting for the actor to bind to a port // block by waiting for the actor to bind to a port
Try(Await.result(handler ? RabitTrackerHandler.RequestBoundFuture, askTimeout.duration) Try(Await.result(handler ? RabitTrackerHandler.RequestBoundFuture, askTimeout.duration)