diff --git a/context.go b/context.go index dd19b436f..84cd2396b 100644 --- a/context.go +++ b/context.go @@ -9,12 +9,11 @@ import ( // V2rayKey is the key type of Instance in Context, exported for test. type V2rayKey int -// V2rayKeyValue is the key value of Instance in Context, exported for test. -const V2rayKeyValue V2rayKey = 1 +const v2rayKey V2rayKey = 1 // FromContext returns an Instance from the given context, or nil if the context doesn't contain one. func FromContext(ctx context.Context) *Instance { - if s, ok := ctx.Value(V2rayKeyValue).(*Instance); ok { + if s, ok := ctx.Value(v2rayKey).(*Instance); ok { return s } return nil diff --git a/functions.go b/functions.go index 178e5dcbe..d08091d8e 100644 --- a/functions.go +++ b/functions.go @@ -16,7 +16,7 @@ import ( func CreateObject(v *Instance, config interface{}) (interface{}, error) { var ctx context.Context if v != nil { - ctx = context.WithValue(v.ctx, V2rayKeyValue, v) + ctx = context.WithValue(v.ctx, v2rayKey, v) } return common.CreateObject(ctx, config) }