|
@@ -6,10 +6,11 @@ import ( |
|
@@ -6,10 +6,11 @@ import ( |
|
6
|
"fmt"
|
6
|
"fmt"
|
|
7
|
"io/ioutil"
|
7
|
"io/ioutil"
|
|
8
|
"strconv"
|
8
|
"strconv"
|
|
|
|
9
|
+ "strings"
|
|
9
|
|
10
|
|
|
10
|
"github.com/astaxie/beego"
|
11
|
"github.com/astaxie/beego"
|
|
11
|
"github.com/astaxie/beego/logs"
|
12
|
"github.com/astaxie/beego/logs"
|
|
12
|
- "github.com/prometheus/common/log"
|
13
|
+
|
|
13
|
adminuserquery "gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/application/adminUser/query"
|
14
|
adminuserquery "gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/application/adminUser/query"
|
|
14
|
adminuserservice "gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/application/adminUser/service"
|
15
|
adminuserservice "gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/application/adminUser/service"
|
|
15
|
"gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/lib"
|
16
|
"gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/lib"
|
|
@@ -87,30 +88,37 @@ func (controller *BaseController) Finish() { |
|
@@ -87,30 +88,37 @@ func (controller *BaseController) Finish() { |
|
87
|
strByte, _ := json.Marshal(controller.Data["json"])
|
88
|
strByte, _ := json.Marshal(controller.Data["json"])
|
|
88
|
length := len(strByte)
|
89
|
length := len(strByte)
|
|
89
|
if length > 1500 {
|
90
|
if length > 1500 {
|
|
90
|
- logs.Info("<====Send to client: RspBodyData: %s......", string(strByte[:1500]))
|
91
|
+ logs.Info("<====Send to client: RspBodyData: ", string(strByte[:1500]))
|
|
91
|
} else {
|
92
|
} else {
|
|
92
|
- logs.Info("<====Send to client: RspBodyData: %s......", string(strByte))
|
93
|
+ logs.Info("<====Send to client: RspBodyData: ", string(strByte))
|
|
93
|
}
|
94
|
}
|
|
94
|
}
|
95
|
}
|
|
95
|
|
96
|
|
|
96
|
////Prepare 重写 beego.Controller 的Prepare方法
|
97
|
////Prepare 重写 beego.Controller 的Prepare方法
|
|
97
|
func (controller *BaseController) Prepare() {
|
98
|
func (controller *BaseController) Prepare() {
|
|
98
|
- log.Info("====>Recv Request:%s Method:%s", controller.Ctx.Input.URI(), controller.Ctx.Input.Method())
|
99
|
+ logs.Info("====>Recv Request:", controller.Ctx.Input.URI())
|
|
99
|
if controller.Ctx.Input.IsPost() || controller.Ctx.Input.IsPut() {
|
100
|
if controller.Ctx.Input.IsPost() || controller.Ctx.Input.IsPut() {
|
|
100
|
bodyByte, _ := ioutil.ReadAll(controller.Ctx.Request.Body)
|
101
|
bodyByte, _ := ioutil.ReadAll(controller.Ctx.Request.Body)
|
|
101
|
controller.Ctx.Input.SetData("requestBody", bodyByte)
|
102
|
controller.Ctx.Input.SetData("requestBody", bodyByte)
|
|
102
|
controller.Ctx.Request.Body.Close()
|
103
|
controller.Ctx.Request.Body.Close()
|
|
103
|
if len(bodyByte) > 1000 {
|
104
|
if len(bodyByte) > 1000 {
|
|
104
|
- log.Info("====>Recv data from client: BodyData:\n %s ...", string(bodyByte[0:1000]))
|
105
|
+ logs.Info("====>Recv data from client: BodyData:\n ", string(bodyByte[0:1000]))
|
|
105
|
} else {
|
106
|
} else {
|
|
106
|
- log.Info("====>Recv data from client: BodyData: \n %s", string(bodyByte))
|
107
|
+ logs.Info("====>Recv data from client: BodyData: \n ", string(bodyByte))
|
|
107
|
}
|
108
|
}
|
|
108
|
}
|
109
|
}
|
|
109
|
return
|
110
|
return
|
|
110
|
}
|
111
|
}
|
|
111
|
|
112
|
|
|
112
|
func (controller *BaseController) GetHeaderToken() string {
|
113
|
func (controller *BaseController) GetHeaderToken() string {
|
|
113
|
- return controller.Ctx.Input.Header("Authorization")
|
114
|
+ authString := controller.Ctx.Input.Header("Authorization")
|
|
|
|
115
|
+ //解出需要的jwt串 例:头Authorization:Bearer 123token456
|
|
|
|
116
|
+ s := strings.Split(authString, "\u0020")
|
|
|
|
117
|
+ var tokenString string
|
|
|
|
118
|
+ if len(s) > 0 {
|
|
|
|
119
|
+ tokenString = s[len(s)-1]
|
|
|
|
120
|
+ }
|
|
|
|
121
|
+ return tokenString
|
|
114
|
}
|
122
|
}
|
|
115
|
|
123
|
|
|
116
|
func (controller *BaseController) ValidJWTToken() bool {
|
124
|
func (controller *BaseController) ValidJWTToken() bool {
|
|
@@ -118,6 +126,7 @@ func (controller *BaseController) ValidJWTToken() bool { |
|
@@ -118,6 +126,7 @@ func (controller *BaseController) ValidJWTToken() bool { |
|
118
|
mytoken := new(lib.MyToken)
|
126
|
mytoken := new(lib.MyToken)
|
|
119
|
err := mytoken.ValidJWTToken(headerToken)
|
127
|
err := mytoken.ValidJWTToken(headerToken)
|
|
120
|
if err != nil {
|
128
|
if err != nil {
|
|
|
|
129
|
+ logs.Error("校验token失败", err)
|
|
121
|
if mytoken.IsJwtErrorExpired(err) {
|
130
|
if mytoken.IsJwtErrorExpired(err) {
|
|
122
|
//token超时
|
131
|
//token超时
|
|
123
|
controller.Data["json"] = protocol.ResponseData{
|
132
|
controller.Data["json"] = protocol.ResponseData{
|