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)
}