ziwen преди 2 години
родител
ревизия
01becb806e
променени са 2 файла, в които са добавени 10 реда и са изтрити 4 реда
  1. 5 2
      middleware/token.go
  2. 5 2
      middleware/token_no_login.go

+ 5 - 2
middleware/token.go

@@ -42,8 +42,11 @@ func Token() gin.HandlerFunc {
 
 		var userInfo services.UserInfo
 
-		if sessionInfo.OpenID != "" {
-			tmpUserInfo, tmpErr := services.GetWxUserItemByOpenId(sessionInfo.OpenID)
+		//先判断pc端登录的情况
+		tmpUserInfo, tmpErr := services.GetWxUserItemByUserId(int(sessionInfo.UserID), 3)
+		userInfo = tmpUserInfo
+		if tmpErr == utils.ErrNoRow && sessionInfo.OpenID != ""{
+			tmpUserInfo, tmpErr = services.GetWxUserItemByOpenId(sessionInfo.OpenID)
 			userInfo = tmpUserInfo
 			err = tmpErr
 		} else {

+ 5 - 2
middleware/token_no_login.go

@@ -42,8 +42,11 @@ func TokenNoLogin() gin.HandlerFunc {
 
 		var userInfo services.UserInfo
 
-		if sessionInfo.OpenID != "" {
-			tmpUserInfo, tmpErr := services.GetWxUserItemByOpenId(sessionInfo.OpenID)
+		//先判断pc端登录的情况
+		tmpUserInfo, tmpErr := services.GetWxUserItemByUserId(int(sessionInfo.UserID), 3)
+		userInfo = tmpUserInfo
+		if tmpErr == utils.ErrNoRow && sessionInfo.OpenID != ""{
+			tmpUserInfo, tmpErr = services.GetWxUserItemByOpenId(sessionInfo.OpenID)
 			userInfo = tmpUserInfo
 			err = tmpErr
 		} else {