u

code/base-project/base-go-v2/internal/reponse/response.go

@@ -0,0 +1,107 @@
+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
+}