You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
28 lines
495 B
Go
28 lines
495 B
Go
package contextx
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/zeromicro/go-zero/core/mapping"
|
|
)
|
|
|
|
const contextTagKey = "ctx"
|
|
|
|
var unmarshaler = mapping.NewUnmarshaler(contextTagKey)
|
|
|
|
type contextValuer struct {
|
|
context.Context
|
|
}
|
|
|
|
func (cv contextValuer) Value(key string) (interface{}, bool) {
|
|
v := cv.Context.Value(key)
|
|
return v, v != nil
|
|
}
|
|
|
|
// For unmarshals ctx into v.
|
|
func For(ctx context.Context, v interface{}) error {
|
|
return unmarshaler.UnmarshalValuer(contextValuer{
|
|
Context: ctx,
|
|
}, v)
|
|
}
|