Skip to content

Commit 226b1dc

Browse files
committed
Making docker -> container
1 parent 7f6d10a commit 226b1dc

File tree

7 files changed

+24
-25
lines changed

7 files changed

+24
-25
lines changed

resource-managers/kubernetes/core/src/main/scala/org/apache/spark/deploy/k8s/Config.scala

Lines changed: 8 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -30,21 +30,20 @@ private[spark] object Config extends Logging {
3030
.stringConf
3131
.createWithDefault("default")
3232

33-
val DRIVER_DOCKER_IMAGE =
34-
ConfigBuilder("spark.kubernetes.driver.docker.image")
35-
.doc("Docker image to use for the driver. Specify this using the standard Docker tag format.")
33+
val DRIVER_CONTAINER_IMAGE =
34+
ConfigBuilder("spark.kubernetes.driver.container.image")
35+
.doc("Container image to use for the driver.")
3636
.stringConf
3737
.createOptional
3838

39-
val EXECUTOR_DOCKER_IMAGE =
40-
ConfigBuilder("spark.kubernetes.executor.docker.image")
41-
.doc("Docker image to use for the executors. Specify this using the standard Docker tag " +
42-
"format.")
39+
val EXECUTOR_CONTAINER_IMAGE =
40+
ConfigBuilder("spark.kubernetes.executor.container.image")
41+
.doc("Container image to use for the executors.")
4342
.stringConf
4443
.createOptional
4544

46-
val DOCKER_IMAGE_PULL_POLICY =
47-
ConfigBuilder("spark.kubernetes.docker.image.pullPolicy")
45+
val CONTAINER_IMAGE_PULL_POLICY =
46+
ConfigBuilder("spark.kubernetes.container.image.pullPolicy")
4847
.doc("Kubernetes image pull policy. Valid values are Always, Never, and IfNotPresent.")
4948
.stringConf
5049
.checkValues(Set("Always", "Never", "IfNotPresent"))

resource-managers/kubernetes/core/src/main/scala/org/apache/spark/deploy/k8s/submit/DriverConfigurationStepsOrchestrator.scala

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ private[spark] class DriverConfigurationStepsOrchestrator(
4949
s"$appName-$uuid".toLowerCase.replaceAll("\\.", "-")
5050
}
5151

52-
private val dockerImagePullPolicy = submissionSparkConf.get(DOCKER_IMAGE_PULL_POLICY)
52+
private val imagePullPolicy = submissionSparkConf.get(CONTAINER_IMAGE_PULL_POLICY)
5353
private val jarsDownloadPath = submissionSparkConf.get(JARS_DOWNLOAD_LOCATION)
5454
private val filesDownloadPath = submissionSparkConf.get(FILES_DOWNLOAD_LOCATION)
5555

@@ -72,7 +72,7 @@ private[spark] class DriverConfigurationStepsOrchestrator(
7272
kubernetesAppId,
7373
kubernetesResourceNamePrefix,
7474
allDriverLabels,
75-
dockerImagePullPolicy,
75+
imagePullPolicy,
7676
appName,
7777
mainClass,
7878
appArgs,

resource-managers/kubernetes/core/src/main/scala/org/apache/spark/deploy/k8s/submit/steps/BaseDriverConfigurationStep.scala

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ private[spark] class BaseDriverConfigurationStep(
3434
kubernetesAppId: String,
3535
kubernetesResourceNamePrefix: String,
3636
driverLabels: Map[String, String],
37-
dockerImagePullPolicy: String,
37+
imagePullPolicy: String,
3838
appName: String,
3939
mainClass: String,
4040
appArgs: Array[String],
@@ -47,7 +47,7 @@ private[spark] class BaseDriverConfigurationStep(
4747
DRIVER_CLASS_PATH)
4848

4949
private val driverDockerImage = submissionSparkConf
50-
.get(DRIVER_DOCKER_IMAGE)
50+
.get(DRIVER_CONTAINER_IMAGE)
5151
.getOrElse(throw new SparkException("Must specify the driver Docker image"))
5252

5353
// CPU settings
@@ -111,7 +111,7 @@ private[spark] class BaseDriverConfigurationStep(
111111
val driverContainer = new ContainerBuilder(driverSpec.driverContainer)
112112
.withName(DRIVER_CONTAINER_NAME)
113113
.withImage(driverDockerImage)
114-
.withImagePullPolicy(dockerImagePullPolicy)
114+
.withImagePullPolicy(imagePullPolicy)
115115
.addAllToEnv(driverCustomEnvs.asJava)
116116
.addToEnv(driverExtraClasspathEnv.toSeq: _*)
117117
.addNewEnv()

resource-managers/kubernetes/core/src/main/scala/org/apache/spark/scheduler/cluster/k8s/ExecutorPodFactory.scala

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -72,10 +72,10 @@ private[spark] class ExecutorPodFactoryImpl(sparkConf: SparkConf)
7272
sparkConf,
7373
KUBERNETES_NODE_SELECTOR_PREFIX)
7474

75-
private val executorDockerImage = sparkConf
76-
.get(EXECUTOR_DOCKER_IMAGE)
77-
.getOrElse(throw new SparkException("Must specify the executor Docker image"))
78-
private val dockerImagePullPolicy = sparkConf.get(DOCKER_IMAGE_PULL_POLICY)
75+
private val executorContainerImage = sparkConf
76+
.get(EXECUTOR_CONTAINER_IMAGE)
77+
.getOrElse(throw new SparkException("Must specify the executor container image"))
78+
private val imagePullPolicy = sparkConf.get(CONTAINER_IMAGE_PULL_POLICY)
7979
private val blockManagerPort = sparkConf
8080
.getInt("spark.blockmanager.port", DEFAULT_BLOCKMANAGER_PORT)
8181

@@ -166,8 +166,8 @@ private[spark] class ExecutorPodFactoryImpl(sparkConf: SparkConf)
166166

167167
val executorContainer = new ContainerBuilder()
168168
.withName("executor")
169-
.withImage(executorDockerImage)
170-
.withImagePullPolicy(dockerImagePullPolicy)
169+
.withImage(executorContainerImage)
170+
.withImagePullPolicy(imagePullPolicy)
171171
.withNewResources()
172172
.addToRequests("memory", executorMemoryQuantity)
173173
.addToLimits("memory", executorMemoryLimitQuantity)

resource-managers/kubernetes/core/src/test/scala/org/apache/spark/deploy/k8s/submit/DriverConfigurationStepsOrchestratorSuite.scala

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
package org.apache.spark.deploy.k8s.submit
1818

1919
import org.apache.spark.{SparkConf, SparkFunSuite}
20-
import org.apache.spark.deploy.k8s.Config.DRIVER_DOCKER_IMAGE
20+
import org.apache.spark.deploy.k8s.Config.DRIVER_CONTAINER_IMAGE
2121
import org.apache.spark.deploy.k8s.submit.steps._
2222

2323
class DriverConfigurationStepsOrchestratorSuite extends SparkFunSuite {
@@ -32,7 +32,7 @@ class DriverConfigurationStepsOrchestratorSuite extends SparkFunSuite {
3232

3333
test("Base submission steps with a main app resource.") {
3434
val sparkConf = new SparkConf(false)
35-
.set(DRIVER_DOCKER_IMAGE, DRIVER_IMAGE)
35+
.set(DRIVER_CONTAINER_IMAGE, DRIVER_IMAGE)
3636
val mainAppResource = JavaMainAppResource("local:///var/apps/jars/main.jar")
3737
val orchestrator = new DriverConfigurationStepsOrchestrator(
3838
NAMESPACE,
@@ -54,7 +54,7 @@ class DriverConfigurationStepsOrchestratorSuite extends SparkFunSuite {
5454

5555
test("Base submission steps without a main app resource.") {
5656
val sparkConf = new SparkConf(false)
57-
.set(DRIVER_DOCKER_IMAGE, DRIVER_IMAGE)
57+
.set(DRIVER_CONTAINER_IMAGE, DRIVER_IMAGE)
5858
val orchestrator = new DriverConfigurationStepsOrchestrator(
5959
NAMESPACE,
6060
APP_ID,

resource-managers/kubernetes/core/src/test/scala/org/apache/spark/deploy/k8s/submit/steps/BaseDriverConfigurationStepSuite.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ class BaseDriverConfigurationStepSuite extends SparkFunSuite {
4747
.set(KUBERNETES_DRIVER_LIMIT_CORES, "4")
4848
.set(org.apache.spark.internal.config.DRIVER_MEMORY.key, "256M")
4949
.set(org.apache.spark.internal.config.DRIVER_MEMORY_OVERHEAD, 200L)
50-
.set(DRIVER_DOCKER_IMAGE, "spark-driver:latest")
50+
.set(DRIVER_CONTAINER_IMAGE, "spark-driver:latest")
5151
.set(s"$KUBERNETES_DRIVER_ANNOTATION_PREFIX$CUSTOM_ANNOTATION_KEY", CUSTOM_ANNOTATION_VALUE)
5252
.set(s"$KUBERNETES_DRIVER_ENV_KEY$DRIVER_CUSTOM_ENV_KEY1", "customDriverEnv1")
5353
.set(s"$KUBERNETES_DRIVER_ENV_KEY$DRIVER_CUSTOM_ENV_KEY2", "customDriverEnv2")

resource-managers/kubernetes/core/src/test/scala/org/apache/spark/scheduler/cluster/k8s/ExecutorPodFactorySuite.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ class ExecutorPodFactorySuite extends SparkFunSuite with BeforeAndAfter with Bef
5050
baseConf = new SparkConf()
5151
.set(KUBERNETES_DRIVER_POD_NAME, driverPodName)
5252
.set(KUBERNETES_EXECUTOR_POD_NAME_PREFIX, executorPrefix)
53-
.set(EXECUTOR_DOCKER_IMAGE, executorImage)
53+
.set(EXECUTOR_CONTAINER_IMAGE, executorImage)
5454
}
5555

5656
test("basic executor pod has reasonable defaults") {

0 commit comments

Comments
 (0)