paddy преди 3 години
родител
ревизия
328393f509
променени са 1 файла, в които са добавени 6 реда и са изтрити 0 реда
  1. 6 0
      middleware/login_ware.go

+ 6 - 0
middleware/login_ware.go

@@ -14,6 +14,12 @@ func LoginWare() gin.HandlerFunc {
 	return func(c *gin.Context) {
 		//验证登陆
 		token := c.GetHeader("token")
+		if token == "" {
+			ts := strings.Split(c.Request.RequestURI, "___t=")
+			if len(ts) > 1 {
+				token = ts[1]
+			}
+		}
 		if token == "" {
 			c.AbortWithStatusJSON(401, sysmodels.SysReturn{401, "token is empty", ""})
 			c.Redirect(302, "/login")