|
|
@@ -12,6 +12,8 @@ import com.ydtech.constants.SysConstants;
|
|
|
import com.ydtech.constants.sys.SysPasswordEnum;
|
|
|
import com.ydtech.core.page.HttpResult;
|
|
|
import com.ydtech.exception.SystemException;
|
|
|
+import com.ydtech.modules.admin.dao.SysPartnerMapper;
|
|
|
+import com.ydtech.modules.admin.model.SysPartner;
|
|
|
import com.ydtech.modules.admin.model.SysPassword;
|
|
|
import com.ydtech.modules.admin.model.SysRole;
|
|
|
import com.ydtech.modules.admin.model.SysUser;
|
|
|
@@ -79,6 +81,8 @@ public class SysLoginController {
|
|
|
|
|
|
private final SysPartnerService sysPartnerService;
|
|
|
|
|
|
+ private final SysPartnerMapper sysPartnerMapper;
|
|
|
+
|
|
|
private final SysAgencyLeaderService sysAgencyLeaderService;
|
|
|
|
|
|
@GetMapping("captcha.jpg")
|
|
|
@@ -270,8 +274,10 @@ public class SysLoginController {
|
|
|
boolean a = sysPartnerService.checkPartner(user.getId());
|
|
|
boolean b = sysAgencyLeaderService.checkUserId(user.getId());
|
|
|
if (a) {
|
|
|
+ SysPartner sysPartner = sysPartnerMapper.selectOne(new LambdaQueryWrapper<SysPartner>().eq(SysPartner::getUserId, user.getId()));
|
|
|
//类型:1.合伙人;2机构负责人
|
|
|
aapLoginVo.setType(1);
|
|
|
+ aapLoginVo.setLevel(sysPartner.getType());
|
|
|
} else if (b) {
|
|
|
aapLoginVo.setType(2);
|
|
|
} else {
|