DKube v1.0

middle/jwt.go

@@ -0,0 +1,45 @@
+package middle
+
+import (
+	"dkube/utils"
+	"github.com/gin-gonic/gin"
+	"net/http"
+)
+
+func JWTAuth() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		if len(c.Request.URL.String()) >= 10 && c.Request.URL.String()[0:10] == "/api/login" {
+			c.Next()
+		} else {
+			token := c.Request.Header.Get("Authorization")
+			if token == "" {
+				c.JSON(http.StatusBadRequest, gin.H{
+					"msg":  "请求未携带token，无权限访问",
+					"data": nil,
+				})
+				c.Abort()
+				return
+			}
+
+			claims, err := utils.JWTToken.ParseToken(token)
+			if err != nil {
+				if err.Error() == "TokenExpired" {
+					c.JSON(http.StatusBadRequest, gin.H{
+						"msg":  "授权已过期",
+						"data": nil,
+					})
+					c.Abort()
+					return
+				}
+				c.JSON(http.StatusBadRequest, gin.H{
+					"msg":  err.Error(),
+					"data": nil,
+				})
+				c.Abort()
+				return
+			}
+			c.Set("claims", claims)
+			c.Next()
+		}
+	}
+}