May1145 hai 1 ano
pai
achega
6b48560400
Modificáronse 1 ficheiros con 12 adicións e 0 borrados
  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()
+	}
+}