108 lines
2.1 KiB
Go
108 lines
2.1 KiB
Go
package response
|
|
|
|
import (
|
|
"base-go-v2/internal/logx"
|
|
"base-go-v2/internal/router"
|
|
"base-go-v2/internal/routinex"
|
|
"base-go-v2/internal/utils/strutil"
|
|
"net/http"
|
|
)
|
|
|
|
type Result struct {
|
|
Code string `json:"code"`
|
|
Message string `json:"message"`
|
|
Data interface{} `json:"data,omitempty"`
|
|
TrackId string `json:"trackId"`
|
|
}
|
|
|
|
type Code struct {
|
|
Value string
|
|
Message string
|
|
}
|
|
|
|
var (
|
|
CodeSuccess = Code{"0000", "请求成功"}
|
|
CodeFail = Code{"0001", "请求失败"}
|
|
CodeNoLogin = Code{"1001", "未登录"}
|
|
CodeLoginExpired = Code{"1002", "登录已过期"}
|
|
CodeInvalidToken = Code{"1003", "Token 无效"}
|
|
CodeInvalidAccount = Code{"1005", "账号或密码错误"}
|
|
)
|
|
|
|
const defaultMessage = "未知错误"
|
|
|
|
type Builder struct {
|
|
c *router.Context
|
|
result Result
|
|
}
|
|
|
|
func Success(c *router.Context) *Builder {
|
|
return &Builder{
|
|
c: c,
|
|
result: Result{
|
|
Code: CodeSuccess.Value,
|
|
Message: CodeSuccess.Message,
|
|
},
|
|
}
|
|
}
|
|
|
|
func Error(c *router.Context) *Builder {
|
|
return &Builder{
|
|
c: c,
|
|
result: Result{
|
|
Code: CodeFail.Value,
|
|
Message: CodeFail.Message,
|
|
},
|
|
}
|
|
}
|
|
|
|
func (b *Builder) Code(code string) *Builder {
|
|
b.result.Code = code
|
|
b.result.Message = GetMessage(code)
|
|
return b
|
|
}
|
|
|
|
func (b *Builder) CodeObj(code Code) *Builder {
|
|
b.result.Code = code.Value
|
|
b.result.Message = code.Message
|
|
return b
|
|
}
|
|
|
|
func (b *Builder) Message(msg string) *Builder {
|
|
b.result.Message = msg
|
|
return b
|
|
}
|
|
|
|
func (b *Builder) Data(data interface{}) *Builder {
|
|
b.result.Data = data
|
|
return b
|
|
}
|
|
|
|
func (b *Builder) Send() error {
|
|
// 默认 Data 为空
|
|
if b.result.Data == nil {
|
|
b.result.Data = ""
|
|
}
|
|
|
|
// 从 routinex 获取 trackId
|
|
trackId := routinex.Get(logx.TrackID)
|
|
if trackId != nil {
|
|
b.result.TrackId = strutil.ToString(trackId)
|
|
}
|
|
|
|
// 发送响应
|
|
return b.c.JSON(http.StatusOK, b.result)
|
|
}
|
|
|
|
func GetMessage(value string) string {
|
|
for _, c := range []Code{
|
|
CodeSuccess, CodeFail, CodeNoLogin,
|
|
CodeLoginExpired, CodeInvalidToken, CodeInvalidAccount,
|
|
} {
|
|
if c.Value == value {
|
|
return c.Message
|
|
}
|
|
}
|
|
return defaultMessage
|
|
}
|