|
@@ -43,7 +43,7 @@ import java.util.concurrent.TimeUnit;
|
|
|
|
|
|
|
|
/**
|
|
/**
|
|
|
* 登录校验方法
|
|
* 登录校验方法
|
|
|
- *
|
|
|
|
|
|
|
+ *
|
|
|
|
|
|
|
|
*/
|
|
*/
|
|
|
@Component
|
|
@Component
|
|
@@ -57,7 +57,7 @@ public class SysLoginService
|
|
|
|
|
|
|
|
@Autowired
|
|
@Autowired
|
|
|
private RedisCache redisCache;
|
|
private RedisCache redisCache;
|
|
|
-
|
|
|
|
|
|
|
+
|
|
|
@Autowired
|
|
@Autowired
|
|
|
private ISysUserService userService;
|
|
private ISysUserService userService;
|
|
|
|
|
|
|
@@ -77,7 +77,7 @@ public class SysLoginService
|
|
|
|
|
|
|
|
/**
|
|
/**
|
|
|
* 登录验证
|
|
* 登录验证
|
|
|
- *
|
|
|
|
|
|
|
+ *
|
|
|
* @param username 用户名
|
|
* @param username 用户名
|
|
|
* @param password 密码
|
|
* @param password 密码
|
|
|
* @param code 验证码
|
|
* @param code 验证码
|
|
@@ -154,7 +154,7 @@ public class SysLoginService
|
|
|
|
|
|
|
|
/**
|
|
/**
|
|
|
* 校验验证码
|
|
* 校验验证码
|
|
|
- *
|
|
|
|
|
|
|
+ *
|
|
|
* @param username 用户名
|
|
* @param username 用户名
|
|
|
* @param code 验证码
|
|
* @param code 验证码
|
|
|
* @param uuid 唯一标识
|
|
* @param uuid 唯一标识
|
|
@@ -227,24 +227,21 @@ public class SysLoginService
|
|
|
|
|
|
|
|
// 默认使用主库
|
|
// 默认使用主库
|
|
|
DynamicDataSourceContextHolder.setDataSourceType(DataSourceType.MASTER.name());
|
|
DynamicDataSourceContextHolder.setDataSourceType(DataSourceType.MASTER.name());
|
|
|
-
|
|
|
|
|
- // ===== 只有传了 tenantCode 才走租户逻辑 =====
|
|
|
|
|
- if (StringUtils.isNotBlank(tenantCode))
|
|
|
|
|
|
|
+
|
|
|
|
|
+ // 查询租户(主库)
|
|
|
|
|
+ tenantInfo = userService.getTenantInfo(tenantCode);
|
|
|
|
|
+ if (BeanUtil.isEmpty(tenantInfo))
|
|
|
{
|
|
{
|
|
|
- // 查询租户(主库)
|
|
|
|
|
- tenantInfo = userService.getTenantInfo(tenantCode);
|
|
|
|
|
- if (BeanUtil.isEmpty(tenantInfo))
|
|
|
|
|
- {
|
|
|
|
|
- throw new ServiceException("企业不存在");
|
|
|
|
|
- }
|
|
|
|
|
- if (!tenantInfo.getStatus().equals(1))
|
|
|
|
|
- {
|
|
|
|
|
- throw new ServiceException("企业已禁用");
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- // 切到租户库
|
|
|
|
|
- tenantDataSourceManager.switchTenant(tenantInfo);
|
|
|
|
|
|
|
+ throw new ServiceException("企业不存在");
|
|
|
}
|
|
}
|
|
|
|
|
+ if (!tenantInfo.getStatus().equals(1))
|
|
|
|
|
+ {
|
|
|
|
|
+ throw new ServiceException("企业已禁用");
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ // 切到租户库
|
|
|
|
|
+ tenantDataSourceManager.switchTenant(tenantInfo);
|
|
|
|
|
+
|
|
|
|
|
|
|
|
try
|
|
try
|
|
|
{
|
|
{
|