From ccd05ae46402fa463c7294dc20e1dc3def11f6e6 Mon Sep 17 00:00:00 2001
From: Dan Molik <dan@danmolik.com>
Date: Fri, 2 Aug 2024 07:24:57 -0400
Subject: [PATCH] check balance nodes return

---
 internal/controller/valkey_controller.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/internal/controller/valkey_controller.go b/internal/controller/valkey_controller.go
index 5113944..48668f8 100644
--- a/internal/controller/valkey_controller.go
+++ b/internal/controller/valkey_controller.go
@@ -787,7 +787,9 @@ fi
 			return err
 		}
 		r.Recorder.Event(valkey, "Normal", "Updated", fmt.Sprintf("StatefulSet %s/%s is updated (replicas)", valkey.Namespace, valkey.Name))
-		r.balanceNodes(ctx, valkey, oldnodes, *sts.Spec.Replicas)
+		if err := r.balanceNodes(ctx, valkey, oldnodes, *sts.Spec.Replicas); err != nil {
+			return err
+		}
 	}
 
 	return nil