init

src/middleware/check_sign.go

@@ -0,0 +1,37 @@
+package middleware
+
+import (
+	"bytes"
+	"github.com/assimon/luuu/config"
+	"github.com/assimon/luuu/util/constant"
+	"github.com/assimon/luuu/util/json"
+	"github.com/assimon/luuu/util/sign"
+	"github.com/labstack/echo/v4"
+	"io/ioutil"
+)
+
+func CheckApiSign() echo.MiddlewareFunc {
+	return func(next echo.HandlerFunc) echo.HandlerFunc {
+		return func(ctx echo.Context) error {
+			params, err := ioutil.ReadAll(ctx.Request().Body)
+			if err != nil {
+				return constant.SignatureErr
+			}
+			m := make(map[string]interface{})
+			err = json.Cjson.Unmarshal(params, &m)
+			signature, ok := m["signature"]
+			if !ok {
+				return constant.SignatureErr
+			}
+			checkSignature, err := sign.Get(m, config.GetApiAuthToken())
+			if err != nil {
+				return constant.SignatureErr
+			}
+			if checkSignature != signature {
+				return constant.SignatureErr
+			}
+			ctx.Request().Body = ioutil.NopCloser(bytes.NewBuffer(params))
+			return next(ctx)
+		}
+	}
+}

src/middleware/request_id.go

@@ -0,0 +1,18 @@
+package middleware
+
+import (
+	"github.com/labstack/echo/v4"
+	"github.com/labstack/echo/v4/middleware"
+	uuid "github.com/satori/go.uuid"
+)
+
+func RequestUUID() echo.MiddlewareFunc {
+	return middleware.RequestIDWithConfig(middleware.RequestIDConfig{
+		Generator: func() string {
+			return uuid.NewV4().String()
+		},
+		RequestIDHandler: func(c echo.Context, s string) {
+			c.Request().Header.Set(echo.HeaderXRequestID, s)
+		},
+	})
+}