Commit
…tRepos (#3102)
- Loading branch information
There are no files selected for viewing
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,233 @@ | ||
| package reconciler | ||
|
|
||
| import ( | ||
| "context" | ||
| "fmt" | ||
| "reflect" | ||
| "sort" | ||
|
|
||
| "github.com/rancher/fleet/internal/cmd/controller/summary" | ||
| "github.com/rancher/fleet/internal/resourcestatus" | ||
| v1alpha1 "github.com/rancher/fleet/pkg/apis/fleet.cattle.io/v1alpha1" | ||
| "github.com/rancher/fleet/pkg/durations" | ||
| "github.com/rancher/fleet/pkg/sharding" | ||
|
|
||
| "k8s.io/apimachinery/pkg/api/errors" | ||
| "k8s.io/apimachinery/pkg/runtime" | ||
| "k8s.io/apimachinery/pkg/types" | ||
| ctrl "sigs.k8s.io/controller-runtime" | ||
| "sigs.k8s.io/controller-runtime/pkg/builder" | ||
| "sigs.k8s.io/controller-runtime/pkg/client" | ||
| "sigs.k8s.io/controller-runtime/pkg/controller" | ||
| "sigs.k8s.io/controller-runtime/pkg/event" | ||
| "sigs.k8s.io/controller-runtime/pkg/handler" | ||
| "sigs.k8s.io/controller-runtime/pkg/log" | ||
| "sigs.k8s.io/controller-runtime/pkg/predicate" | ||
| ) | ||
|
|
||
| type StatusReconciler struct { | ||
| client.Client | ||
| Scheme *runtime.Scheme | ||
| Workers int | ||
| ShardID string | ||
| } | ||
|
|
||
| func (r *StatusReconciler) SetupWithManager(mgr ctrl.Manager) error { | ||
| return ctrl.NewControllerManagedBy(mgr). | ||
| For(&v1alpha1.GitRepo{}). | ||
| Watches( | ||
| // Fan out from bundle to gitrepo | ||
| &v1alpha1.Bundle{}, | ||
| handler.EnqueueRequestsFromMapFunc(func(ctx context.Context, a client.Object) []ctrl.Request { | ||
| repo := a.GetLabels()[v1alpha1.RepoLabel] | ||
| if repo != "" { | ||
| return []ctrl.Request{{ | ||
| NamespacedName: types.NamespacedName{ | ||
| Namespace: a.GetNamespace(), | ||
| Name: repo, | ||
| }, | ||
| }} | ||
| } | ||
|
|
||
| return []ctrl.Request{} | ||
| }), | ||
| builder.WithPredicates(bundleStatusChangedPredicate()), | ||
| ). | ||
| WithEventFilter(sharding.FilterByShardID(r.ShardID)). | ||
| WithOptions(controller.Options{MaxConcurrentReconciles: r.Workers}). | ||
| Named("GitRepoStatus"). | ||
| Complete(r) | ||
| } | ||
|
|
||
| // Reconcile reads the stat of the GitRepo and BundleDeployments and | ||
| // computes status fields for the GitRepo. This status is used to | ||
| // display information to the user. | ||
| func (r *StatusReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) { | ||
| logger := log.FromContext(ctx).WithName("gitops-status") | ||
| gitrepo := &v1alpha1.GitRepo{} | ||
|
|
||
| if err := r.Get(ctx, req.NamespacedName, gitrepo); err != nil && !errors.IsNotFound(err) { | ||
| return ctrl.Result{}, err | ||
| } else if errors.IsNotFound(err) { | ||
| logger.V(1).Info("Gitrepo deleted, cleaning up poll jobs") | ||
| return ctrl.Result{}, nil | ||
| } | ||
|
|
||
| // Restrictions / Overrides, gitrepo reconciler is responsible for setting error in status | ||
| if err := AuthorizeAndAssignDefaults(ctx, r.Client, gitrepo); err != nil { | ||
| // the gitjob_controller will handle the error | ||
| return ctrl.Result{}, nil | ||
| } | ||
|
|
||
| if !gitrepo.DeletionTimestamp.IsZero() { | ||
| // the gitjob_controller will handle deletion | ||
| return ctrl.Result{}, nil | ||
| } | ||
|
|
||
| if gitrepo.Spec.Repo == "" { | ||
| return ctrl.Result{}, nil | ||
| } | ||
|
|
||
| logger = logger.WithValues("generation", gitrepo.Generation, "commit", gitrepo.Status.Commit).WithValues("conditions", gitrepo.Status.Conditions) | ||
| ctx = log.IntoContext(ctx, logger) | ||
|
|
||
| logger.V(1).Info("Reconciling GitRepo status") | ||
|
|
||
| bdList := &v1alpha1.BundleDeploymentList{} | ||
| err := r.List(ctx, bdList, client.MatchingLabels{ | ||
| v1alpha1.RepoLabel: gitrepo.Name, | ||
| v1alpha1.BundleNamespaceLabel: gitrepo.Namespace, | ||
| }) | ||
| if err != nil { | ||
| return ctrl.Result{}, err | ||
| } | ||
|
|
||
| err = setStatus(bdList, gitrepo) | ||
| if err != nil { | ||
| return ctrl.Result{}, err | ||
| } | ||
|
|
||
| if gitrepo.Status.GitJobStatus != "Current" { | ||
| gitrepo.Status.Display.State = "GitUpdating" | ||
| } | ||
|
|
||
| err = r.Client.Status().Update(ctx, gitrepo) | ||
| if err != nil { | ||
| logger.Error(err, "Reconcile failed update to git repo status", "status", gitrepo.Status) | ||
| return ctrl.Result{RequeueAfter: durations.GitRepoStatusDelay}, nil | ||
|
Check failure on line 117 in internal/cmd/controller/gitops/reconciler/status_controller.go
|
||
| } | ||
|
|
||
| return ctrl.Result{}, nil | ||
| } | ||
|
|
||
| // bundleStatusChangedPredicate returns true if the bundle | ||
| // status has changed, or the bundle was created | ||
| func bundleStatusChangedPredicate() predicate.Funcs { | ||
|
Check failure on line 125 in internal/cmd/controller/gitops/reconciler/status_controller.go
|
||
| return predicate.Funcs{ | ||
| CreateFunc: func(e event.CreateEvent) bool { | ||
| return true | ||
| }, | ||
| UpdateFunc: func(e event.UpdateEvent) bool { | ||
| n, isBundle := e.ObjectNew.(*v1alpha1.Bundle) | ||
| if !isBundle { | ||
| return false | ||
| } | ||
| o := e.ObjectOld.(*v1alpha1.Bundle) | ||
| if n == nil || o == nil { | ||
| return false | ||
| } | ||
| return !reflect.DeepEqual(n.Status, o.Status) | ||
| }, | ||
| DeleteFunc: func(e event.DeleteEvent) bool { | ||
| return true | ||
| }, | ||
| } | ||
| } | ||
|
|
||
| func setStatus(list *v1alpha1.BundleDeploymentList, gitrepo *v1alpha1.GitRepo) error { | ||
| // sort for resourceKey? | ||
| sort.Slice(list.Items, func(i, j int) bool { | ||
| return list.Items[i].UID < list.Items[j].UID | ||
| }) | ||
|
|
||
| err := setFields(list, gitrepo) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| resourcestatus.SetGitRepoResources(list, gitrepo) | ||
|
|
||
| summary.SetReadyConditions(&gitrepo.Status, "Bundle", gitrepo.Status.Summary) | ||
|
|
||
| gitrepo.Status.Display.ReadyBundleDeployments = fmt.Sprintf("%d/%d", | ||
| gitrepo.Status.Summary.Ready, | ||
| gitrepo.Status.Summary.DesiredReady) | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| // setFields sets bundledeployment related status fields: | ||
| // Summary, ReadyClusters, DesiredReadyClusters, Display.State, Display.Message, Display.Error | ||
| func setFields(list *v1alpha1.BundleDeploymentList, gitrepo *v1alpha1.GitRepo) error { | ||
| var ( | ||
| maxState v1alpha1.BundleState | ||
| message string | ||
| count = map[client.ObjectKey]int{} | ||
| readyCount = map[client.ObjectKey]int{} | ||
| ) | ||
|
|
||
| gitrepo.Status.Summary = v1alpha1.BundleSummary{} | ||
|
|
||
| for _, bd := range list.Items { | ||
| state := summary.GetDeploymentState(&bd) | ||
| summary.IncrementState(&gitrepo.Status.Summary, bd.Name, state, summary.MessageFromDeployment(&bd), bd.Status.ModifiedStatus, bd.Status.NonReadyStatus) | ||
| gitrepo.Status.Summary.DesiredReady++ | ||
| if v1alpha1.StateRank[state] > v1alpha1.StateRank[maxState] { | ||
| maxState = state | ||
| message = summary.MessageFromDeployment(&bd) | ||
| } | ||
|
|
||
| // gather status per cluster | ||
| // try to avoid old bundle deployments, which might be missing the labels | ||
| if bd.Labels == nil { | ||
| // this should not happen | ||
| continue | ||
| } | ||
|
|
||
| name := bd.Labels[v1alpha1.ClusterLabel] | ||
| namespace := bd.Labels[v1alpha1.ClusterNamespaceLabel] | ||
| if name == "" || namespace == "" { | ||
| // this should not happen | ||
| continue | ||
| } | ||
|
|
||
| key := client.ObjectKey{Name: name, Namespace: namespace} | ||
| count[key]++ | ||
| if state == v1alpha1.Ready { | ||
| readyCount[key]++ | ||
| } | ||
| } | ||
|
|
||
| // unique number of clusters from bundledeployments | ||
| gitrepo.Status.DesiredReadyClusters = len(count) | ||
|
|
||
| // number of clusters where all deployments are ready | ||
| readyClusters := 0 | ||
| for key, n := range readyCount { | ||
| if count[key] == n { | ||
| readyClusters++ | ||
| } | ||
| } | ||
| gitrepo.Status.ReadyClusters = readyClusters | ||
|
|
||
| if maxState == v1alpha1.Ready { | ||
| maxState = "" | ||
| message = "" | ||
| } | ||
|
|
||
| gitrepo.Status.Display.State = string(maxState) | ||
| gitrepo.Status.Display.Message = message | ||
| gitrepo.Status.Display.Error = len(message) > 0 | ||
|
|
||
| return nil | ||
| } | ||
