force_ua.go 438 B

12345678910111213141516171819202122
  1. package middlewares
  2. import (
  3. "errors"
  4. "github.com/gin-gonic/gin"
  5. "github.com/runningwater/gohub/pkg/response"
  6. )
  7. // ForceUA 强制设置UA, 请求必须附带 User-Agent 标志头
  8. func ForceUA() gin.HandlerFunc {
  9. return func(c *gin.Context) {
  10. if c.Request.Header.Get("User-Agent") == "" {
  11. response.BadRequest(c, errors.New("User-Agent is required"), "请求必须附带 User-Agent 标志头")
  12. return
  13. }
  14. c.Next()
  15. }
  16. }