50 lines
1.0 KiB
Go
50 lines
1.0 KiB
Go
// Package infra 提供基础设施功能
|
|
// 包括统一响应处理、健康检查、日志等
|
|
package infra
|
|
|
|
import (
|
|
"github.com/gin-gonic/gin"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
var Response Reply
|
|
|
|
// Reply 统一响应结构体
|
|
type Reply struct {
|
|
Code int32 `json:"code"` // 响应码
|
|
Message string `json:"message"` // 响应消息
|
|
Result any `json:"result"` // 响应数据
|
|
}
|
|
|
|
// Success 返回成功响应
|
|
// ctx: Gin上下文
|
|
// data: 响应数据
|
|
func (reply *Reply) Success(ctx *gin.Context, data any) {
|
|
reply.Code = 0
|
|
reply.Result = data
|
|
reply.Message = ""
|
|
if data == nil {
|
|
reply.Result = ""
|
|
}
|
|
ctx.JSON(200, reply)
|
|
}
|
|
|
|
// Error 返回错误响应
|
|
// ctx: Gin上下文
|
|
// err: 错误对象
|
|
func (reply *Reply) Error(ctx *gin.Context, err error) {
|
|
reply.Code = 500
|
|
reply.Result = ""
|
|
// 默认状态码为500
|
|
e, ok := status.FromError(err)
|
|
if ok {
|
|
reply.Code = int32(e.Code())
|
|
reply.Message = e.Message()
|
|
} else {
|
|
reply.Message = err.Error()
|
|
}
|
|
|
|
// 发送错误响应
|
|
ctx.JSON(200, reply)
|
|
}
|