|
...
|
...
|
@@ -6,10 +6,11 @@ import ( |
|
|
|
"fmt"
|
|
|
|
"io/ioutil"
|
|
|
|
"strconv"
|
|
|
|
"strings"
|
|
|
|
|
|
|
|
"github.com/astaxie/beego"
|
|
|
|
"github.com/astaxie/beego/logs"
|
|
|
|
"github.com/prometheus/common/log"
|
|
|
|
|
|
|
|
adminuserquery "gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/application/adminUser/query"
|
|
|
|
adminuserservice "gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/application/adminUser/service"
|
|
|
|
"gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/lib"
|
|
...
|
...
|
@@ -87,30 +88,37 @@ func (controller *BaseController) Finish() { |
|
|
|
strByte, _ := json.Marshal(controller.Data["json"])
|
|
|
|
length := len(strByte)
|
|
|
|
if length > 1500 {
|
|
|
|
logs.Info("<====Send to client: RspBodyData: %s......", string(strByte[:1500]))
|
|
|
|
logs.Info("<====Send to client: RspBodyData: ", string(strByte[:1500]))
|
|
|
|
} else {
|
|
|
|
logs.Info("<====Send to client: RspBodyData: %s......", string(strByte))
|
|
|
|
logs.Info("<====Send to client: RspBodyData: ", string(strByte))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
////Prepare 重写 beego.Controller 的Prepare方法
|
|
|
|
func (controller *BaseController) Prepare() {
|
|
|
|
log.Info("====>Recv Request:%s Method:%s", controller.Ctx.Input.URI(), controller.Ctx.Input.Method())
|
|
|
|
logs.Info("====>Recv Request:", controller.Ctx.Input.URI())
|
|
|
|
if controller.Ctx.Input.IsPost() || controller.Ctx.Input.IsPut() {
|
|
|
|
bodyByte, _ := ioutil.ReadAll(controller.Ctx.Request.Body)
|
|
|
|
controller.Ctx.Input.SetData("requestBody", bodyByte)
|
|
|
|
controller.Ctx.Request.Body.Close()
|
|
|
|
if len(bodyByte) > 1000 {
|
|
|
|
log.Info("====>Recv data from client: BodyData:\n %s ...", string(bodyByte[0:1000]))
|
|
|
|
logs.Info("====>Recv data from client: BodyData:\n ", string(bodyByte[0:1000]))
|
|
|
|
} else {
|
|
|
|
log.Info("====>Recv data from client: BodyData: \n %s", string(bodyByte))
|
|
|
|
logs.Info("====>Recv data from client: BodyData: \n ", string(bodyByte))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
func (controller *BaseController) GetHeaderToken() string {
|
|
|
|
return controller.Ctx.Input.Header("Authorization")
|
|
|
|
authString := controller.Ctx.Input.Header("Authorization")
|
|
|
|
//解出需要的jwt串 例:头Authorization:Bearer 123token456
|
|
|
|
s := strings.Split(authString, "\u0020")
|
|
|
|
var tokenString string
|
|
|
|
if len(s) > 0 {
|
|
|
|
tokenString = s[len(s)-1]
|
|
|
|
}
|
|
|
|
return tokenString
|
|
|
|
}
|
|
|
|
|
|
|
|
func (controller *BaseController) ValidJWTToken() bool {
|
|
...
|
...
|
@@ -118,6 +126,7 @@ func (controller *BaseController) ValidJWTToken() bool { |
|
|
|
mytoken := new(lib.MyToken)
|
|
|
|
err := mytoken.ValidJWTToken(headerToken)
|
|
|
|
if err != nil {
|
|
|
|
logs.Error("校验token失败", err)
|
|
|
|
if mytoken.IsJwtErrorExpired(err) {
|
|
|
|
//token超时
|
|
|
|
controller.Data["json"] = protocol.ResponseData{
|
...
|
...
|
|