mirror of
https://github.com/PengZhangs2017582/novel.git
synced 2025-06-16 15:31:22 +08:00
50 lines
1.1 KiB
Go
50 lines
1.1 KiB
Go
package bizctx
|
||
|
||
import (
|
||
"context"
|
||
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
"github.com/gogf/gf/v2/net/ghttp"
|
||
|
||
"novel/internal/consts"
|
||
"novel/internal/model"
|
||
"novel/internal/service"
|
||
)
|
||
|
||
type sBizCtx struct{}
|
||
|
||
func init() {
|
||
service.RegisterBizCtx(New())
|
||
}
|
||
|
||
func New() *sBizCtx {
|
||
return &sBizCtx{}
|
||
}
|
||
|
||
// Init 初始化上下文对象指针到上下文对象中，以便后续的请求流程中可以修改。
|
||
func (s *sBizCtx) Init(r *ghttp.Request, customCtx *model.Context) {
|
||
r.SetCtxVar(consts.ContextKey, customCtx)
|
||
}
|
||
|
||
// Get 获得上下文变量，如果没有设置，那么返回nil
|
||
func (s *sBizCtx) Get(ctx context.Context) *model.Context {
|
||
value := ctx.Value(consts.ContextKey)
|
||
if value == nil {
|
||
return nil
|
||
}
|
||
if localCtx, ok := value.(*model.Context); ok {
|
||
return localCtx
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// SetUser 将上下文信息设置到上下文请求中，注意是完整覆盖
|
||
func (s *sBizCtx) SetUser(ctx context.Context, ctxUser *model.ContextUser) {
|
||
s.Get(ctx).User = ctxUser
|
||
}
|
||
|
||
// SetData 将上下文信息设置到上下文请求中，注意是完整覆盖
|
||
func (s *sBizCtx) SetData(ctx context.Context, data g.Map) {
|
||
s.Get(ctx).Data = data
|
||
}
|