12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455 |
- package base
- import (
- "github.com/gogf/gf/v2/net/ghttp"
- "golang.org/x/net/context"
- "nodeMonitor/internal/consts"
- "nodeMonitor/internal/model"
- )
- type IContext interface {
- Init(r *ghttp.Request, customCtx *model.Context)
- Get(ctx context.Context) *model.Context
- SetUser(ctx context.Context, ctxUser *model.ContextUser)
- GetLoginUser(ctx context.Context) *model.ContextUser
- }
- // Context 上下文管理服务
- var contextService = contextImpl{}
- type contextImpl struct{}
- func Context() IContext {
- return IContext(&contextService)
- }
- // Init 初始化上下文对象指针到上下文对象中,以便后续的请求流程中可以修改。
- func (s *contextImpl) Init(r *ghttp.Request, customCtx *model.Context) {
- r.SetCtxVar(consts.CtxKey, customCtx)
- }
- // Get 获得上下文变量,如果没有设置,那么返回nil
- func (s *contextImpl) Get(ctx context.Context) *model.Context {
- value := ctx.Value(consts.CtxKey)
- if value == nil {
- return nil
- }
- if localCtx, ok := value.(*model.Context); ok {
- return localCtx
- }
- return nil
- }
- // SetUser 将上下文信息设置到上下文请求中,注意是完整覆盖
- func (s *contextImpl) SetUser(ctx context.Context, ctxUser *model.ContextUser) {
- s.Get(ctx).User = ctxUser
- }
- // GetLoginUser 获取当前登陆用户信息
- func (s *contextImpl) GetLoginUser(ctx context.Context) *model.ContextUser {
- context := s.Get(ctx)
- if context == nil {
- return nil
- }
- return context.User
- }
|