package cn.iocoder.yudao.module.system.service.tenant;
|
|
import cn.hutool.core.collection.CollUtil;
|
import cn.hutool.core.util.StrUtil;
|
import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
|
import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
|
import cn.iocoder.yudao.module.system.controller.admin.tenant.vo.packages.TenantPackagePageReqVO;
|
import cn.iocoder.yudao.module.system.controller.admin.tenant.vo.packages.TenantPackageSaveReqVO;
|
import cn.iocoder.yudao.module.system.dal.dataobject.tenant.TenantDO;
|
import cn.iocoder.yudao.module.system.dal.dataobject.tenant.TenantPackageDO;
|
import cn.iocoder.yudao.module.system.dal.mysql.tenant.TenantPackageMapper;
|
import com.baomidou.dynamic.datasource.annotation.DSTransactional;
|
import com.google.common.annotations.VisibleForTesting;
|
import jakarta.annotation.Resource;
|
import org.springframework.context.annotation.Lazy;
|
import org.springframework.stereotype.Service;
|
import org.springframework.validation.annotation.Validated;
|
|
import java.util.List;
|
|
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
|
import static cn.iocoder.yudao.module.system.enums.ErrorCodeConstants.*;
|
|
/**
|
* 租户套餐 Service 实现类
|
*
|
* @author 芋道源码
|
*/
|
@Service
|
@Validated
|
public class TenantPackageServiceImpl implements TenantPackageService {
|
|
@Resource
|
private TenantPackageMapper tenantPackageMapper;
|
|
@Resource
|
@Lazy // 避免循环依赖的报错
|
private TenantService tenantService;
|
|
@Override
|
public Long createTenantPackage(TenantPackageSaveReqVO createReqVO) {
|
// 校验套餐名是否重复
|
validateTenantPackageNameUnique(null, createReqVO.getName());
|
// 插入
|
TenantPackageDO tenantPackage = BeanUtils.toBean(createReqVO, TenantPackageDO.class);
|
tenantPackageMapper.insert(tenantPackage);
|
// 返回
|
return tenantPackage.getId();
|
}
|
|
@Override
|
@DSTransactional // 多数据源，使用 @DSTransactional 保证本地事务，以及数据源的切换
|
public void updateTenantPackage(TenantPackageSaveReqVO updateReqVO) {
|
// 校验存在
|
TenantPackageDO tenantPackage = validateTenantPackageExists(updateReqVO.getId());
|
// 校验套餐名是否重复
|
validateTenantPackageNameUnique(updateReqVO.getId(), updateReqVO.getName());
|
// 更新
|
TenantPackageDO updateObj = BeanUtils.toBean(updateReqVO, TenantPackageDO.class);
|
tenantPackageMapper.updateById(updateObj);
|
// 如果菜单发生变化，则修改每个租户的菜单
|
if (!CollUtil.isEqualList(tenantPackage.getMenuIds(), updateReqVO.getMenuIds())) {
|
List<TenantDO> tenants = tenantService.getTenantListByPackageId(tenantPackage.getId());
|
tenants.forEach(tenant -> tenantService.updateTenantRoleMenu(tenant.getId(), updateReqVO.getMenuIds()));
|
}
|
}
|
|
@Override
|
public void deleteTenantPackage(Long id) {
|
// 校验存在
|
validateTenantPackageExists(id);
|
// 校验正在使用
|
validateTenantUsed(id);
|
// 删除
|
tenantPackageMapper.deleteById(id);
|
}
|
|
@Override
|
public void deleteTenantPackageList(List<Long> ids) {
|
// 1. 校验是否有租户正在使用该套餐
|
for (Long id : ids) {
|
if (tenantService.getTenantCountByPackageId(id) > 0) {
|
throw exception(TENANT_PACKAGE_USED);
|
}
|
}
|
|
// 2. 批量删除
|
tenantPackageMapper.deleteByIds(ids);
|
}
|
|
private TenantPackageDO validateTenantPackageExists(Long id) {
|
TenantPackageDO tenantPackage = tenantPackageMapper.selectById(id);
|
if (tenantPackage == null) {
|
throw exception(TENANT_PACKAGE_NOT_EXISTS);
|
}
|
return tenantPackage;
|
}
|
|
private void validateTenantUsed(Long id) {
|
if (tenantService.getTenantCountByPackageId(id) > 0) {
|
throw exception(TENANT_PACKAGE_USED);
|
}
|
}
|
|
@Override
|
public TenantPackageDO getTenantPackage(Long id) {
|
return tenantPackageMapper.selectById(id);
|
}
|
|
@Override
|
public PageResult<TenantPackageDO> getTenantPackagePage(TenantPackagePageReqVO pageReqVO) {
|
return tenantPackageMapper.selectPage(pageReqVO);
|
}
|
|
@Override
|
public TenantPackageDO validTenantPackage(Long id) {
|
TenantPackageDO tenantPackage = tenantPackageMapper.selectById(id);
|
if (tenantPackage == null) {
|
throw exception(TENANT_PACKAGE_NOT_EXISTS);
|
}
|
if (tenantPackage.getStatus().equals(CommonStatusEnum.DISABLE.getStatus())) {
|
throw exception(TENANT_PACKAGE_DISABLE, tenantPackage.getName());
|
}
|
return tenantPackage;
|
}
|
|
@Override
|
public List<TenantPackageDO> getTenantPackageListByStatus(Integer status) {
|
return tenantPackageMapper.selectListByStatus(status);
|
}
|
|
|
@VisibleForTesting
|
void validateTenantPackageNameUnique(Long id, String name) {
|
if (StrUtil.isBlank(name)) {
|
return;
|
}
|
TenantPackageDO tenantPackage = tenantPackageMapper.selectByName(name);
|
if (tenantPackage == null) {
|
return;
|
}
|
// 如果 id 为空，说明不用比较是否为相同 id 的用户
|
if (id == null) {
|
throw exception(TENANT_PACKAGE_NAME_DUPLICATE);
|
}
|
if (!tenantPackage.getId().equals(id)) {
|
throw exception(TENANT_PACKAGE_NAME_DUPLICATE);
|
}
|
}
|
|
}
|
