48 lines
1.0 KiB
Go
48 lines
1.0 KiB
Go
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)
|
|
}
|