use context if available

sni · Apr 1, 2024 · cc21c29 · cc21c29
1 parent 83b79c3
commit cc21c29
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pkg/lmd/nodes.go b/pkg/lmd/nodes.go
@@ -255,7 +255,7 @@ func (n *Nodes) checkNodeAvailability(ctx context.Context) {
 		waitGroup.Add(1)
 		go func(waitGroup *sync.WaitGroup, node *NodeAddress) {
 			defer n.lmd.logPanicExit()
-			isOnline, forceRedistribute := n.sendPing(node, initializing, requestData)
+			isOnline, forceRedistribute := n.sendPing(ctx, node, initializing, requestData)
 			if forceRedistribute {
 				redistribute = true
 			}
@@ -528,10 +528,9 @@ func generateUUID() (uuid string) {
 	return
 }
 
-func (n *Nodes) sendPing(node *NodeAddress, initializing bool, requestData map[string]interface{}) (isOnline, forceRedistribute bool) {
+func (n *Nodes) sendPing(ctx context.Context, node *NodeAddress, initializing bool, requestData map[string]interface{}) (isOnline, forceRedistribute bool) {
 	done := make(chan bool)
 	ownIdentifier := n.ID
-	ctx := context.Background()
 	err := n.SendQuery(ctx, node, "ping", requestData, func(responseData interface{}) {
 		defer func() { done <- true }()
 		// Parse response