s_cherry_e/sr/internal/code/data_result.go

package code

import (
	cherryGin "github.com/cherry-game/components/gin"
	"github.com/gin-gonic/gin"
)

type DataResult struct {
	Code    int32  `json:"code"`
	Message string `json:"message"`
	Data    any    `json:"data"`
}

func NewDataResult(code int32) DataResult {
	result := DataResult{
		Code:    code,
		Message: GetMessage(code),
		Data:    []string{},
	}
	return result
}

func (p *DataResult) SetCode(code int32) {
	p.Code = code
	p.Message = GetMessage(code)
}

func JSONResult(c *cherryGin.Context, statusCode int32, data ...any) {
	result := NewDataResult(statusCode)
	if len(data) > 0 {
		result.Data = data[0]
	}
	c.JSON200(result)
}

func JSONError(c *cherryGin.Context, message ...string) {
	result := NewDataResult(Error)
	if len(message) > 0 {
		result.Message = message[0]
	}
	c.JSON200(result)
}

func Direct404(c *cherryGin.Context, statusCode int32) {
	c.HTML200("404.html", gin.H{
		"title":   "404",
		"message": GetMessage(statusCode),
	})
}

func Paging(c *cherryGin.Context, list any, count int64) {
	c.JSON200(gin.H{
		"code":  0,
		"msg":   "",
		"count": count,
		"data":  list,
	})
}