SPARK-1176 Add port configuration for HttpBroadcast

Uses spark.broadcast.port
apache · Jul 25, 2014 · f34115d · f34115d
1 parent 49ee29b
commit f34115d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/core/src/main/scala/org/apache/spark/broadcast/HttpBroadcast.scala b/core/src/main/scala/org/apache/spark/broadcast/HttpBroadcast.scala
@@ -152,7 +152,8 @@ private[broadcast] object HttpBroadcast extends Logging {
 
   private def createServer(conf: SparkConf) {
     broadcastDir = Utils.createTempDir(Utils.getLocalDir(conf))
-    server = new HttpServer(broadcastDir, securityManager)
+    val broadcastListenPort: Int = conf.getInt("spark.broadcast.port", 0)
+    server = new HttpServer(broadcastDir, securityManager, broadcastListenPort)
     server.start()
     serverUri = server.uri
     logInfo("Broadcast server started at " + serverUri)