m

code/base-project/base-go/pkg/reponse/response.go

@@ -0,0 +1,115 @@
+package reponse
+
+import (
+	"base-framework/pkg/router"
+	"net/http"
+)
+
+type ErrorInfo struct {
+	TrackId  string `json:"trackId,omitempty"`
+	Method   string `json:"method,omitempty"`
+	Path     string `json:"path,omitempty"`
+	Message  string `json:"message,omitempty"`
+	Location string `json:"location,omitempty"`
+}
+
+// Result 统一响应结构体
+type Result struct {
+	Code    string      `json:"code"`
+	Message string      `json:"message"`
+	Data    interface{} `json:"data,omitempty"`
+	Error   ErrorInfo   `json:"error,omitempty"`
+}
+
+const (
+	CodeSuccess = "0000" // 成功
+	CodeFail    = "0001" // 失败
+
+	CodeNoLogin        = "1001" // 未登录
+	CodeLoginExpired   = "1002" // 登录过期
+	CodeInvalidToken   = "1003" // Token 无效
+	CodeInvalidOrgCode = "1004" // 机构码有误
+	CodeInvalidAccount = "1005" // 账号或密码有误
+)
+
+var codeMessages = map[string]string{
+	CodeSuccess:        "请求成功",
+	CodeFail:           "请求失败",
+	CodeNoLogin:        "未登录",
+	CodeLoginExpired:   "登录已过期",
+	CodeInvalidToken:   "Token 无效",
+	CodeInvalidOrgCode: "机构码有误",
+	CodeInvalidAccount: "账号或密码有误",
+}
+
+// 兜底提示
+const defaultMessage = "未知错误"
+
+// Builder 链式构造器
+type Builder struct {
+	c      *router.Context
+	result Result
+}
+
+// Success 构造成功响应
+func Success(c *router.Context) *Builder {
+	return &Builder{
+		c: c,
+		result: Result{
+			Code:    CodeSuccess,
+			Message: getMessage(CodeSuccess),
+		},
+	}
+}
+
+// Error 构造失败响应
+func Error(c *router.Context) *Builder {
+	return &Builder{
+		c: c,
+		result: Result{
+			Code:    CodeFail,
+			Message: getMessage(CodeFail),
+		},
+	}
+}
+
+// Code 设置状态码（自动填充默认提示，除非后面手动改）
+func (b *Builder) Code(code string) *Builder {
+	b.result.Code = code
+	b.result.Message = getMessage(code)
+	return b
+}
+
+// Message 设置提示消息（覆盖默认的）
+func (b *Builder) Message(msg string) *Builder {
+	b.result.Message = msg
+	return b
+}
+
+// Data 设置返回数据
+func (b *Builder) Data(data interface{}) *Builder {
+	b.result.Data = data
+	return b
+}
+
+// ErrorInfo 设置错误信息
+func (b *Builder) ErrorInfo(err ErrorInfo) *Builder {
+	b.result.Error = err
+	return b
+}
+
+// Send 发送响应
+func (b *Builder) Send() {
+	if b.result.Data == nil {
+		b.result.Data = ""
+	}
+	b.c.JSON(http.StatusOK, b.result)
+}
+
+// getMessage 获取状态码对应提示，兜底 defaultMessage
+func getMessage(code string) string {
+	if msg, ok := codeMessages[code]; ok {
+		return msg
+	}
+	return defaultMessage
+}