75 lines
1.7 KiB
Go
75 lines
1.7 KiB
Go
/*
|
||
* @desc:context-service
|
||
* @company:云南奇讯科技有限公司
|
||
* @Author: yixiaohu<yxh669@qq.com>
|
||
* @Date: 2022/9/23 14:51
|
||
*/
|
||
|
||
package context
|
||
|
||
import (
|
||
"context"
|
||
"github.com/gogf/gf/v2/net/ghttp"
|
||
"github.com/tiger1103/gfast/v3/internal/app/system/model"
|
||
"github.com/tiger1103/gfast/v3/internal/app/wechat/consts"
|
||
"github.com/tiger1103/gfast/v3/internal/app/wechat/service"
|
||
)
|
||
|
||
func init() {
|
||
service.RegisterContext(New())
|
||
}
|
||
|
||
type sContext struct{}
|
||
|
||
func New() service.IContext {
|
||
return &sContext{}
|
||
}
|
||
|
||
// Init 初始化上下文对象指针到上下文对象中，以便后续的请求流程中可以修改。
|
||
func (s *sContext) Init(r *ghttp.Request, customCtx *model.Context) {
|
||
r.SetCtxVar(consts.CtxKey, customCtx)
|
||
}
|
||
|
||
// Get 获得上下文变量，如果没有设置，那么返回nil
|
||
func (s *sContext) 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 *sContext) SetUser(ctx context.Context, ctxUser *model.ContextUser) {
|
||
s.Get(ctx).User = ctxUser
|
||
}
|
||
|
||
// GetLoginUser 获取当前登陆用户信息
|
||
func (s *sContext) GetLoginUser(ctx context.Context) *model.ContextUser {
|
||
context := s.Get(ctx)
|
||
if context == nil {
|
||
return nil
|
||
}
|
||
return context.User
|
||
}
|
||
|
||
// GetUserId 获取当前登录用户id
|
||
func (s *sContext) GetUserId(ctx context.Context) uint64 {
|
||
user := s.GetLoginUser(ctx)
|
||
if user != nil {
|
||
return user.Id
|
||
}
|
||
return 0
|
||
}
|
||
|
||
func (s *sContext) GetDeptId(ctx context.Context) uint64 {
|
||
user := s.GetLoginUser(ctx)
|
||
if user != nil {
|
||
return user.DeptId
|
||
}
|
||
return 0
|
||
}
|