gateway/middleware/request.go

package middleware

import (
	"gateway/core"
	"github.com/gin-gonic/gin"
	"github.com/nacos-group/nacos-sdk-go/inner/uuid"
)

// GenRequestId 生成RequestId
func GenRequestId() gin.HandlerFunc {
	return func(c *gin.Context) {
		requestId := c.Request.Header.Get("X-Request-Id")
		if requestId == "" {
			u, err := uuid.NewV4()
			if err != nil {
				core.Log.Error("生成UUID错误: %v\n", err.Error())
			} else {
				requestId = u.String()
			}
		}
		c.Set("X-Request-Id", requestId)
		c.Writer.Header().Set("X-Request-Id", requestId)
	}
}