From 4229657c5b6be8a1da9e59a40e2426cf6b0995c2 Mon Sep 17 00:00:00 2001
From: 云 <2163098428@qq.com>
Date: 星期日, 25 十月 2026 12:01:55 +0800
Subject: [PATCH] feat(hrm): 新增人事管理模块功能
---
yudao-module-infra/src/main/java/cn/iocoder/yudao/module/infra/service/db/DataSourceConfigServiceImpl.java | 66 +++++++++++++++++++++++++++++++++
1 files changed, 66 insertions(+), 0 deletions(-)
diff --git a/yudao-module-infra/src/main/java/cn/iocoder/yudao/module/infra/service/db/DataSourceConfigServiceImpl.java b/yudao-module-infra/src/main/java/cn/iocoder/yudao/module/infra/service/db/DataSourceConfigServiceImpl.java
index 75ba722..58f2d43 100644
--- a/yudao-module-infra/src/main/java/cn/iocoder/yudao/module/infra/service/db/DataSourceConfigServiceImpl.java
+++ b/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;
+ }
+
}
--
Gitblit v1.9.3