Merge pull request #23 from libp2p/context-discover-gateway

pass a context to DiscoverGateway
libp2p · Sep 19, 2021 · 73b4c1e · 73b4c1e
2 parents 558869f + d0d64ea
commit 73b4c1e
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/nat.go b/nat.go
@@ -85,10 +85,7 @@ func DiscoverNATs(ctx context.Context) <-chan NAT {
 }
 
 // DiscoverGateway attempts to find a gateway device.
-func DiscoverGateway() (NAT, error) {
-	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
-	defer cancel()
-
+func DiscoverGateway(ctx context.Context) (NAT, error) {
 	var nats []NAT
 	for nat := range DiscoverNATs(ctx) {
 		nats = append(nats, nat)