micro/config/source/service/options.go

39 lines
778 B
Go
Raw Normal View History

package service
2020-01-17 00:10:15 +08:00
import (
"context"
"github.com/micro/go-micro/v2/config/source"
2020-01-17 00:10:15 +08:00
)
type serviceNameKey struct{}
2020-01-20 18:31:18 +08:00
type keyKey struct{}
type pathKey struct{}
2020-01-17 00:10:15 +08:00
2020-01-20 18:31:18 +08:00
func ServiceName(name string) source.Option {
2020-01-17 00:10:15 +08:00
return func(o *source.Options) {
if o.Context == nil {
o.Context = context.Background()
}
2020-01-20 18:31:18 +08:00
o.Context = context.WithValue(o.Context, serviceNameKey{}, name)
}
}
func Key(key string) source.Option {
return func(o *source.Options) {
if o.Context == nil {
o.Context = context.Background()
}
o.Context = context.WithValue(o.Context, keyKey{}, key)
}
}
func Path(path string) source.Option {
return func(o *source.Options) {
if o.Context == nil {
o.Context = context.Background()
}
o.Context = context.WithValue(o.Context, pathKey{}, path)
2020-01-17 00:10:15 +08:00
}
}