- add accessUUID for Context

This commit is contained in:
gazebo
2019-02-13 17:20:41 +08:00
parent b2150a6938
commit 78e50f16d6

View File

@@ -4,6 +4,7 @@ import (
"net/http" "net/http"
"sync" "sync"
"git.rosy.net.cn/baseapi/utils"
"git.rosy.net.cn/jx-callback/business/jxcallback/auth" "git.rosy.net.cn/jx-callback/business/jxcallback/auth"
"git.rosy.net.cn/jx-callback/business/jxutils/tasksch" "git.rosy.net.cn/jx-callback/business/jxutils/tasksch"
"git.rosy.net.cn/jx-callback/business/model" "git.rosy.net.cn/jx-callback/business/model"
@@ -12,13 +13,14 @@ import (
) )
type Context struct { type Context struct {
rootTask tasksch.ITask rootTask tasksch.ITask
token string token string
userInfo *auth.LoginInfo accessUUID string
w http.ResponseWriter userInfo *auth.LoginInfo
r *http.Request w http.ResponseWriter
mapData map[interface{}]interface{} r *http.Request
locker sync.RWMutex mapData map[interface{}]interface{}
locker sync.RWMutex
} }
const ( const (
@@ -35,22 +37,24 @@ func init() {
func NewWithUserName(rootTask tasksch.ITask, userName string, w http.ResponseWriter, r *http.Request) (ctx *Context) { func NewWithUserName(rootTask tasksch.ITask, userName string, w http.ResponseWriter, r *http.Request) (ctx *Context) {
ctx = &Context{ ctx = &Context{
rootTask: rootTask, rootTask: rootTask,
token: userName, token: userName,
w: w, w: w,
r: r, r: r,
mapData: make(map[interface{}]interface{}), mapData: make(map[interface{}]interface{}),
accessUUID: utils.GetUUID(),
} }
return ctx return ctx
} }
func New(rootTask tasksch.ITask, token string, w http.ResponseWriter, r *http.Request) (ctx *Context, errCode string, err error) { func New(rootTask tasksch.ITask, token string, w http.ResponseWriter, r *http.Request) (ctx *Context, errCode string, err error) {
ctx = &Context{ ctx = &Context{
rootTask: rootTask, rootTask: rootTask,
token: token, token: token,
w: w, w: w,
r: r, r: r,
mapData: make(map[interface{}]interface{}), mapData: make(map[interface{}]interface{}),
accessUUID: utils.GetUUID(),
} }
ctx.userInfo, err = auth.GetUserInfo(token) ctx.userInfo, err = auth.GetUserInfo(token)
if err != nil && beego.BConfig.RunMode == "prod" { if err != nil && beego.BConfig.RunMode == "prod" {
@@ -90,6 +94,10 @@ func (ctx *Context) GetToken() string {
return ctx.token return ctx.token
} }
func (ctx *Context) GetAccessUUID() string {
return ctx.accessUUID
}
func (ctx *Context) GetLoginInfo() *auth.LoginInfo { func (ctx *Context) GetLoginInfo() *auth.LoginInfo {
return ctx.userInfo return ctx.userInfo
} }