@@ -30,6 +30,7 @@ import (
3030 "sigs.k8s.io/controller-runtime/pkg/log"
3131)
3232
33+ //+kubebuilder:rbac:groups="core",resources=namespaces,verbs=get;list;patch;watch
3334//+kubebuilder:rbac:groups="core",resources=nodes,verbs=get;watch
3435//+kubebuilder:rbac:groups=kmm.sigs.x-k8s.io,resources=nodemodulesconfigs,verbs=get;list;watch;patch;create;delete
3536
@@ -47,6 +48,7 @@ type schedulingData struct {
4748
4849type ModuleNMCReconciler struct {
4950 filter * filter.Filter
51+ nsLabeler namespaceLabeler
5052 reconHelper moduleNMCReconcilerHelperAPI
5153}
5254
@@ -59,6 +61,7 @@ func NewModuleNMCReconciler(client client.Client,
5961 reconHelper := newModuleNMCReconcilerHelper (client , kernelAPI , registryAPI , nmcHelper , scheme )
6062 return & ModuleNMCReconciler {
6163 filter : filter ,
64+ nsLabeler : newNamespaceLabeler (client ),
6265 reconHelper : reconHelper ,
6366 }
6467}
@@ -78,13 +81,21 @@ func (mnr *ModuleNMCReconciler) Reconcile(ctx context.Context, req ctrl.Request)
7881 }
7982 if mod .GetDeletionTimestamp () != nil {
8083 //Module is being deleted
84+ if err = mnr .nsLabeler .tryRemovingLabel (ctx , mod .Namespace , mod .Name ); err != nil {
85+ return ctrl.Result {}, fmt .Errorf ("error while trying to remove the label on namespace %s: %v" , mod .Namespace , err )
86+ }
87+
8188 err = mnr .reconHelper .finalizeModule (ctx , mod )
8289 if err != nil {
8390 return ctrl.Result {}, fmt .Errorf ("failed to finalize %s Module: %v" , req .NamespacedName , err )
8491 }
8592 return ctrl.Result {}, nil
8693 }
8794
95+ if err = mnr .nsLabeler .setLabel (ctx , mod .Namespace ); err != nil {
96+ return ctrl.Result {}, fmt .Errorf ("could not set label %q on namespace %s: %v" , constants .NamespaceLabelKey , mod .Namespace , err )
97+ }
98+
8899 err = mnr .reconHelper .setFinalizerAndStatus (ctx , mod )
89100 if err != nil {
90101 return ctrl.Result {}, fmt .Errorf ("failed to set finalizer on %s Module: %v" , req .NamespacedName , err )
@@ -145,7 +156,7 @@ func (mnr *ModuleNMCReconciler) SetupWithManager(mgr ctrl.Manager) error {
145156 Complete (mnr )
146157}
147158
148- //go:generate mockgen -source=module_nmc_reconciler.go -package=controllers -destination=mock_module_nmc_reconciler.go moduleNMCReconcilerHelperAPI
159+ //go:generate mockgen -source=module_nmc_reconciler.go -package=controllers -destination=mock_module_nmc_reconciler.go moduleNMCReconcilerHelperAPI,namespaceLabeler
149160
150161type moduleNMCReconcilerHelperAPI interface {
151162 setFinalizerAndStatus (ctx context.Context , mod * kmmv1beta1.Module ) error
@@ -439,6 +450,80 @@ func (mnrh *moduleNMCReconcilerHelper) moduleUpdateWorkerPodsStatus(ctx context.
439450 return mnrh .client .Status ().Patch (ctx , mod , client .MergeFrom (unmodifiedMod ))
440451}
441452
453+ type namespaceLabeler interface {
454+ setLabel (ctx context.Context , name string ) error
455+ tryRemovingLabel (ctx context.Context , name , moduleName string ) error
456+ }
457+
458+ type namespaceLabelerImpl struct {
459+ client client.Client
460+ }
461+
462+ func newNamespaceLabeler (client client.Client ) namespaceLabeler {
463+ return & namespaceLabelerImpl {client : client }
464+ }
465+
466+ func (h * namespaceLabelerImpl ) setLabel (ctx context.Context , name string ) error {
467+ logger := log .FromContext (ctx )
468+
469+ ns := v1.Namespace {}
470+
471+ if err := h .client .Get (ctx , types.NamespacedName {Name : name }, & ns ); err != nil {
472+ return fmt .Errorf ("could not get namespace %s: %v" , name , err )
473+ }
474+
475+ if ! meta .HasLabel (& ns , constants .NamespaceLabelKey ) {
476+ nsCopy := ns .DeepCopy ()
477+
478+ logger .Info ("Setting namespace label" )
479+ meta .SetLabel (& ns , constants .NamespaceLabelKey , "" )
480+
481+ return h .client .Patch (ctx , & ns , client .MergeFrom (nsCopy ))
482+ }
483+
484+ return nil
485+ }
486+
487+ func (h * namespaceLabelerImpl ) tryRemovingLabel (ctx context.Context , name , moduleName string ) error {
488+ logger := log .FromContext (ctx )
489+
490+ modList := kmmv1beta1.ModuleList {}
491+
492+ opt := client .InNamespace (name )
493+
494+ if err := h .client .List (ctx , & modList , opt ); err != nil {
495+ return fmt .Errorf ("could not list modules in namespace %s: %v" , name , err )
496+ }
497+
498+ if count := len (modList .Items ); count > 1 {
499+ logger .Info ("Namespace still contains modules; not removing the label" , "count" , count )
500+
501+ if verboseLogger := logger .V (1 ); verboseLogger .Enabled () {
502+ modNames := make ([]string , 0 , count )
503+
504+ for _ , m := range modList .Items {
505+ modNames = append (modNames , m .Name )
506+ }
507+
508+ verboseLogger .Info ("Remaining modules" , "names" , modNames )
509+ }
510+
511+ return nil
512+ }
513+
514+ ns := & v1.Namespace {}
515+
516+ if err := h .client .Get (ctx , types.NamespacedName {Name : name }, ns ); err != nil {
517+ return fmt .Errorf ("could not get namespace %s: %v" , name , err )
518+ }
519+
520+ nsCopy := ns .DeepCopy ()
521+
522+ meta .RemoveLabel (ns , constants .NamespaceLabelKey )
523+
524+ return h .client .Patch (ctx , ns , client .MergeFrom (nsCopy ))
525+ }
526+
442527func prepareNodeSchedulingData (node v1.Node , mld * api.ModuleLoaderData , currentNMCs sets.Set [string ]) schedulingData {
443528 versionLabel := ""
444529 present := false
0 commit comments