rbac/middleware/auth.go
zhangkun9038@dingtalk.com 375b305ab4 aliyun-deepseek-r1
2025-02-20 11:29:57 +08:00

28 lines
521 B
Go

package middlewares
import (
"github.com/gin-gonic/gin"
"net/http"
)
func AuthMiddleware() gin.HandlerFunc {
return func(c *gin.Context) {
tokenString := c.GetHeader("Authorization")
if tokenString == "" {
utils.Error(c, http.StatusUnauthorized, "Authorization header required")
c.Abort()
return
}
claims, err := utils.ParseToken(tokenString)
if err != nil {
utils.Error(c, http.StatusUnauthorized, "Invalid token")
c.Abort()
return
}
c.Set("userID", claims.UserID)
c.Next()
}
}