|
@@ -9,6 +9,7 @@ import cn.hutool.core.util.IdUtil;
|
|
|
import cn.hutool.core.util.ObjectUtil;
|
|
import cn.hutool.core.util.ObjectUtil;
|
|
|
import cn.hutool.core.util.StrUtil;
|
|
import cn.hutool.core.util.StrUtil;
|
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|
|
|
|
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
|
|
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
|
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
|
|
import com.baomidou.mybatisplus.core.toolkit.StringPool;
|
|
import com.baomidou.mybatisplus.core.toolkit.StringPool;
|
|
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
|
@@ -22,6 +23,8 @@ import com.xjrsoft.common.utils.RedisUtil;
|
|
|
import com.xjrsoft.common.utils.WeChatUtil;
|
|
import com.xjrsoft.common.utils.WeChatUtil;
|
|
|
import com.xjrsoft.config.CommonPropertiesConfig;
|
|
import com.xjrsoft.config.CommonPropertiesConfig;
|
|
|
import com.xjrsoft.config.LicenseConfig;
|
|
import com.xjrsoft.config.LicenseConfig;
|
|
|
|
|
+import com.xjrsoft.module.base.entity.WhitelistManagement;
|
|
|
|
|
+import com.xjrsoft.module.base.service.IWhitelistManagementService;
|
|
|
import com.xjrsoft.module.organization.entity.Department;
|
|
import com.xjrsoft.module.organization.entity.Department;
|
|
|
import com.xjrsoft.module.organization.entity.Post;
|
|
import com.xjrsoft.module.organization.entity.Post;
|
|
|
import com.xjrsoft.module.organization.entity.User;
|
|
import com.xjrsoft.module.organization.entity.User;
|
|
@@ -85,6 +88,8 @@ public class LoginServiceImpl implements ILoginService {
|
|
|
|
|
|
|
|
private final CommonPropertiesConfig propertiesConfig;
|
|
private final CommonPropertiesConfig propertiesConfig;
|
|
|
|
|
|
|
|
|
|
+ private final IWhitelistManagementService whitelistManagementService;
|
|
|
|
|
+
|
|
|
@Override
|
|
@Override
|
|
|
public LoginVo login(LoginDto dto) throws Exception {
|
|
public LoginVo login(LoginDto dto) throws Exception {
|
|
|
if (licenseConfig.getEnabled()) {
|
|
if (licenseConfig.getEnabled()) {
|
|
@@ -147,6 +152,14 @@ public class LoginServiceImpl implements ILoginService {
|
|
|
}
|
|
}
|
|
|
LoginVo loginVo = getLoginInfo(loginUser, "WX-MP");
|
|
LoginVo loginVo = getLoginInfo(loginUser, "WX-MP");
|
|
|
result.setToken(loginVo.getToken());
|
|
result.setToken(loginVo.getToken());
|
|
|
|
|
+ result.setWhitelistStatus(0);
|
|
|
|
|
+ List<WhitelistManagement> managementList = whitelistManagementService.list(
|
|
|
|
|
+ new QueryWrapper<WhitelistManagement>().lambda()
|
|
|
|
|
+ .eq(WhitelistManagement::getUserId, loginUser.getId())
|
|
|
|
|
+ );
|
|
|
|
|
+ if(!managementList.isEmpty()){
|
|
|
|
|
+ result.setWhitelistStatus(1);
|
|
|
|
|
+ }
|
|
|
result.setUserType(loginVo.getUserType());
|
|
result.setUserType(loginVo.getUserType());
|
|
|
|
|
|
|
|
return result;
|
|
return result;
|