init

src/util/http/response.go

@@ -0,0 +1,73 @@
+package http
+
+import (
+	"github.com/assimon/luuu/util/constant"
+	"github.com/assimon/luuu/util/page"
+	"github.com/labstack/echo/v4"
+	"net/http"
+)
+
+type Resp struct{}
+
+type Response struct {
+	StatusCode int         `json:"status_code"`
+	Message    string      `json:"message"`
+	Data       interface{} `json:"data"`
+	RequestID  string      `json:"request_id"`
+}
+
+func (r *Resp) View(e echo.Context, code int, html string) error {
+	return e.HTML(code, html)
+}
+
+// SucView 成功返回
+func (r *Resp) SucView(e echo.Context, html string) error {
+	return r.View(e, http.StatusOK, html)
+}
+
+func (r *Resp) Json(e echo.Context, code int, data interface{}) error {
+	return e.JSON(code, data)
+}
+
+// SucJson 成功返回json
+func (r *Resp) SucJson(e echo.Context, data interface{}, message ...string) error {
+	rp := new(Response)
+	rp.StatusCode = http.StatusOK
+	if len(message) == 0 {
+		rp.Message = "success"
+	} else {
+		for _, m := range message {
+			rp.Message += "," + m
+		}
+	}
+	rp.Data = data
+	rp.RequestID = e.Request().Header.Get(echo.HeaderXRequestID)
+	return r.Json(e, http.StatusOK, rp)
+}
+
+// SucJsonPage 分页封装返回json
+func (r *Resp) SucJsonPage(e echo.Context, data interface{}, pagination page.Pagination, message ...string) error {
+	type PageData struct {
+		List       interface{}     `json:"list"`
+		Pagination page.Pagination `json:"pagination"`
+	}
+	pageDate := PageData{
+		List:       data,
+		Pagination: pagination,
+	}
+	return r.SucJson(e, pageDate, message...)
+}
+
+// FailJson 失败json
+func (r *Resp) FailJson(e echo.Context, err error) error {
+	rr := new(Response)
+	switch err.(type) {
+	case *constant.RspError:
+		rr.StatusCode, rr.Message = err.(*constant.RspError).Render()
+	default:
+		rr.StatusCode = 400
+		rr.Message = err.Error()
+	}
+	rr.RequestID = e.Request().Header.Get(echo.HeaderXRequestID)
+	return r.Json(e, http.StatusOK, &rr)
+}