errcode

infra/response.go

@@ -0,0 +1,69 @@
+package infra
+
+import (
+	"net/http"
+	"time"
+
+	"git.apinb.com/bsm-sdk/core/vars"
+	"github.com/gofiber/fiber/v2"
+)
+
+var Response Reply
+
+type Reply struct {
+	Ret  int         `json:"ret"`
+	Msg  string      `json:"msg"`
+	Data interface{} `json:"data"`
+}
+
+type ReplyIdent struct {
+	ID      uint   `json:"id"`
+	Identiy string `json:"identity"`
+}
+
+func (r *Reply) Error(ctx *fiber.Ctx, err error) error {
+	// Status code defaults to 500
+	reply := Reply{
+		Ret: http.StatusInternalServerError,
+	}
+	if e, ok := err.(*fiber.Error); ok {
+		reply.Ret = e.Code
+	}
+
+	// 获取执行信息
+	ExcuteTime(ctx)
+
+	reply.Msg = err.Error()
+
+	// Send error
+	return ctx.Status(fiber.StatusInternalServerError).JSON(reply)
+}
+func (r *Reply) Success(ctx *fiber.Ctx, data interface{}) error {
+	reply := Reply{
+		Ret: http.StatusOK,
+	}
+
+	if data == nil {
+		reply.Data = ""
+	} else {
+		reply.Data = data
+	}
+
+	// 获取执行信息
+	ExcuteTime(ctx)
+
+	return ctx.Status(fiber.StatusOK).JSON(reply)
+}
+
+func ExcuteTime(ctx *fiber.Ctx) error {
+	// 从context中获取开始时间
+	startTime, ok := ctx.Locals("startTime").(time.Time)
+	if !ok {
+		// 如果转换失败，则默认为当前时间
+		startTime = time.Now()
+	}
+
+	exectime := time.Since(startTime)
+	ctx.Response().Header.Add("Execute-Node", vars.HostName+","+exectime.String())
+	return nil
+}