| | |
| | | * 校验企业名称是否唯一 |
| | | */ |
| | | @Override |
| | | public String checkCompanyNameUnique(SysTenantBo bo) { |
| | | public boolean checkCompanyNameUnique(SysTenantBo bo) { |
| | | boolean exist = baseMapper.exists(new LambdaQueryWrapper<SysTenant>() |
| | | .eq(SysTenant::getCompanyName, bo.getCompanyName()) |
| | | .ne(ObjectUtil.isNotNull(bo.getTenantId()), SysTenant::getTenantId, bo.getTenantId())); |
| | | if (exist) { |
| | | return TenantConstants.NOT_PASS; |
| | | } |
| | | return TenantConstants.PASS; |
| | | return !exist; |
| | | } |
| | | |
| | | /** |
| | | * 校验账号余额 |
| | | */ |
| | | @Override |
| | | public String checkAccountBalance(String tenantId) { |
| | | public boolean checkAccountBalance(String tenantId) { |
| | | SysTenantVo tenant = SpringUtils.getAopProxy(this).queryByTenantId(tenantId); |
| | | // 如果余额为-1代表不限制 |
| | | if (tenant.getAccountCount() == -1) { |
| | | return TenantConstants.PASS; |
| | | return true; |
| | | } |
| | | Long userNumber = sysUserMapper.selectCount(new LambdaQueryWrapper<>()); |
| | | // 如果余额大于0代表还有可用名额 |
| | | if (tenant.getAccountCount() - userNumber > 0) { |
| | | return TenantConstants.PASS; |
| | | } |
| | | return TenantConstants.NOT_PASS; |
| | | return tenant.getAccountCount() - userNumber > 0; |
| | | } |
| | | |
| | | /** |
| | | * 校验有效期 |
| | | */ |
| | | @Override |
| | | public String checkExpireTime(String tenantId) { |
| | | public boolean checkExpireTime(String tenantId) { |
| | | SysTenantVo tenant = SpringUtils.getAopProxy(this).queryByTenantId(tenantId); |
| | | // 如果未设置过期时间代表不限制 |
| | | if (ObjectUtil.isNull(tenant.getExpireTime())) { |
| | | return TenantConstants.PASS; |
| | | return true; |
| | | } |
| | | // 如果当前时间在过期时间之前则通过 |
| | | if (new Date().before(tenant.getExpireTime())) { |
| | | return TenantConstants.PASS; |
| | | } |
| | | return TenantConstants.NOT_PASS; |
| | | return new Date().before(tenant.getExpireTime()); |
| | | } |
| | | |
| | | /** |
| | | * 同步租户套餐 |
| | | */ |
| | | @Override |
| | | @Transactional(rollbackFor = Exception.class) |
| | | public Boolean syncTenantPackage(String tenantId, String packageId) { |
| | | TenantHelper.enableIgnore(); |
| | | SysTenantPackage tenantPackage = sysTenantPackageMapper.selectById(packageId); |
| | | List<SysRole> roles = sysRoleMapper.selectList( |
| | | new LambdaQueryWrapper<SysRole>().eq(SysRole::getTenantId, tenantId)); |
| | | List<Long> roleIds = new ArrayList<>(roles.size() - 1); |
| | | List<Long> menuIds = StringUtils.splitTo(tenantPackage.getMenuIds(), Convert::toLong); |
| | | roles.forEach(item -> { |
| | | if (TenantConstants.TENANT_ADMIN_ROLE_KEY.equals(item.getRoleKey())) { |
| | | List<SysRoleMenu> roleMenus = new ArrayList<>(menuIds.size()); |
| | | menuIds.forEach(menuId -> { |
| | | SysRoleMenu roleMenu = new SysRoleMenu(); |
| | | roleMenu.setRoleId(item.getRoleId()); |
| | | roleMenu.setMenuId(menuId); |
| | | roleMenus.add(roleMenu); |
| | | }); |
| | | sysRoleMenuMapper.delete(new LambdaQueryWrapper<SysRoleMenu>().eq(SysRoleMenu::getRoleId, item.getRoleId())); |
| | | sysRoleMenuMapper.insertBatch(roleMenus); |
| | | } else { |
| | | roleIds.add(item.getRoleId()); |
| | | } |
| | | }); |
| | | if (!roleIds.isEmpty()) { |
| | | sysRoleMenuMapper.delete( |
| | | new LambdaQueryWrapper<SysRoleMenu>().in(SysRoleMenu::getRoleId, roleIds).notIn(!menuIds.isEmpty(), SysRoleMenu::getMenuId, menuIds)); |
| | | } |
| | | TenantHelper.disableIgnore(); |
| | | return true; |
| | | } |
| | | } |