package controllers import ( "encoding/json" "github.com/beego/beego/v2/server/web" "hongze/hongze_clpt/models" "hongze/hongze_clpt/utils" "net/http" "net/url" "strings" ) type BaseCommonController struct { web.Controller //User *models.WxUserItem Token string } func (this *BaseCommonController) Prepare() { var requestBody string method := this.Ctx.Input.Method() if method == "GET" { requestBody = this.Ctx.Request.RequestURI } else { requestBody, _ = url.QueryUnescape(string(this.Ctx.Input.RequestBody)) } ip := this.Ctx.Input.IP() apiLog.Println("请求地址:", this.Ctx.Input.URI(), "RequestBody:", requestBody, "IP:", ip) authorization := this.Ctx.Input.Header("Authorization") if authorization == "" { cookie := this.Ctx.GetCookie("rddp_access_token") utils.FileLog.Info("authorization:%s,cookie:%s", authorization, cookie) authorization = cookie } uri := this.Ctx.Input.URI() utils.FileLog.Info("URI:%s", uri) if strings.Contains(uri, "/api/wechat/login") { authorization = "" } if authorization != "" { //session, err := models.GetSessionByToken(authorization) //if err != nil { // if err.Error() == utils.ErrNoRow() { // this.JSON(models.BaseResponse{Ret: 408, Msg: "信息已变更,请重新登陆!", ErrMsg: "Token 信息已变更:Token: " + authorization}, false, false) // this.StopRun() // return // } // this.JSON(models.BaseResponse{Ret: 408, Msg: "网络异常,请稍后重试!", ErrMsg: "获取用户信息异常,Eerr:" + err.Error()}, false, false) // this.StopRun() // return //} //if session == nil { // this.JSON(models.BaseResponse{Ret: 408, Msg: "网络异常,请稍后重试!", ErrMsg: "sesson is empty "}, false, false) // this.StopRun() // return //} //wxUser, err := models.GetWxUserItemByUserId(session.UserId) //if err != nil { // if err.Error() == utils.ErrNoRow() { // this.JSON(models.BaseResponse{Ret: 408, Msg: "信息已变更,请重新登陆!", ErrMsg: "获取admin 信息失败 " + strconv.Itoa(session.UserId)}, false, false) // this.StopRun() // return // } // this.JSON(models.BaseResponse{Ret: 408, Msg: "网络异常,请稍后重试!", ErrMsg: "获取wx_user信息异常,Eerr:" + err.Error()}, false, false) // this.StopRun() // return //} //if wxUser == nil { // this.JSON(models.BaseResponse{Ret: 408, Msg: "网络异常,请稍后重试!", ErrMsg: "admin is empty "}, false, false) // this.StopRun() // return //} //this.User = wxUser } //this.Token = authorization } func (c *BaseCommonController) ServeJSON(encoding ...bool) { 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(utils.APPNAME+" "+utils.RunMode+"异常提醒", "接口:"+"URI:"+c.Ctx.Input.URI()+";无返回值", utils.EmailSendToUsers) return } baseRes := c.Data["json"].(*models.BaseResponse) if baseRes != nil && !baseRes.Success && baseRes.IsSendEmail { go utils.SendEmail(utils.APPNAME+" "+utils.RunMode+" 失败提醒", "URI:"+c.Ctx.Input.URI()+" ErrMsg:"+baseRes.ErrMsg+";Msg"+baseRes.Msg, 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, _ := url.QueryUnescape(string(c.Ctx.Input.RequestBody)) apiLog.Println("请求地址:", c.Ctx.Input.URI(), "Authorization:", c.Ctx.Input.Header("Authorization"), "RequestBody:", requestBody, "ResponseBody", string(content), "IP:", ip) if coding { content = []byte(utils.StringsToJSON(string(content))) } return c.Ctx.Output.Body(content) }