初始化

response.go

@@ -0,0 +1,133 @@
+package responsex
+
+// Author: Yun
+// Version: 2023年6月12日19:25:54
+
+import (
+	"context"
+	"encoding/json"
+	"math"
+	"net/http"
+
+	"github.com/yuninks/langx"
+	"github.com/zeromicro/go-zero/core/trace"
+)
+
+type pagination struct {
+	Page       int64 `json:"page"`
+	Size       int64 `json:"size"`
+	TotalPage  int64 `json:"total_page"`
+	TotalCount int64 `json:"total_count"`
+}
+
+// 计算DB分页偏移量
+func GetOffset(page, size int) (offset int, limit int) {
+	if page <= 0 {
+		page = 1
+	}
+	offset = (page - 1) * size
+	limit = size
+	if limit == 0 {
+		// 默认查10条
+		limit = 10
+	}
+	return
+}
+
+// 计算响应分页
+func GetPage(page, size, totalCount int64) pagination {
+	pa := pagination{}
+	totalPage := 0
+	if size != 0 {
+		totalPage = int(math.Ceil(float64(totalCount) / float64(size)))
+	}
+	pa.Page = page
+	pa.Size = size
+	pa.TotalPage = int64(totalPage)
+	pa.TotalCount = totalCount
+	return pa
+}
+
+// Msg格式化响应
+func FormatMessage(ctx context.Context, w http.ResponseWriter, message string, format map[string]string, data interface{}) {
+	code, msg := langx.GetTrans(message, message, format)
+	ResponseCtx(ctx, w, code, msg, data)
+}
+
+// 成功的响应
+func Success(ctx context.Context, w http.ResponseWriter, data interface{}, info ...interface{}) {
+	ResponseCtx(ctx, w, 200, "请求成功", data, info)
+}
+
+func SuccessWithPage(ctx context.Context, w http.ResponseWriter, data interface{}, page pagination) {
+	ResponseCtx(ctx, w, 200, "请求成功", data, page)
+}
+
+// 失败的响应
+func Error(ctx context.Context, w http.ResponseWriter, err error) {
+	code := 400
+	msg := "请求失败"
+	if err != nil {
+		val, ok := err.(*langx.LangError)
+		if ok {
+			code = val.Code()
+			msg = err.Error()
+		}
+	}
+	ResponseCtx(ctx, w, code, msg, "")
+}
+
+func ErrorStr(ctx context.Context, w http.ResponseWriter, msg string) {
+	code, msg := langx.GetTrans("zh-CN", msg, nil)
+	ResponseCtx(ctx, w, code, msg, "")
+}
+
+// 基础的响应
+func ResponseCtx(ctx context.Context, w http.ResponseWriter, code int, message string, data interface{}, other ...interface{}) {
+
+	res := make(map[string]interface{})
+	res["code"] = code
+	res["message"] = message
+
+	// 分页的内容
+	var page *pagination
+	for _, value := range other {
+		vv, ok := value.(pagination)
+		if ok {
+			page = &vv
+		}
+	}
+
+	if page == nil {
+		// 不需要分页
+		res["data"] = data
+	} else {
+		res["data"] = data
+		res["pagination"] = page
+	}
+
+	// 获取trace_id
+	if op.traceId != "" {
+		t := ctx.Value(op.traceId)
+		if t != nil {
+			res["trace_id"] = t
+		} else {
+			// go-zero
+			res["trace_id"] = trace.TraceIDFromContext(ctx)
+		}
+	}
+
+	if !op.ignoreLog {
+		// 记录日志
+		op.logger.Info(ctx, res)
+	}
+
+	bs, _ := json.Marshal(res)
+
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	w.WriteHeader(200)
+	n, err := w.Write(bs)
+	if err != nil || n != len(bs) {
+		op.logger.Error(ctx, "write response failed err:%+v", err)
+	}
+}