group

pkg/common/middleware/gin.go

@@ -0,0 +1,42 @@
+package middleware
+
+import (
+	"bytes"
+	"encoding/json"
+	"github.com/gin-gonic/gin"
+	"io/ioutil"
+	"net/http"
+)
+
+func GinParseOperationID(c *gin.Context) {
+	if c.Request.Method == http.MethodPost {
+		operationID := c.Request.Header.Get("operationID")
+		if operationID == "" {
+			body, err := ioutil.ReadAll(c.Request.Body)
+			if err != nil {
+				c.String(400, "read request body error: "+err.Error())
+				c.Abort()
+				return
+			}
+			req := struct {
+				OperationID string `json:"operationID"`
+			}{}
+			if err := json.Unmarshal(body, &req); err != nil {
+				c.String(400, "get operationID error: "+err.Error())
+				c.Abort()
+				return
+			}
+			if req.OperationID == "" {
+				c.String(400, "operationID empty")
+				c.Abort()
+				return
+			}
+			c.Request.Body = ioutil.NopCloser(bytes.NewReader(body))
+			operationID = req.OperationID
+		}
+		c.Set("operationID", operationID)
+		c.Next()
+		return
+	}
+	c.Next()
+}