Browse Source

jwt补充

May1145 1 năm trước cách đây
mục cha
commit
6b48560400
1 tập tin đã thay đổi với 12 bổ sung0 xóa
  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()
+	}
+}