作者 yangfu

1.中间间修改

@@ -2,14 +2,13 @@ package appsvr @@ -2,14 +2,13 @@ package appsvr
2 2
3 import ( 3 import (
4 "github.com/astaxie/beego" 4 "github.com/astaxie/beego"
5 - "github.com/linmadan/egglib-go/web/beego/filters"  
6 . "gitlab.fjmaimaimai.com/mmm-go/partner/pkg/log" 5 . "gitlab.fjmaimaimai.com/mmm-go/partner/pkg/log"
7 "gitlab.fjmaimaimai.com/mmm-go/partner/pkg/port/appsvr/middleware" 6 "gitlab.fjmaimaimai.com/mmm-go/partner/pkg/port/appsvr/middleware"
8 _ "gitlab.fjmaimaimai.com/mmm-go/partner/pkg/port/appsvr/routers" 7 _ "gitlab.fjmaimaimai.com/mmm-go/partner/pkg/port/appsvr/routers"
9 ) 8 )
10 9
11 func init() { 10 func init() {
12 - beego.InsertFilter("/*", beego.BeforeExec, filters.CreateRequestBodyFilter()) 11 + beego.InsertFilter("/*", beego.BeforeExec, middleware.CreateRequestBodyFilter())
13 beego.InsertFilter("/*", beego.BeforeExec, middleware.CreateRequstLogFilter(Logger)) 12 beego.InsertFilter("/*", beego.BeforeExec, middleware.CreateRequstLogFilter(Logger))
14 beego.InsertFilter("/*", beego.AfterExec, middleware.CreateResponseLogFilter(Logger), false) 13 beego.InsertFilter("/*", beego.AfterExec, middleware.CreateResponseLogFilter(Logger), false)
15 } 14 }
@@ -20,7 +20,7 @@ func (this *UserController) UserInfo() { @@ -20,7 +20,7 @@ func (this *UserController) UserInfo() {
20 }() 20 }()
21 var request *protocol.UserInfoRequest 21 var request *protocol.UserInfoRequest
22 if err := this.JsonUnmarshal(&request); err != nil { 22 if err := this.JsonUnmarshal(&request); err != nil {
23 - msg = protocol.BadRequestParam(1) 23 + msg = protocol.BadRequestParam(2)
24 return 24 return
25 } 25 }
26 if b, m := this.Valid(request); !b { 26 if b, m := this.Valid(request); !b {
@@ -5,6 +5,7 @@ import ( @@ -5,6 +5,7 @@ import (
5 "fmt" 5 "fmt"
6 "github.com/astaxie/beego/context" 6 "github.com/astaxie/beego/context"
7 "github.com/astaxie/beego/logs" 7 "github.com/astaxie/beego/logs"
  8 + "io/ioutil"
8 "time" 9 "time"
9 ) 10 )
10 11
@@ -30,3 +31,20 @@ func CreateResponseLogFilter(logger *logs.BeeLogger) func(ctx *context.Context) @@ -30,3 +31,20 @@ func CreateResponseLogFilter(logger *logs.BeeLogger) func(ctx *context.Context)
30 logger.Debug(fmt.Sprintf("<====Send RequestId:%v BodyData:%s", requestId, body)) 31 logger.Debug(fmt.Sprintf("<====Send RequestId:%v BodyData:%s", requestId, body))
31 } 32 }
32 } 33 }
  34 +
  35 +func CreateRequestBodyFilter() func(ctx *context.Context) {
  36 + return func(ctx *context.Context) {
  37 + if len(ctx.Input.RequestBody) == 0 {
  38 + if ctx.Request.Method == "POST" || ctx.Request.Method == "PUT" {
  39 + body, _ := ioutil.ReadAll(ctx.Request.Body)
  40 + if len(body) == 0 {
  41 + body = []byte("{}")
  42 + }
  43 + ctx.Input.SetData("requestBody", body)
  44 + ctx.Request.Body.Close()
  45 + }
  46 + } else {
  47 + ctx.Input.SetData("requestBody", ctx.Input.RequestBody)
  48 + }
  49 + }
  50 +}