Merge pull request #227 from mfreeman451/updates/snmp_checker_ctx

cleanup
carverauto · Feb 20, 2025 · 79019da · 79019da
2 parents 171343b + 0552e7b
commit 79019da
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pkg/agent/registry.go b/pkg/agent/registry.go
@@ -47,12 +47,12 @@ func initRegistry() checker.Registry {
 	})
 
 	// Register the SNMP checker
-	registry.Register("snmp", func(_ context.Context, _, details string) (checker.Checker, error) {
+	registry.Register("snmp", func(ctx context.Context, _, details string) (checker.Checker, error) {
 		if details == "" {
 			return nil, errDetailsRequiredSNMP
 		}
 
-		return NewSNMPChecker(details)
+		return NewSNMPChecker(ctx, details)
 	})
 
 	return registry

diff --git a/pkg/agent/snmp_checker.go b/pkg/agent/snmp_checker.go
@@ -35,7 +35,7 @@ type SNMPChecker struct {
 }
 
 // NewSNMPChecker creates a new SNMP checker that connects to an external SNMP checker process.
-func NewSNMPChecker(address string) (checker.Checker, error) {
+func NewSNMPChecker(ctx context.Context, address string) (checker.Checker, error) {
 	log.Printf("Creating new SNMP checker client for address: %s", address)
 
 	// Load configuration
@@ -51,7 +51,7 @@ func NewSNMPChecker(address string) (checker.Checker, error) {
 
 	// Create gRPC client connection to the SNMP checker process
 	client, err := grpc.NewClient(
-		context.Background(),
+		ctx,
 		address,
 		grpc.WithMaxRetries(grpcRetries),
 	)