ping: use context.Afterfunc to avoid a lingering goroutine

libp2p · Mar 5, 2024 · 3999d9b · 3999d9b
1 parent 6130b9c
commit 3999d9b
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/p2p/protocol/ping/ping.go b/p2p/protocol/ping/ping.go
@@ -158,12 +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
 }