check balance nodes return

hyperspike · Aug 2, 2024 · ccd05ae · ccd05ae
1 parent 152db68
commit ccd05ae
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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