Auto recreate redis replication statefulset when update failed (#586)

Signed-off-by: drivebyer <[email protected]>
OT-CONTAINER-KIT · Aug 20, 2023 · c826f89 · c826f89
1 parent 97378b6
commit c826f89
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 2 deletions.
diff --git a/k8sutils/const.go b/k8sutils/const.go
@@ -0,0 +1,5 @@
+package k8sutils
+
+const (
+	AnnotationKeyRecreateStatefulset = "redis.opstreelabs.in/recreate-statefulset"
+)
diff --git a/k8sutils/redis-cluster.go b/k8sutils/redis-cluster.go
@@ -52,7 +52,7 @@ func generateRedisClusterParams(cr *redisv1beta1.RedisCluster, replicas int32, e
 	if externalConfig != nil {
 		res.ExternalConfig = externalConfig
 	}
-	if _, found := cr.ObjectMeta.GetAnnotations()["redis.opstreelabs.in/recreate-statefulset"]; found {
+	if _, found := cr.ObjectMeta.GetAnnotations()[AnnotationKeyRecreateStatefulset]; found {
 		res.RecreateStatefulSet = true
 	}
 	return res

diff --git a/k8sutils/redis-replication.go b/k8sutils/redis-replication.go
@@ -93,6 +93,9 @@ func generateRedisReplicationParams(cr *redisv1beta1.RedisReplication) statefulS
 	if cr.Spec.ServiceAccountName != nil {
 		res.ServiceAccountName = cr.Spec.ServiceAccountName
 	}
+	if _, found := cr.ObjectMeta.GetAnnotations()[AnnotationKeyRecreateStatefulset]; found {
+		res.RecreateStatefulSet = true
+	}
 	return res
 }
 

diff --git a/k8sutils/redis-standalone.go b/k8sutils/redis-standalone.go
@@ -98,7 +98,7 @@ func generateRedisStandaloneParams(cr *redisv1beta1.Redis) statefulSetParameters
 	if cr.Spec.ServiceAccountName != nil {
 		res.ServiceAccountName = cr.Spec.ServiceAccountName
 	}
-	if _, found := cr.ObjectMeta.GetAnnotations()["redis.opstreelabs.in/recreate-statefulset"]; found {
+	if _, found := cr.ObjectMeta.GetAnnotations()[AnnotationKeyRecreateStatefulset]; found {
 		res.RecreateStatefulSet = true
 	}
 	return res