Clean up resourceSlices during exit

Signed-off-by: Carlos Eduardo Arango Gutierrez <[email protected]>

Author: ArangoGutierrez

cmd/nvidia-dra-controller/imex.go

@@ -19,6 +19,9 @@ package main
 import (
 	"context"
 	"fmt"
+	"os"
+	"os/signal"
+	"syscall"
 	"time"
 
 	v1 "k8s.io/api/core/v1"
@@ -94,6 +97,10 @@ func manageResourceSlices(ctx context.Context, clientset kubernetes.Interface, o
 		return fmt.Errorf("error starting resource slice controller: %w", err)
 	}
 
+	// Setup signal catching
+	sigs := make(chan os.Signal, 1)
+	signal.Notify(sigs, syscall.SIGTERM, syscall.SIGINT)
+
 	go func() {
 		for {
 			select {
@@ -109,7 +116,12 @@ func manageResourceSlices(ctx context.Context, clientset kubernetes.Interface, o
 				delete(newDriverResources.Pools, removedDomain)
 				controller.Update(&newDriverResources)
 				driverResources = newDriverResources
-			case <-ctx.Done():
+			case <-sigs:
+				controller.Stop()
+				err = cleanupImexChannels(ctx, clientset)
+				if err != nil {
+					klog.Errorf("error cleaning up resource slices: %v", err)
+				}
 				return
 			}
 		}
@@ -259,3 +271,25 @@ func generateImexChannelPool(imexDomain string, numChannels int) resourceslice.P
 
 	return pool
 }
+
+// cleanupImexChannels removes all resource slices created by the IMEX manager.
+func cleanupImexChannels(ctx context.Context, clientset kubernetes.Interface) error {
+	// Delete all resource slices created by the IMEX manager
+	ops := metav1.ListOptions{
+		FieldSelector: fmt.Sprintf("%s=%s", resourceapi.ResourceSliceSelectorDriver, DriverName),
+	}
+	l, err := clientset.ResourceV1alpha3().ResourceSlices().List(ctx, ops)
+	if err != nil {
+		return fmt.Errorf("error listing resource slices: %w", err)
+	}
+
+	for _, rs := range l.Items {
+		klog.Info("Deleting resource slice: ", rs.Name)
+		err := clientset.ResourceV1alpha3().ResourceSlices().Delete(ctx, rs.Name, metav1.DeleteOptions{})
+		if err != nil {
+			return fmt.Errorf("error deleting resource slice %s: %w", rs.Name, err)
+		}
+	}
+
+	return nil
+}