base.go
1.1 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
package controllers
import (
"bytes"
"encoding/json"
"github.com/astaxie/beego"
"github.com/astaxie/beego/context"
"gitlab.fjmaimaimai.com/mmm-go/godevp/pkg/protocol"
)
type BaseController struct {
beego.Controller
}
func (controller BaseController) JsonUnmarshal(v interface{}) error {
body := controller.Ctx.Input.RequestBody
if len(body) == 0 {
body = []byte("{}")
}
newDecoder := json.NewDecoder(bytes.NewReader(body))
newDecoder.UseNumber()
err := newDecoder.Decode(v)
return err
}
func (controller *BaseController) Resp(msg *protocol.ResponseMessage) {
if msg.Errno != 0 {
msg.Errno = -1
}
controller.Data["json"] = msg
controller.Ctx.Input.SetData("outputData", msg)
controller.ServeJSON()
}
func (controller BaseController) GetLimitInfo() (offset int, limit int) {
offset, _ = controller.GetInt("offset")
limit, _ = controller.GetInt("limit")
return
}
//获取请求头信息
func (controller *BaseController) GetRequestHeader(ctx *context.Context) *protocol.RequestHeader {
h := &protocol.RequestHeader{}
if ctx.Input.GetData("x-mmm-id") != nil {
v := ctx.Input.GetData("x-mmm-id")
h.UserId = int64(v.(int))
}
return h
}