libp2p/go-libp2p-core
1
0
Fork 0
mirror of https://github.com/libp2p/go-libp2p-core.git synced 2024-12-27 23:40:11 +08:00
Code Issues Projects Releases Wiki Activity

make UseTransient context option take a reason argument, for consistency with other options

Browse Source
This commit is contained in:
vyzo 2021-02-17 11:19:56 +02:00
parent a21b06e646
commit 980a554aa5
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
network/context.go
View File

@ -71,15 +71,15 @@ func WithDialPeerTimeout(ctx context.Context, timeout time.Duration) context.Con
// WithUseTransient constructs a new context with an option that instructs the network // WithUseTransient constructs a new context with an option that instructs the network
// that it is acceptable to use a transient connection when opening a new stream. // that it is acceptable to use a transient connection when opening a new stream.
func WithUseTransient(ctx context.Context) context.Context { func WithUseTransient(ctx context.Context, reason string) context.Context {
return context.WithValue(ctx, useTransient, true) return context.WithValue(ctx, useTransient, reason)
} }
// GetUseTransient returns true if the use transient option is set in the context. // GetUseTransient returns true if the use transient option is set in the context.
func GetUseTransient(ctx context.Context) bool { func GetUseTransient(ctx context.Context) (usetransient bool, reason string) {
v := ctx.Value(useTransient) v := ctx.Value(useTransient)
if v != nil { if v != nil {
return true return true, v.(string)
} }
return false return false, ""
} }