go-wxhelper/pkg/response/response.go

package response

import (
	"github.com/gin-gonic/gin"
	"net/http"
	"wechat-robot/pkg/validator"
)

// 定义状态码
const (
	fail    = http.StatusInternalServerError
	success = http.StatusOK
)

// Response
// @description: 返回结果
type Response struct {
	ctx    *gin.Context
	code   int
	data   any
	msg    string
	errMsg string
}

// New
// @description: 返回结果实现
// @param ctx
// @return Response
func New(ctx *gin.Context) *Response {
	var r Response
	r.ctx = ctx

	return &r
}

// SetCode
// @description: 设置状态码
// @receiver r
// @param code
// @return *Response
func (r *Response) SetCode(code int) *Response {
	r.code = code
	return r
}

// SetData
// @description: 设置返回数据
// @receiver r
// @param data
// @return *Response
func (r *Response) SetData(data any) *Response {
	r.data = data
	return r
}

// SetMsg
// @description: 设置返回消息
// @receiver r
// @param msg
// @return *Response
func (r *Response) SetMsg(msg string) *Response {
	r.msg = msg
	return r
}

// SetError
// @description: 设置错误信息
// @receiver r
// @param err
// @return *Response
func (r *Response) SetError(err error) *Response {
	if err != nil {
		err = validator.Translate(err)
		r.errMsg = err.Error()
	}
	return r
}