关于碰见的this.ServeJSON()问题

场景:
刚学习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()

去除以后,结果正常了

你可能感兴趣的:(Golang,beego)