|
@@ -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()
|
|
|
+ }
|
|
|
+}
|