68 lines
1.4 KiB
Go
68 lines
1.4 KiB
Go
package response
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
type Response struct {
|
|
Code int `json:"code"`
|
|
Message string `json:"message"`
|
|
Total int64 `json:"total,omitempty"`
|
|
Data any `json:"data"`
|
|
}
|
|
|
|
func getErrorCodeAndMessage(err error) (int, string) {
|
|
if st, ok := status.FromError(err); ok {
|
|
return int(st.Code()), st.Message()
|
|
}
|
|
return CodeInternalServerError, "未知错误类型或内部服务错误"
|
|
}
|
|
|
|
func Result(c *gin.Context, err error, data any) {
|
|
if err == nil {
|
|
c.JSON(http.StatusOK, Response{Code: CodeSuccess, Message: MsgSuccess, Data: data})
|
|
return
|
|
}
|
|
|
|
code, message := getErrorCodeAndMessage(err)
|
|
|
|
httpStatus := http.StatusOK
|
|
|
|
if code == CodeInternalServerError || !errors.Is(err, ErrSuccess) && code == 0 {
|
|
httpStatus = http.StatusInternalServerError
|
|
}
|
|
|
|
c.JSON(httpStatus, Response{
|
|
Code: code,
|
|
Message: message,
|
|
Data: data,
|
|
})
|
|
}
|
|
|
|
func PageResult(c *gin.Context, err error, total int64, data any) {
|
|
if err == nil {
|
|
c.JSON(http.StatusOK, Response{Code: CodeSuccess, Message: MsgSuccess, Total: total, Data: data})
|
|
return
|
|
}
|
|
|
|
code, message := getErrorCodeAndMessage(err)
|
|
|
|
httpStatus := http.StatusOK
|
|
|
|
if code == CodeInternalServerError || !errors.Is(err, ErrSuccess) && code == 0 {
|
|
httpStatus = http.StatusInternalServerError
|
|
}
|
|
|
|
c.JSON(httpStatus, Response{
|
|
Code: code,
|
|
Message: message,
|
|
Total: total,
|
|
Data: data,
|
|
})
|
|
}
|
|
|