79 lines
1.4 KiB
Go
79 lines
1.4 KiB
Go
package jxcontext
|
|
|
|
import (
|
|
"net/http"
|
|
"sync"
|
|
|
|
"git.rosy.net.cn/jx-callback/business/jxutils/tasksch"
|
|
)
|
|
|
|
type Context struct {
|
|
rootTask tasksch.ITask
|
|
token string
|
|
w http.ResponseWriter
|
|
r *http.Request
|
|
mapData map[interface{}]interface{}
|
|
locker sync.RWMutex
|
|
}
|
|
|
|
func New(rootTask tasksch.ITask, token string, w http.ResponseWriter, r *http.Request) (ctx *Context, errCode string, err error) {
|
|
return &Context{
|
|
rootTask: rootTask,
|
|
token: token,
|
|
w: w,
|
|
r: r,
|
|
mapData: make(map[interface{}]interface{}),
|
|
}, "", nil
|
|
}
|
|
|
|
func (ctx *Context) GetUserName() string {
|
|
return ctx.token
|
|
}
|
|
|
|
func (ctx *Context) GetUserID() string {
|
|
return ctx.token
|
|
}
|
|
|
|
func (ctx *Context) GetRootTask() tasksch.ITask {
|
|
ctx.locker.RLock()
|
|
defer ctx.locker.RUnlock()
|
|
|
|
return ctx.rootTask
|
|
}
|
|
|
|
func (ctx *Context) TaskCreated(task tasksch.ITask) bool {
|
|
ctx.locker.Lock()
|
|
|
|
if ctx.rootTask == nil {
|
|
ctx.rootTask = task
|
|
ctx.locker.Unlock()
|
|
return true
|
|
}
|
|
ctx.locker.Unlock()
|
|
|
|
ctx.rootTask.AddChild(task)
|
|
return false
|
|
}
|
|
|
|
func (ctx *Context) GetResponseWriter() http.ResponseWriter {
|
|
return ctx.w
|
|
}
|
|
|
|
func (ctx *Context) GetRequest() *http.Request {
|
|
return ctx.r
|
|
}
|
|
|
|
func (ctx *Context) SetValue(key, value interface{}) {
|
|
ctx.locker.Lock()
|
|
defer ctx.locker.Unlock()
|
|
|
|
ctx.mapData[key] = value
|
|
}
|
|
|
|
func (ctx *Context) GetValue(key interface{}) interface{} {
|
|
ctx.locker.RLock()
|
|
defer ctx.locker.RUnlock()
|
|
|
|
return ctx.mapData[key]
|
|
}
|