121 lines
3.5 KiB
Go
121 lines
3.5 KiB
Go
package rest
|
||
|
||
import (
|
||
"fmt"
|
||
"github.com/eyebluecn/tank/code/core"
|
||
"github.com/eyebluecn/tank/code/tool/result"
|
||
"github.com/json-iterator/go"
|
||
"go/types"
|
||
"net/http"
|
||
)
|
||
|
||
type BaseController struct {
|
||
BaseBean
|
||
userDao *UserDao
|
||
sessionDao *SessionDao
|
||
}
|
||
|
||
func (this *BaseController) Init() {
|
||
|
||
this.BaseBean.Init()
|
||
|
||
//手动装填本实例的Bean.
|
||
b := core.CONTEXT.GetBean(this.userDao)
|
||
if b, ok := b.(*UserDao); ok {
|
||
this.userDao = b
|
||
}
|
||
|
||
b = core.CONTEXT.GetBean(this.sessionDao)
|
||
if b, ok := b.(*SessionDao); ok {
|
||
this.sessionDao = b
|
||
}
|
||
|
||
}
|
||
|
||
//注册自己的路由。
|
||
func (this *BaseController) RegisterRoutes() map[string]func(writer http.ResponseWriter, request *http.Request) {
|
||
//每个Controller需要主动注册自己的路由。
|
||
return make(map[string]func(writer http.ResponseWriter, request *http.Request))
|
||
}
|
||
|
||
//处理一些特殊的接口，比如参数包含在路径中,一般情况下，controller不将参数放在url路径中
|
||
func (this *BaseController) HandleRoutes(writer http.ResponseWriter, request *http.Request) (func(writer http.ResponseWriter, request *http.Request), bool) {
|
||
return nil, false
|
||
}
|
||
|
||
//需要进行登录验证的wrap包装
|
||
func (this *BaseController) Wrap(f func(writer http.ResponseWriter, request *http.Request) *result.WebResult, qualifiedRole string) func(w http.ResponseWriter, r *http.Request) {
|
||
|
||
return func(writer http.ResponseWriter, request *http.Request) {
|
||
|
||
//writer和request赋值给自己。
|
||
|
||
var webResult *result.WebResult = nil
|
||
|
||
//只有游客接口不需要登录
|
||
if qualifiedRole != USER_ROLE_GUEST {
|
||
user := this.checkUser(writer, request)
|
||
|
||
if user.Status == USER_STATUS_DISABLED {
|
||
//判断用户是否被禁用。
|
||
webResult = result.ConstWebResult(result.USER_DISABLED)
|
||
} else {
|
||
if qualifiedRole == USER_ROLE_ADMINISTRATOR && user.Role != USER_ROLE_ADMINISTRATOR {
|
||
webResult = result.ConstWebResult(result.UNAUTHORIZED)
|
||
} else {
|
||
webResult = f(writer, request)
|
||
}
|
||
}
|
||
|
||
} else {
|
||
webResult = f(writer, request)
|
||
}
|
||
|
||
//输出的是json格式
|
||
if webResult != nil {
|
||
//返回的内容申明是json，utf-8
|
||
writer.Header().Set("Content-Type", "application/json;charset=UTF-8")
|
||
|
||
//用json的方式输出返回值。
|
||
b, err := jsoniter.ConfigCompatibleWithStandardLibrary.Marshal(webResult)
|
||
|
||
this.PanicError(err)
|
||
|
||
writer.WriteHeader(result.FetchHttpStatus(webResult.Code))
|
||
|
||
_, err = fmt.Fprintf(writer, string(b))
|
||
this.PanicError(err)
|
||
} else {
|
||
//输出的内容是二进制的。
|
||
|
||
}
|
||
|
||
}
|
||
}
|
||
|
||
//返回成功的结果。支持放置三种类型 1.字符串 2. WebResult对象 3.空指针 4.任意类型
|
||
func (this *BaseController) Success(data interface{}) *result.WebResult {
|
||
var webResult *result.WebResult = nil
|
||
if value, ok := data.(string); ok {
|
||
//返回一句普通的消息
|
||
webResult = &result.WebResult{Code: result.OK.Code, Msg: value}
|
||
} else if value, ok := data.(*result.WebResult); ok {
|
||
//返回一个webResult对象
|
||
webResult = value
|
||
} else if _, ok := data.(types.Nil); ok {
|
||
//返回一个空指针
|
||
webResult = result.ConstWebResult(result.OK)
|
||
} else {
|
||
//返回的类型不明确。
|
||
webResult = &result.WebResult{Code: result.OK.Code, Data: data}
|
||
}
|
||
return webResult
|
||
}
|
||
|
||
//允许跨域请求
|
||
func (this *BaseController) allowCORS(writer http.ResponseWriter) {
|
||
writer.Header().Add("Access-Control-Allow-Origin", "*")
|
||
writer.Header().Add("Access-Control-Allow-Methods", "POST, GET, OPTIONS, DELETE")
|
||
writer.Header().Add("Access-Control-Max-Age", "3600")
|
||
}
|