23 lines
507 B
Go
23 lines
507 B
Go
|
package micro
|
||
|
|
|
||
|
|
import "context"
|
||
|
|
|
||
|
|
type serviceKey struct{}
|
||
|
|
|
||
|
|
// FromContext retrieves a Service from the Context.
|
||
|
|
func FromContext(ctx context.Context) (Service, bool) {
|
||
|
|
if ctx == nil {
|
||
|
|
return nil, false
|
||
|
|
}
|
||
|
|
s, ok := ctx.Value(serviceKey{}).(Service)
|
||
|
|
return s, ok
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewContext returns a new Context with the Service embedded within it.
|
||
|
|
func NewContext(ctx context.Context, s Service) context.Context {
|
||
|
|
if ctx == nil {
|
||
|
|
ctx = context.Background()
|
||
|
|
}
|
||
|
|
return context.WithValue(ctx, serviceKey{}, s)
|
||
|
|
}
|