登录优化

app/admin/controller/User.php

@@ -41,14 +41,18 @@ class User extends Base
                 break;
 
             case 2:
+                //获取默认公司
+                $default = array_column($rs['data']['company_relaton'], 'companyNo', 'is_main');
+                if (!isset($default) || !isset($default[1])) return json_show(1004, '没有默认公司,无法登录');
+
                 $role = Db::name('user_role')
                     ->alias('a')
                     ->field('a.roleid role_id,b.role_name,b.role_level')
                     ->leftJoin('role b', 'b.id=a.roleid')
-                    ->where(['a.uid' => $rs['data']['id'], 'a.is_main' => 1, 'a.is_del' => 0])
+                    ->where(['a.uid' => $rs['data']['id'], 'a.companyNo' => $default[1], 'a.is_del' => 0])
                     ->findOrEmpty();
 
-                if (empty($role)) return json_show(1004, '没有默认角色，无法登录，请联系管理员');
+                if (empty($role)) return json_show(1004, '没有默认角色,无法登录,请联系管理员');
                 break;
 
             case 3: