refactor(rm): migrate CheckHealth interface to context.Context

Update ResourceManager interface and implementations to use
context.Context for cancellation instead of a stop channel.
This is more idiomatic Go and allows for better lifecycle control.

Signed-off-by: Carlos Eduardo Arango Gutierrez <eduardoa@nvidia.com>

Author: ArangoGutierrez

internal/rm/nvml_manager.go

@@ -17,6 +17,7 @@
 package rm
 
 import (
+	"context"
 	"fmt"
 
 	"github.com/NVIDIA/go-gpuallocator/gpuallocator"
@@ -91,8 +92,8 @@ func (r *nvmlResourceManager) GetDevicePaths(ids []string) []string {
 }
 
 // CheckHealth performs health checks on a set of devices, writing to the 'unhealthy' channel with any unhealthy devices
-func (r *nvmlResourceManager) CheckHealth(stop <-chan interface{}, unhealthy chan<- *Device) error {
-	return r.checkHealth(stop, r.devices, unhealthy)
+func (r *nvmlResourceManager) CheckHealth(ctx context.Context, unhealthy chan<- *Device) error {
+	return r.checkHealth(ctx, r.devices, unhealthy)
 }
 
 // getPreferredAllocation runs an allocation algorithm over the inputs.

internal/rm/rm.go

@@ -17,6 +17,7 @@
 package rm
 
 import (
+	"context"
 	"errors"
 	"fmt"
 	"strings"
@@ -44,7 +45,7 @@ type ResourceManager interface {
 	Devices() Devices
 	GetDevicePaths([]string) []string
 	GetPreferredAllocation(available, required []string, size int) ([]string, error)
-	CheckHealth(stop <-chan interface{}, unhealthy chan<- *Device) error
+	CheckHealth(ctx context.Context, unhealthy chan<- *Device) error
 	ValidateRequest(AnnotatedIDs) error
 }
 

internal/rm/rm_mock.go

@@ -17,6 +17,7 @@
 package rm
 
 import (
+	"context"
 	"fmt"
 
 	spec "github.com/NVIDIA/k8s-device-plugin/api/config/v1"
@@ -71,6 +72,6 @@ func (r *tegraResourceManager) GetDevicePaths(ids []string) []string {
 }
 
 // CheckHealth is disabled for the tegraResourceManager
-func (r *tegraResourceManager) CheckHealth(stop <-chan interface{}, unhealthy chan<- *Device) error {
+func (r *tegraResourceManager) CheckHealth(_ context.Context, _ chan<- *Device) error {
 	return nil
 }