强制 User Agent 中间件

Author: summerblue

app/http/middlewares/force_ua.go

@@ -0,0 +1,23 @@
+// Package middlewares Gin 中间件
+package middlewares
+
+import (
+	"errors"
+	"gohub/pkg/response"
+
+	"github.com/gin-gonic/gin"
+)
+
+// ForceUA 中间件，强制请求必须附带 User-Agent 标头
+func ForceUA() gin.HandlerFunc {
+	return func(c *gin.Context) {
+
+		// 获取 User-Agent 标头信息
+		if len(c.Request.Header["User-Agent"]) == 0 {
+			response.BadRequest(c, errors.New("User-Agent 标头未找到"), "请求必须附带 User-Agent 标头")
+			return
+		}
+
+		c.Next()
+	}
+}

bootstrap/route.go

@@ -27,6 +27,7 @@ func registerGlobalMiddleWare(router *gin.Engine) {
 	router.Use(
 		middlewares.Logger(),
 		middlewares.Recovery(),
+		middlewares.ForceUA(),
 	)
 }