任务三

pkg/response/response.go

@@ -0,0 +1,47 @@
+package response
+
+import (
+	"encoding/json"
+	"net/http"
+
+	"github.com/go-kratos/kratos/v2/errors"
+)
+
+// Response 统一响应格式
+type Response struct {
+	Code int         `json:"code"`
+	Msg  string      `json:"msg"`
+	Data interface{} `json:"data,omitempty"`
+}
+
+// Encoder 响应编码器，包装响应为 {code, msg, data} 格式
+func Encoder(w http.ResponseWriter, r *http.Request, v interface{}) error {
+	if v == nil {
+		v = struct{}{}
+	}
+	reply := Response{
+		Code: 0,
+		Msg:  "success",
+		Data: v,
+	}
+	w.Header().Set("Content-Type", "application/json")
+	return json.NewEncoder(w).Encode(reply)
+}
+
+// ErrorEncoder 错误编码器，统一错误响应格式
+func ErrorEncoder(w http.ResponseWriter, r *http.Request, err error) {
+	se := errors.FromError(err)
+	code := int(se.Code)
+	if code >= 400 && code < 500 {
+		code = 7
+	} else if code >= 500 {
+		code = 7
+	}
+	reply := Response{
+		Code: code,
+		Msg:  se.Message,
+	}
+	w.Header().Set("Content-Type", "application/json")
+	w.WriteHeader(int(se.Code))
+	json.NewEncoder(w).Encode(reply)
+}