From 63b197d83c7d055280f550d50f3216cf695666b2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?S=C3=B6ren=20Henning?= <soeren.henning@jku.at>
Date: Wed, 20 Mar 2024 14:14:47 +0100
Subject: [PATCH] Add pullPolicy configuration for image patcher
---
.../theodolite/kubernetes/patcher/ImagePatcher.kt | 11 ++++++++++-
.../theodolite/kubernetes/patcher/PatcherFactory.kt | 3 ++-
2 files changed, 12 insertions(+), 2 deletions(-)
diff --git a/theodolite/src/main/kotlin/rocks/theodolite/kubernetes/patcher/ImagePatcher.kt b/theodolite/src/main/kotlin/rocks/theodolite/kubernetes/patcher/ImagePatcher.kt
index ce23aaa0..ea1b6160 100644
--- a/theodolite/src/main/kotlin/rocks/theodolite/kubernetes/patcher/ImagePatcher.kt
+++ b/theodolite/src/main/kotlin/rocks/theodolite/kubernetes/patcher/ImagePatcher.kt
@@ -10,23 +10,32 @@ import io.fabric8.kubernetes.api.model.apps.StatefulSet
*
* @param container Container to be patched.
*/
-class ImagePatcher(private val container: String) : AbstractStringPatcher() {
+class ImagePatcher(private val container: String, private val pullPolicy: String? = null) : AbstractStringPatcher() {
override fun patchSingleResource(resource: HasMetadata, value: String): HasMetadata {
when (resource) {
is Deployment -> {
resource.spec.template.spec.containers.filter { it.name == container }.forEach {
it.image = value
+ if (pullPolicy != null) {
+ it.imagePullPolicy = pullPolicy
+ }
}
}
is StatefulSet -> {
resource.spec.template.spec.containers.filter { it.name == container }.forEach {
it.image = value
+ if (pullPolicy != null) {
+ it.imagePullPolicy = pullPolicy
+ }
}
}
is Pod -> {
resource.spec.containers.filter { it.name == container }.forEach {
it.image = value
+ if (pullPolicy != null) {
+ it.imagePullPolicy = pullPolicy
+ }
}
}
}
diff --git a/theodolite/src/main/kotlin/rocks/theodolite/kubernetes/patcher/PatcherFactory.kt b/theodolite/src/main/kotlin/rocks/theodolite/kubernetes/patcher/PatcherFactory.kt
index d996b545..7d47dc8b 100644
--- a/theodolite/src/main/kotlin/rocks/theodolite/kubernetes/patcher/PatcherFactory.kt
+++ b/theodolite/src/main/kotlin/rocks/theodolite/kubernetes/patcher/PatcherFactory.kt
@@ -75,7 +75,8 @@ class PatcherFactory {
?: throwInvalid(patcher)
)
"ImagePatcher" -> ImagePatcher(
- container = patcher.properties["container"] ?: throwInvalid(patcher)
+ container = patcher.properties["container"] ?: throwInvalid(patcher),
+ pullPolicy = patcher.properties["pullPolicy"]
)
"ConfigMapYamlPatcher" -> DecoratingPatcher(
ConfigMapYamlPatcher(
--
GitLab