fix WithNoDial to return the context

libp2p · Apr 10, 2019 · 0415ad3 · 0415ad3
1 parent 673b626
commit 0415ad3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/options.go b/options.go
@@ -10,8 +10,8 @@ var noDial = noDialCtxKey{}
 
 // WithNoDial constructs a new context with an option that instructs the network
 // to not attempt a new dial when opening a stream.
-func WithNoDial(ctx context.Context, reason string) {
-	context.WithValue(ctx, noDial, reason)
+func WithNoDial(ctx context.Context, reason string) context.Context {
+	return context.WithValue(ctx, noDial, reason)
 }
 
 // GetNoDial returns true if the no dial option is set in the context.