package controllers import ( "encoding/json" "eta/eta_forum_admin/services/alarm_msg" "net/http" "net/url" "github.com/beego/beego/v2/server/web" "eta/eta_forum_admin/models" "eta/eta_forum_admin/utils" ) type BaseCommonController struct { web.Controller } func (c *BaseCommonController) ServeJSON(encoding ...bool) { // 方法处理完后,需要后置处理的业务逻辑 //if handlerList, ok := AfterHandlerUrlMap[c.Ctx.Request.URL.Path]; ok { // for _, handler := range handlerList { // handler(c.Ctx.Input.RequestBody) // } //} var ( hasIndent = false hasEncoding = false ) if web.BConfig.RunMode == web.PROD { hasIndent = false } if len(encoding) > 0 && encoding[0] == true { hasEncoding = true } if c.Data["json"] == nil { go alarm_msg.SendAlarmMsg("接口:URI:"+c.Ctx.Input.URI()+";无返回值", 3) return } baseRes := c.Data["json"].(*models.BaseResponse) if baseRes != nil && baseRes.Ret != 408 { body, _ := json.Marshal(baseRes) var requestBody string method := c.Ctx.Input.Method() if method == "GET" { requestBody = c.Ctx.Request.RequestURI } else { requestBody, _ = url.QueryUnescape(string(c.Ctx.Input.RequestBody)) } if baseRes.Ret != 200 && baseRes.IsSendEmail { //go utils.SendEmail(utils.APP_NAME_CN+"【"+utils.RunMode+"】"+"失败提醒", "URI:"+c.Ctx.Input.URI()+"
"+"Params"+requestBody+"
"+"ErrMsg:"+baseRes.ErrMsg+";
Msg:"+baseRes.Msg+";
Body:"+string(body)+"
", utils.EmailSendToUsers) go alarm_msg.SendAlarmMsg("失败提醒:URI:"+c.Ctx.Input.URI()+";请求入参:Params"+requestBody+" ErrMsg:"+baseRes.ErrMsg+";Msg"+baseRes.Msg+"; Body:"+string(body), 3) } } c.JSON(c.Data["json"], hasIndent, hasEncoding) } func (c *BaseCommonController) JSON(data interface{}, hasIndent bool, coding bool) error { c.Ctx.Output.Header("Content-Type", "application/json; charset=utf-8") var content []byte var err error if hasIndent { content, err = json.MarshalIndent(data, "", " ") } else { content, err = json.Marshal(data) } if err != nil { http.Error(c.Ctx.Output.Context.ResponseWriter, err.Error(), http.StatusInternalServerError) return err } ip := c.Ctx.Input.IP() requestBody, err := url.QueryUnescape(string(c.Ctx.Input.RequestBody)) if err != nil { requestBody = string(c.Ctx.Input.RequestBody) } if requestBody == "" { requestBody = c.Ctx.Input.URI() } authorization := c.Ctx.Input.Header("authorization") if authorization == "" { authorization = c.Ctx.Input.Header("Authorization") } utils.ApiLog.Info("uri:%s, authorization:%s, requestBody:%s, responseBody:%s, ip:%s", c.Ctx.Input.URI(), authorization, requestBody, content, ip) if utils.RunMode != "debug" { content = utils.DesBase64Encrypt(content, utils.DesKey) // get请求时,不加双引号就获取不到数据,不知道什么原因,所以还是在前后加上双引号吧 content = []byte(`"` + string(content) + `"`) } if coding { content = []byte(utils.StringsToJSON(string(content))) } return c.Ctx.Output.Body(content) }