Init job creates from InstanceSpec image if specified (#336)

* Init job creates from PrimaryMasters.InstanceSpec image if specified * Add helper function for image selection * change helper image func specification
ytsaurus · Sep 12, 2024 · d211c4a · d211c4a
1 parent 49534d9
commit d211c4a
Show file tree

Hide file tree

Showing 8 changed files with 17 additions and 20 deletions.
diff --git a/pkg/components/helpers.go b/pkg/components/helpers.go
@@ -227,3 +227,10 @@ func AddInitContainersToPodSpec(initContainers []string, podSpec *corev1.PodSpec
 	podSpec.InitContainers = append(containers, podSpec.InitContainers...)
 	return nil
 }
+
+func getImageWithDefault(componentImage *string, defaultImage string) string {
+	if componentImage != nil {
+		return *componentImage
+	}
+	return defaultImage
+}
diff --git a/pkg/components/master.go b/pkg/components/master.go
@@ -69,7 +69,7 @@ func NewMaster(cfgen *ytconfig.Generator, ytsaurus *apiproxy.Ytsaurus) *Master {
 		resource.Spec.ImagePullSecrets,
 		"default",
 		consts.ClientConfigFileName,
-		resource.Spec.CoreImage,
+		getImageWithDefault(resource.Spec.PrimaryMasters.InstanceSpec.Image, resource.Spec.CoreImage),
 		cfgen.GetNativeClientConfig)
 
 	exitReadOnlyJob := NewInitJob(
@@ -79,7 +79,7 @@ func NewMaster(cfgen *ytconfig.Generator, ytsaurus *apiproxy.Ytsaurus) *Master {
 		resource.Spec.ImagePullSecrets,
 		"exit-read-only",
 		consts.ClientConfigFileName,
-		resource.Spec.CoreImage,
+		getImageWithDefault(resource.Spec.PrimaryMasters.InstanceSpec.Image, resource.Spec.CoreImage),
 		cfgen.GetNativeClientConfig,
 	)
 

diff --git a/pkg/components/query_tracker.go b/pkg/components/query_tracker.go
@@ -66,11 +66,6 @@ func NewQueryTracker(
 		}),
 	)
 
-	image := ytsaurus.GetResource().Spec.CoreImage
-	if resource.Spec.QueryTrackers.InstanceSpec.Image != nil {
-		image = *resource.Spec.QueryTrackers.InstanceSpec.Image
-	}
-
 	return &QueryTracker{
 		localServerComponent: newLocalServerComponent(&l, ytsaurus, srv),
 		cfgen:                cfgen,
@@ -84,7 +79,7 @@ func NewQueryTracker(
 			resource.Spec.ImagePullSecrets,
 			"qt-state",
 			consts.ClientConfigFileName,
-			image,
+			getImageWithDefault(resource.Spec.QueryTrackers.InstanceSpec.Image, resource.Spec.CoreImage),
 			cfgen.GetNativeClientConfig),
 		secret: resources.NewStringSecret(
 			l.GetSecretName(),

diff --git a/pkg/components/queue_agent.go b/pkg/components/queue_agent.go
@@ -68,11 +68,6 @@ func NewQueueAgent(
 		}),
 	)
 
-	image := ytsaurus.GetResource().Spec.CoreImage
-	if resource.Spec.QueueAgents.InstanceSpec.Image != nil {
-		image = *resource.Spec.QueueAgents.InstanceSpec.Image
-	}
-
 	return &QueueAgent{
 		localServerComponent: newLocalServerComponent(&l, ytsaurus, srv),
 		cfgen:                cfgen,
@@ -87,7 +82,7 @@ func NewQueueAgent(
 			resource.Spec.ImagePullSecrets,
 			"qa-state",
 			consts.ClientConfigFileName,
-			image,
+			getImageWithDefault(resource.Spec.QueueAgents.InstanceSpec.Image, resource.Spec.CoreImage),
 			cfgen.GetNativeClientConfig),
 		secret: resources.NewStringSecret(
 			l.GetSecretName(),

diff --git a/pkg/components/scheduler.go b/pkg/components/scheduler.go
@@ -79,7 +79,7 @@ func NewScheduler(
 			resource.Spec.ImagePullSecrets,
 			"user",
 			consts.ClientConfigFileName,
-			resource.Spec.CoreImage,
+			getImageWithDefault(resource.Spec.Schedulers.InstanceSpec.Image, resource.Spec.CoreImage),
 			cfgen.GetNativeClientConfig),
 		initOpArchive: NewInitJob(
 			&l,
@@ -88,7 +88,7 @@ func NewScheduler(
 			resource.Spec.ImagePullSecrets,
 			"op-archive",
 			consts.ClientConfigFileName,
-			resource.Spec.CoreImage,
+			getImageWithDefault(resource.Spec.Schedulers.InstanceSpec.Image, resource.Spec.CoreImage),
 			cfgen.GetNativeClientConfig),
 		secret: resources.NewStringSecret(
 			l.GetSecretName(),

diff --git a/pkg/components/strawberry_controller.go b/pkg/components/strawberry_controller.go
@@ -83,7 +83,7 @@ func NewStrawberryController(
 			ytsaurus.GetResource().Spec.ImagePullSecrets,
 			"user",
 			consts.ClientConfigFileName,
-			resource.Spec.CoreImage,
+			getImageWithDefault(resource.Spec.StrawberryController.Image, resource.Spec.CoreImage),
 			cfgen.GetNativeClientConfig),
 		initChytClusterJob: NewInitJob(
 			&l,
@@ -92,7 +92,7 @@ func NewStrawberryController(
 			resource.Spec.ImagePullSecrets,
 			"cluster",
 			ChytInitClusterJobConfigFileName,
-			image,
+			getImageWithDefault(resource.Spec.StrawberryController.Image, resource.Spec.CoreImage),
 			cfgen.GetChytInitClusterConfig),
 		secret: resources.NewStringSecret(
 			l.GetSecretName(),

diff --git a/pkg/components/ui.go b/pkg/components/ui.go
@@ -83,7 +83,7 @@ func NewUI(cfgen *ytconfig.Generator, ytsaurus *apiproxy.Ytsaurus, master Compon
 			resource.Spec.ImagePullSecrets,
 			"default",
 			consts.ClientConfigFileName,
-			resource.Spec.CoreImage,
+			getImageWithDefault(resource.Spec.UI.Image, resource.Spec.CoreImage),
 			cfgen.GetNativeClientConfig),
 		secret: resources.NewStringSecret(
 			l.GetSecretName(),

diff --git a/pkg/components/yql_agent.go b/pkg/components/yql_agent.go
@@ -68,7 +68,7 @@ func NewYQLAgent(cfgen *ytconfig.Generator, ytsaurus *apiproxy.Ytsaurus, master
 			resource.Spec.ImagePullSecrets,
 			"yql-agent-environment",
 			consts.ClientConfigFileName,
-			resource.Spec.CoreImage,
+			getImageWithDefault(resource.Spec.YQLAgents.Image, resource.Spec.CoreImage),
 			cfgen.GetNativeClientConfig),
 		secret: resources.NewStringSecret(
 			l.GetSecretName(),