feat(hrm): 新增人事管理模块功能

云

2026-10-25 4229657c5b6be8a1da9e59a40e2426cf6b0995c2

 yudao-module-infra/src/main/java/cn/iocoder/yudao/module/infra/service/db/DataSourceConfigServiceImpl.java

@@ -1,7 +1,13 @@
package cn.iocoder.yudao.module.infra.service.db;

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.util.object.BeanUtils;
import cn.iocoder.yudao.framework.mybatis.core.util.JdbcUtils;
import cn.iocoder.yudao.module.infra.controller.admin.db.vo.DataSourceConfigImportExcelVO;
import cn.iocoder.yudao.module.infra.controller.admin.db.vo.DataSourceConfigImportReqVO;
import cn.iocoder.yudao.module.infra.controller.admin.db.vo.DataSourceConfigImportRespVO;
import cn.iocoder.yudao.module.infra.controller.admin.db.vo.DataSourceConfigSaveReqVO;
import cn.iocoder.yudao.module.infra.dal.dataobject.db.DataSourceConfigDO;
import cn.iocoder.yudao.module.infra.dal.mysql.db.DataSourceConfigMapper;
@@ -9,12 +15,19 @@
import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DynamicDataSourceProperties;
import jakarta.annotation.Resource;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.validation.annotation.Validated;

import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Objects;

import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.filterList;
import static cn.iocoder.yudao.module.infra.enums.ErrorCodeConstants.DATA_SOURCE_CONFIG_IMPORT_LIST_IS_EMPTY;
import static cn.iocoder.yudao.module.infra.enums.ErrorCodeConstants.DATA_SOURCE_CONFIG_IMPORT_NAME_EXISTS;
import static cn.iocoder.yudao.module.infra.enums.ErrorCodeConstants.DATA_SOURCE_CONFIG_DB_TYPE_UNSUPPORTED;
import static cn.iocoder.yudao.module.infra.enums.ErrorCodeConstants.DATA_SOURCE_CONFIG_NOT_EXISTS;
import static cn.iocoder.yudao.module.infra.enums.ErrorCodeConstants.DATA_SOURCE_CONFIG_NOT_OK;

@@ -93,9 +106,19 @@
    }

    private void validateConnectionOK(DataSourceConfigDO config) {
        validateDatabaseTypeSupported(config.getUrl());
        boolean success = JdbcUtils.isConnectionOK(config.getUrl(), config.getUsername(), config.getPassword());
        if (!success) {
            throw exception(DATA_SOURCE_CONFIG_NOT_OK);
        }
    }

    /**
     * 默认轻量包只携带 MySQL 驱动。提前给出明确提示，避免缺少驱动时表现成含糊的连接失败。
     */
    private void validateDatabaseTypeSupported(String url) {
        if (!StrUtil.startWithIgnoreCase(url, "jdbc:mysql:")) {
            throw exception(DATA_SOURCE_CONFIG_DB_TYPE_UNSUPPORTED, url);
        }
    }

@@ -108,4 +131,47 @@
                .setPassword(dataSourceProperty.getPassword());
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public DataSourceConfigImportRespVO importDataSourceConfigList(List<DataSourceConfigImportExcelVO> importDataSourceConfigs,
                                                                   DataSourceConfigImportReqVO importReqVO) {
        // 1. 校验非空
        importDataSourceConfigs = filterList(importDataSourceConfigs, item -> StrUtil.isNotBlank(item.getName()));
        if (CollUtil.isEmpty(importDataSourceConfigs)) {
            throw exception(DATA_SOURCE_CONFIG_IMPORT_LIST_IS_EMPTY);
        }
        // 2. 逐条处理
        DataSourceConfigImportRespVO respVO = DataSourceConfigImportRespVO.builder()
                .createDataSourceConfigNames(new ArrayList<>())
                .updateDataSourceConfigNames(new ArrayList<>())
                .failureDataSourceConfigNames(new LinkedHashMap<>())
                .build();
        importDataSourceConfigs.forEach(importConfig -> {
            try {
                validateDatabaseTypeSupported(importConfig.getUrl());
                DataSourceConfigDO existConfig = dataSourceConfigMapper.selectByName(importConfig.getName());
                if (existConfig == null) {
                    // 新增
                    DataSourceConfigDO config = BeanUtils.toBean(importConfig, DataSourceConfigDO.class);
                    dataSourceConfigMapper.insert(config);
                    respVO.getCreateDataSourceConfigNames().add(importConfig.getName());
                    return;
                }
                // 已存在，判断是否允许更新
                if (!importReqVO.getUpdateSupport()) {
                    respVO.getFailureDataSourceConfigNames().put(importConfig.getName(),
                            StrUtil.format(DATA_SOURCE_CONFIG_IMPORT_NAME_EXISTS.getMsg(), importConfig.getName()));
                    return;
                }
                DataSourceConfigDO updateConfig = BeanUtils.toBean(importConfig, DataSourceConfigDO.class)
                        .setId(existConfig.getId());
                dataSourceConfigMapper.updateById(updateConfig);
                respVO.getUpdateDataSourceConfigNames().add(importConfig.getName());
            } catch (ServiceException ex) {
                respVO.getFailureDataSourceConfigNames().put(importConfig.getName(), ex.getMessage());
            }
        });
        return respVO;
    }

}