From 59bcf0b6f064269f46c621f1773e8469bcb7ec8d Mon Sep 17 00:00:00 2001 From: iamabhishek-dubey Date: Fri, 7 Oct 2022 18:30:27 +0530 Subject: [PATCH] Added code for statefulset finalizer Signed-off-by: iamabhishek-dubey --- k8sutils/finalizer.go | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) diff --git a/k8sutils/finalizer.go b/k8sutils/finalizer.go index c01a96642..43bb3b927 100644 --- a/k8sutils/finalizer.go +++ b/k8sutils/finalizer.go @@ -34,6 +34,9 @@ func HandleRedisFinalizer(cr *redisv1beta1.Redis, cl client.Client) error { if err := finalizeRedisPVC(cr); err != nil { return err } + if err := finalizeRedisStatefulSet(cr); err != nil { + return err + } controllerutil.RemoveFinalizer(cr, RedisFinalizer) if err := cl.Update(context.TODO(), cr); err != nil { logger.Error(err, "Could not remove finalizer "+RedisFinalizer) @@ -55,6 +58,9 @@ func HandleRedisClusterFinalizer(cr *redisv1beta1.RedisCluster, cl client.Client if err := finalizeRedisClusterPVC(cr); err != nil { return err } + if err := finalizeRedisClusterStatefulSets(cr); err != nil { + return err + } controllerutil.RemoveFinalizer(cr, RedisClusterFinalizer) if err := cl.Update(context.TODO(), cr); err != nil { logger.Error(err, "Could not remove finalizer "+RedisClusterFinalizer) @@ -138,3 +144,27 @@ func finalizeRedisClusterPVC(cr *redisv1beta1.RedisCluster) error { } return nil } + +// finalizeRedisStatefulSet delete statefulset for Redis +func finalizeRedisStatefulSet(cr *redisv1beta1.Redis) error { + logger := finalizerLogger(cr.Namespace, RedisFinalizer) + err := generateK8sClient().AppsV1().StatefulSets(cr.Namespace).Delete(context.TODO(), cr.Name, metav1.DeleteOptions{}) + if err != nil && !errors.IsNotFound(err) { + logger.Error(err, "Could not delete StatefulSets "+cr.Name) + return err + } + return nil +} + +// finalizeRedisClusterStatefulSets delete statefulset for Redis Cluster +func finalizeRedisClusterStatefulSets(cr *redisv1beta1.RedisCluster) error { + logger := finalizerLogger(cr.Namespace, RedisClusterFinalizer) + for _, sts := range []string{cr.Name + "-leader", cr.Name + "-follower"} { + err := generateK8sClient().AppsV1().StatefulSets(cr.Namespace).Delete(context.TODO(), sts, metav1.DeleteOptions{}) + if err != nil && !errors.IsNotFound(err) { + logger.Error(err, "Could not delete statefulset "+sts) + return err + } + } + return nil +}