package cn.iocoder.yudao.module.bi.service.config;
|
|
import cn.hutool.core.collection.CollUtil;
|
import cn.hutool.core.util.StrUtil;
|
import cn.iocoder.yudao.framework.common.exception.ServiceException;
|
import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
|
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
|
import cn.iocoder.yudao.module.bi.controller.admin.config.vo.BiDataSourceConfigImportExcelVO;
|
import cn.iocoder.yudao.module.bi.controller.admin.config.vo.BiDataSourceConfigImportReqVO;
|
import cn.iocoder.yudao.module.bi.controller.admin.config.vo.BiDataSourceConfigImportRespVO;
|
import cn.iocoder.yudao.module.bi.dal.dataobject.BiDataSourceConfigDO;
|
import cn.iocoder.yudao.module.bi.dal.mysql.BiDataSourceConfigMapper;
|
import cn.iocoder.yudao.module.bi.enums.ErrorCodeConstants;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import jakarta.annotation.Resource;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import java.util.ArrayList;
|
import java.util.LinkedHashMap;
|
import java.util.List;
|
|
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
|
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.filterList;
|
|
/**
|
* 数据源配置 Service 实现
|
*
|
* @author 超级管理员
|
*/
|
@Service
|
public class BiDataSourceConfigServiceImpl implements BiDataSourceConfigService {
|
|
@Resource
|
private BiDataSourceConfigMapper mapper;
|
|
@Override
|
public Long create(BiDataSourceConfigDO config) {
|
mapper.insert(config);
|
return config.getId();
|
}
|
|
@Override
|
public void update(BiDataSourceConfigDO config) {
|
validateExists(config.getId());
|
mapper.updateById(config);
|
}
|
|
@Override
|
public void delete(Long id) {
|
validateExists(id);
|
mapper.deleteById(id);
|
}
|
|
@Override
|
public BiDataSourceConfigDO get(Long id) {
|
return mapper.selectById(id);
|
}
|
|
@Override
|
public PageResult<BiDataSourceConfigDO> getPage(Integer pageNo, Integer pageSize) {
|
IPage<BiDataSourceConfigDO> page = mapper.selectPage(new Page<>(pageNo, pageSize),
|
new LambdaQueryWrapperX<BiDataSourceConfigDO>()
|
.orderByDesc(BiDataSourceConfigDO::getCreateTime));
|
return new PageResult<>(page.getRecords(), page.getTotal());
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public BiDataSourceConfigImportRespVO importDataSourceConfigList(List<BiDataSourceConfigImportExcelVO> importList,
|
BiDataSourceConfigImportReqVO importReqVO) {
|
// 1. 校验非空
|
importList = filterList(importList, item -> StrUtil.isNotBlank(item.getName()));
|
if (CollUtil.isEmpty(importList)) {
|
throw exception(ErrorCodeConstants.BI_DATA_SOURCE_CONFIG_IMPORT_LIST_IS_EMPTY);
|
}
|
// 2. 逐条处理
|
BiDataSourceConfigImportRespVO respVO = BiDataSourceConfigImportRespVO.builder()
|
.createDataSourceConfigNames(new ArrayList<>())
|
.updateDataSourceConfigNames(new ArrayList<>())
|
.failureDataSourceConfigNames(new LinkedHashMap<>())
|
.build();
|
importList.forEach(item -> {
|
try {
|
BiDataSourceConfigDO exist = mapper.selectByName(item.getName());
|
if (exist == null) {
|
// 2.1 新增
|
BiDataSourceConfigDO config = BeanUtils.toBean(item, BiDataSourceConfigDO.class);
|
mapper.insert(config);
|
respVO.getCreateDataSourceConfigNames().add(item.getName());
|
return;
|
}
|
// 2.2 已存在，判断是否允许更新
|
if (!importReqVO.getUpdateSupport()) {
|
respVO.getFailureDataSourceConfigNames().put(item.getName(),
|
StrUtil.format(ErrorCodeConstants.BI_DATA_SOURCE_CONFIG_NAME_EXISTS.getMsg(), item.getName()));
|
return;
|
}
|
BiDataSourceConfigDO updateObj = BeanUtils.toBean(item, BiDataSourceConfigDO.class)
|
.setId(exist.getId());
|
mapper.updateById(updateObj);
|
respVO.getUpdateDataSourceConfigNames().add(item.getName());
|
} catch (ServiceException ex) {
|
respVO.getFailureDataSourceConfigNames().put(item.getName(), ex.getMessage());
|
}
|
});
|
return respVO;
|
}
|
|
private void validateExists(Long id) {
|
if (mapper.selectById(id) == null) {
|
throw exception(ErrorCodeConstants.BI_DATA_SOURCE_CONFIG_NOT_EXISTS);
|
}
|
}
|
|
}
|
