Added docker build arguments

Author: liyinan926

integration-test/src/test/scala/org/apache/spark/deploy/k8s/integrationtest/docker/SparkDockerImageBuilder.scala

@@ -17,9 +17,10 @@
 package org.apache.spark.deploy.k8s.integrationtest.docker
 
 import java.net.URI
+import java.net.URLEncoder
 import java.nio.file.Paths
 
-import com.spotify.docker.client.{DefaultDockerClient, DockerCertificates, LoggingBuildHandler}
+import com.spotify.docker.client.{DockerClient, DefaultDockerClient, DockerCertificates, LoggingBuildHandler}
 import org.apache.http.client.utils.URIBuilder
 import org.scalatest.concurrent.{Eventually, PatienceConfiguration}
 import org.scalatest.time.{Minutes, Seconds, Span}
@@ -61,17 +62,19 @@ private[spark] class SparkDockerImageBuilder
 
   def buildSparkDockerImages(): Unit = {
     Eventually.eventually(TIMEOUT, INTERVAL) { dockerClient.ping() }
-    buildImage("spark-base", BASE_DOCKER_FILE)
-    buildImage("spark-driver", DRIVER_DOCKER_FILE)
-    buildImage("spark-executor", EXECUTOR_DOCKER_FILE)
+    buildImage("spark-base", BASE_DOCKER_FILE,
+      "{\"spark_jars\":\"jars\",\"img_path\":\"kubernetes/dockerfiles\"}")
+    buildImage("spark-driver", DRIVER_DOCKER_FILE, "{\"base_image\":\"spark-base\"}")
+    buildImage("spark-executor", EXECUTOR_DOCKER_FILE, "{\"base_image\":\"spark-base\"}")
   }
 
-  private def buildImage(name: String, dockerFile: String): Unit = {
+  private def buildImage(name: String, dockerFile: String, buildArgs: String): Unit = {
     dockerClient.build(
       DOCKER_BUILD_PATH,
       name,
       dockerFile,
-      new LoggingBuildHandler())
+      new LoggingBuildHandler(),
+      DockerClient.BuildParam.create("buildargs", URLEncoder.encode(buildArgs, "UTF-8")))
     logInfo(s"Built $name docker image")
   }
 }