check rate-limit before patch new partition (#884)

Signed-off-by: Kento Takeuchi <kento-takeuchi@cybozu.co.jp>

Author: lrf141

controllers/partition_controller.go

@@ -54,11 +54,6 @@ type StatefulSetPartitionReconciler struct {
 func (r *StatefulSetPartitionReconciler) Reconcile(ctx context.Context, req reconcile.Request) (reconcile.Result, error) {
 	log := crlog.FromContext(ctx)
 
-	if !r.RateLimiter.Allow() {
-		metrics.PartitionUpdateRetriesTotalVec.WithLabelValues(req.Namespace).Inc()
-		return reconcile.Result{RequeueAfter: r.UpdateInterval}, nil
-	}
-
 	sts := &appsv1.StatefulSet{}
 	err := r.Get(ctx, req.NamespacedName, sts)
 	if err != nil {
@@ -96,6 +91,11 @@ func (r *StatefulSetPartitionReconciler) Reconcile(ctx context.Context, req reco
 		return reconcile.Result{RequeueAfter: 30 * time.Second}, nil
 	}
 
+	if !r.RateLimiter.Allow() {
+		metrics.PartitionUpdateRetriesTotalVec.WithLabelValues(req.Namespace).Inc()
+		return reconcile.Result{RequeueAfter: r.UpdateInterval}, nil
+	}
+
 	if err := r.patchNewPartition(ctx, sts); err != nil {
 		log.Error(err, "failed to apply new partition")
 		return reconcile.Result{}, err