init project
This commit is contained in:
@@ -0,0 +1,40 @@
|
||||
package request
|
||||
|
||||
import "context"
|
||||
|
||||
type requestInfoKey int
|
||||
|
||||
const (
|
||||
key requestInfoKey = iota
|
||||
)
|
||||
|
||||
type RequestInfo struct {
|
||||
UserId int64
|
||||
TraceId string
|
||||
Permissions map[string]bool
|
||||
Role string
|
||||
}
|
||||
|
||||
func WithRequestInfo(ctx context.Context, info RequestInfo) context.Context {
|
||||
return context.WithValue(ctx, key, info)
|
||||
}
|
||||
|
||||
func GetRequestInfo(ctx context.Context) (requestInfo RequestInfo, ok bool) {
|
||||
requestInfo, ok = ctx.Value(key).(RequestInfo)
|
||||
return
|
||||
}
|
||||
|
||||
type reqInfoKeyType = string
|
||||
|
||||
const (
|
||||
reqInfoKey reqInfoKeyType = "request-info"
|
||||
)
|
||||
|
||||
func GetReqInfo(c context.Context) RequestInfo {
|
||||
info := c.Value(reqInfoKey)
|
||||
if info != nil {
|
||||
return info.(RequestInfo)
|
||||
}
|
||||
|
||||
return RequestInfo{}
|
||||
}
|
||||
Reference in New Issue
Block a user