场景:
刚学习beego,想写一个练手的项目
1、api服务
2、过滤器在controller执行以后,对数据进行再次封装、返回json格式数据
代码大致如下:
在controller
//@router /info/:id [get]
func (user *UserController) UserInfo() {
//do something
user.ServeJSON()
}
在filter
func NsJsonFilter(ctx *context.Context) {
rst := ctx.Input.GetData("json")
//json.MarshalIndent 不支持map[interface {}]interface {}类型,故在此处修改key为string
packageRst := make(map[string]interface{})
if _, ok := rst.(string); ok {
packageRst["code"] = 1
packageRst["errMsg"] = rst
} else {
packageRst["code"] = 0
packageRst["data"] = rst
}
//将封装后的数据重新写回body中
ctx.Output.Reset(ctx)
ctx.Output.JSON(packageRst, true, false)
}
在main中:
beego.InsertFilter("/*", beego.FinishRouter, filter.NsJsonFilter,false)
碰见问题:
在filter中,数据获取正常,修改也正常。但是返回结果始终无法修改,一直是contriller返回的结果。
原因:
原来在output.Body中有这个设置:
func (output *BeegoOutput) Body(content []byte) error {
var encoding string
var buf = &bytes.Buffer{}
if output.EnableGzip {
encoding = ParseEncoding(output.Context.Request)
}
if b, n, _ := WriteBody(encoding, buf, content); b {
output.Header("Content-Encoding", n)
output.Header("Content-Length", strconv.Itoa(buf.Len()))
} else {
output.Header("Content-Length", strconv.Itoa(len(content)))
}
// Write status code if it has been set manually
// Set it to 0 afterwards to prevent "multiple response.WriteHeader calls"
if output.Status != 0 {
output.Context.ResponseWriter.WriteHeader(output.Status)
output.Status = 0
} else {
output.Context.ResponseWriter.Started = true
}
io.Copy(output.Context.ResponseWriter, buf)
return nil
}
output.Context.ResponseWriter.Started = true
即调用一次this.ServeJSON()以后,会把状态设置为true、然后不会再对output中的body结果进行修改。
最后将controller中调用的
user.ServeJSON()
去除以后,结果正常了