package controllers import ( "encoding/json" "net/http" "net/url" "github.com/beego/beego/v2/server/web" "eta/eta_index_lib/models" "eta/eta_index_lib/utils" ) type BaseCommonController struct { web.Controller Lang string `description:"当前语言类型,中文:zh;英文:en;默认:zh"` } func (c *BaseCommonController) Prepare() { //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)) //} // 当前语言 { lang := c.Ctx.Input.Header("Lang") if lang == "" { lang = utils.ZhLangVersion } c.Lang = lang } //ip := c.Ctx.Input.IP() //utils.ApiLog.Info("uri:%s, requestBody:%s, responseBody:%s, ip:%s", c.Ctx.Input.URI(), requestBody, ip) } 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 utils.SendEmail("异常提醒:", "接口:"+"URI:"+c.Ctx.Input.URI()+";无返回值", utils.EmailSendToUsers) 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) } } 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 coding { content = []byte(utils.StringsToJSON(string(content))) } return c.Ctx.Output.Body(content) }