24 lines
482 B
Go
24 lines
482 B
Go
|
package dialid
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"github.com/ooni/probe-cli/v3/internal/engine/atomicx"
|
||
|
|
)
|
||
|
|
|
||
|
|
type contextkey struct{}
|
||
|
|
|
||
|
|
var id = atomicx.NewInt64()
|
||
|
|
|
||
|
|
// WithDialID returns a copy of ctx with DialID
|
||
|
|
func WithDialID(ctx context.Context) context.Context {
|
||
|
|
return context.WithValue(
|
||
|
|
ctx, contextkey{}, id.Add(1),
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
// ContextDialID returns the DialID of the context, or zero
|
||
|
|
func ContextDialID(ctx context.Context) int64 {
|
||
|
|
id, _ := ctx.Value(contextkey{}).(int64)
|
||
|
|
return id
|
||
|
|
}
|