1

controllers/cms_store.go

@@ -727,13 +727,15 @@ func (c *StoreController) GetStoreCategoryMap() {
 				return nil, "", err
 			}
 
-			_, total, err := dao.GetUsers(dao.GetDB(), 1, "", []string{userAuth.UserID}, nil, nil, 0, 1)
-			//userRole, err := permission.GetUserRole(params.Ctx, userAuth.UserID)
+			user, total, err := dao.GetUsers(dao.GetDB(), 1, "", []string{userAuth.UserID}, nil, nil, 0, 1)
 			if err != nil {
 				return nil, "", err
 			}
-			if total < 0 { // 普通用户，无用户权限
-				return nil, "", errors.New("用户为普通用户，无法查看B2B门店信息")
+			if total != model.YES {
+				return nil, "", errors.New("未注册用户")
+			}
+			if user[0].Type == model.YES { // 普通用户，无用户权限
+				return nil, "", errors.New("用户为普通用户，无法查看B2B门店,物料店信息")
 			}
 		} else {
 			result := make([]*model.StoreCategoryMap, 0, 0)