diff --git a/internal/controller/build.go b/internal/controller/build.go index 62f9429..b158c04 100644 --- a/internal/controller/build.go +++ b/internal/controller/build.go @@ -2,6 +2,7 @@ package controller import ( "context" + locustCluster "git.lilpenguins.com/crichardson/locust-operator/api/v1alpha1" "github.com/go-logr/logr" ) diff --git a/internal/controller/common.go b/internal/controller/common.go index ddd0482..127b353 100644 --- a/internal/controller/common.go +++ b/internal/controller/common.go @@ -2,8 +2,9 @@ package controller import ( "fmt" - "sigs.k8s.io/controller-runtime/pkg/client" "time" + + "sigs.k8s.io/controller-runtime/pkg/client" ) const ( diff --git a/internal/controller/locustcluster_controller.go b/internal/controller/locustcluster_controller.go index 7d471f6..60223f4 100644 --- a/internal/controller/locustcluster_controller.go +++ b/internal/controller/locustcluster_controller.go @@ -70,9 +70,14 @@ func (r *LocustClusterReconciler) Reconcile(ctx context.Context, req ctrl.Reques } podList, err := r.IsLeaderUp(log, locustCluster, ctx) + if err != nil { + return ctrl.Result{Requeue: true}, err + } - r.CreateUpdateLeader(log, locustCluster, ctx, podList) - + err = r.CreateUpdateLeader(log, locustCluster, ctx, podList) + if err != nil { + return ctrl.Result{Requeue: true}, err + } return ctrl.Result{}, nil }