ping: use context.Afterfunc to avoid a lingering goroutine

libp2p · Mar 5, 2024 · 229de81 · 229de81
1 parent 6130b9c
commit 229de81
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/p2p/protocol/ping/ping.go b/p2p/protocol/ping/ping.go
@@ -158,11 +158,10 @@ func Ping(ctx context.Context, h host.Host, p peer.ID) <-chan Result {
 			}
 		}
 	}()
-	go func() {
+	context.AfterFunc(ctx, func() {
 		// forces the ping to abort.
-		<-ctx.Done()
 		s.Reset()
-	}()
+	})
 
 	return out
 }