mayfly-go/server/pkg/model/result.go

package model

import (
	"encoding/json"
	"fmt"
	"mayfly-go/pkg/errorx"
)

const (
	SuccessCode = 200
	SuccessMsg  = "success"
)

// 统一返回结果结构体
type Result struct {
	Code int16  `json:"code"`
	Msg  string `json:"msg"`
	Data any    `json:"data"`
}

// 将Result转为json字符串
func (r *Result) ToJson() string {
	jsonData, err := json.Marshal(r)
	if err != nil {
		fmt.Println("data转json错误")
	}
	return string(jsonData)
}

// 判断该Result是否为成功状态
func (r *Result) IsSuccess() bool {
	return r.Code == SuccessCode
}

// 返回成功状态的Result
// @param data 成功附带的数据消息
func Success(data any) *Result {
	return &Result{Code: SuccessCode, Msg: SuccessMsg, Data: data}
}

// 返回成功状态的Result
// @param data 成功不附带数据
func SuccessNoData() *Result {
	return &Result{Code: SuccessCode, Msg: SuccessMsg}
}

func Error(bizerr *errorx.BizError) *Result {
	return &Result{Code: bizerr.Code(), Msg: bizerr.Error()}
}

// 返回服务器错误Result
func ServerError(msg string) *Result {
	serverErr := errorx.NewBizCode(errorx.ServerError.Code(), msg)
	return Error(serverErr)
}

func TokenError() *Result {
	return Error(errorx.PermissionErr)
}

func ErrorBy(code int16, msg string) *Result {
	return &Result{Code: code, Msg: msg}
}