eta_mini_ht_api/controllers/base_controller.go

package controllers

import (
	"encoding/json"
	"errors"
	"eta_mini_ht_api/common/component/cache"
	logger "eta_mini_ht_api/common/component/log"
	"eta_mini_ht_api/common/exception"
	"eta_mini_ht_api/common/http"
	"github.com/beego/beego/v2/server/web"
)

var (
	redis *cache.RedisCache
)

type BaseController struct {
	web.Controller
}

type BaseResponse struct {
	Ret         int `description:"返回状态码"`
	Msg         string
	ErrMsg      string
	ErrCode     int
	Data        interface{}
	Success     bool `description:"true 执行成功，false 执行失败"`
	IsSendEmail bool `json:"-" description:"true 发送邮件，false 不发送邮件"`
	IsAddLog    bool `json:"-" description:"true 新增操作日志，false 不新增操作日志" `
}

func (b *BaseController) FailResponse(errInfo error) {

	var retData BaseResponse
	var etaError *exception.EtaError
	if !errors.As(errInfo, &etaError) {
		etaError = exception.New(exception.UnknownError)
	}
	retData = BaseResponse{
		Ret:     200,
		Msg:     "",
		ErrMsg:  etaError.ErrorMsg,
		ErrCode: etaError.ErrorCode,
		Data:    nil}
	b.Data["json"] = retData
	b.ServeJSON()
}

// JsonResult /*
func (b *BaseController) JsonResult(status int, errCode int, errMsg string, msg string, success bool, data interface{}) {
	retData := BaseResponse{
		Ret:     status,
		Msg:     msg,
		ErrMsg:  errMsg,
		ErrCode: errCode,
		Data:    data,
		Success: success}

	b.Ctx.Output.SetStatus(status)
	b.Data["json"] = retData
	b.ServeJSON()
}

func (b *BaseController) Forbidden() {

}

func (b *BaseController) GetPostParams(data interface{}) {
	err := json.Unmarshal(b.Ctx.Input.RequestBody, data)
	if err != nil {
		logger.Error("解析请求参数失败:%v", err)
		data = nil
	}
}

// Wrap ControllerWrap 是一个用于封装控制器方法的函数
func Wrap(a *BaseController, fn func() (*WrapData, error)) {
	result, err := fn()
	if err != nil {
		logger.Error("%v", err)
		a.FailResponse(err)
		return
	}
	a.JsonResult(http.GetHttpStatusByAlias("ok"), http.ErrOK, "", result.Msg, http.Success, result.Data)
}

type WrapData struct {
	Msg  string
	Data interface{}
}

func (b *BaseController) Cache() *cache.RedisCache {
	if redis == nil {
		redis = cache.GetInstance()
	}
	return redis
}