瀏覽代碼

jwt补充

May1145 1 年之前
父節點
當前提交
6b48560400
共有 1 個文件被更改,包括 12 次插入0 次删除
  1. 12 0
      trading-go/controller/middleware.go

+ 12 - 0
trading-go/controller/middleware.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"github.com/gin-gonic/gin"
 	"net/http"
+	"trading-go/util"
 )
 
 func Cors(c *gin.Context) {
@@ -20,3 +21,14 @@ func Cors(c *gin.Context) {
 	}
 	c.Next()
 }
+
+func VerifyToken(c *gin.Context) {
+	var tokenString string
+	tokenString = c.GetHeader("token")
+	t, _, _ := util.ParseToken(tokenString)
+	if t.Valid {
+		c.Next()
+	} else {
+		c.Abort()
+	}
+}