From 8a897e1066ecccabc0201ecc3955e765e75305f3 Mon Sep 17 00:00:00 2001
From: chao <3072464591@QQ.com>
Date: Fri, 26 Apr 2024 10:36:04 +0800
Subject: [PATCH] =?UTF-8?q?feat():=20=E5=A2=9E=E5=8A=A0=E5=BC=82=E6=AD=A5?=
 =?UTF-8?q?=E5=90=8C=E6=AD=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/com/etl/ETLDataSourceApplication.java    |  2 ++
 .../service/impl/DataSourceServiceImpl.java       | 15 +++++++++++----
 2 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/etl-modules/etl-modules-data-source/etl-modules-data-source-system/src/main/java/com/etl/ETLDataSourceApplication.java b/etl-modules/etl-modules-data-source/etl-modules-data-source-system/src/main/java/com/etl/ETLDataSourceApplication.java
index 47261ea..86d2335 100644
--- a/etl-modules/etl-modules-data-source/etl-modules-data-source-system/src/main/java/com/etl/ETLDataSourceApplication.java
+++ b/etl-modules/etl-modules-data-source/etl-modules-data-source-system/src/main/java/com/etl/ETLDataSourceApplication.java
@@ -5,6 +5,7 @@ import com.etl.common.security.annotation.EnableMyFeignClients;
 import com.etl.common.swagger.annotation.EnableCustomSwagger2;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.scheduling.annotation.EnableAsync;
 
 /**
  * 数据源模块
@@ -15,6 +16,7 @@ import org.springframework.boot.autoconfigure.SpringBootApplication;
 @EnableCustomSwagger2
 @EnableMyFeignClients
 @SpringBootApplication
+@EnableAsync
 public class ETLDataSourceApplication {
     public static void main (String[] args) {
         SpringApplication.run(ETLDataSourceApplication.class, args);
diff --git a/etl-modules/etl-modules-data-source/etl-modules-data-source-system/src/main/java/com/etl/data/source/service/impl/DataSourceServiceImpl.java b/etl-modules/etl-modules-data-source/etl-modules-data-source-system/src/main/java/com/etl/data/source/service/impl/DataSourceServiceImpl.java
index 80f15b6..ec8f7b6 100644
--- a/etl-modules/etl-modules-data-source/etl-modules-data-source-system/src/main/java/com/etl/data/source/service/impl/DataSourceServiceImpl.java
+++ b/etl-modules/etl-modules-data-source/etl-modules-data-source-system/src/main/java/com/etl/data/source/service/impl/DataSourceServiceImpl.java
@@ -19,6 +19,7 @@ import com.etl.data.type.domain.DataType;
 import com.etl.data.type.service.IDataTypeService;
 import lombok.extern.log4j.Log4j2;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.scheduling.annotation.Async;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 import redis.clients.jedis.Jedis;
@@ -370,6 +371,7 @@ public class DataSourceServiceImpl extends ServiceImpl<DataSourceMapper, DataSou
      * @param dataSource     数据源信息对象
      * @param dataType       数据类型对象
      */
+    @Async
     public Map<String, Object> jdbcSync(AssetStructure assetStructure, DataSource dataSource, DataType dataType) {
         if (dataType.getDriverManager() != null && dataType.getJdbcPre() != null) {
             if ("mysql".equals(dataType.getDataType())) {
@@ -481,6 +483,7 @@ public class DataSourceServiceImpl extends ServiceImpl<DataSourceMapper, DataSou
      * @param assetStructureTableList 数据表列表
      * @param conn                    连接
      */
+    @Async
     public void assetTableDetailsSync(List<AssetStructureTable> assetStructureTableList, Connection conn) {
         List<AssetTableDetails> assetTableDetails = new ArrayList<>();
         try {
@@ -514,7 +517,8 @@ public class DataSourceServiceImpl extends ServiceImpl<DataSourceMapper, DataSou
      * @return
      * @throws SQLException
      */
-    private ResultSet getColumns(DatabaseMetaData metaData, Connection conn, AssetStructureTable assetStructureTable) throws SQLException {
+    @Async
+    public ResultSet getColumns(DatabaseMetaData metaData, Connection conn, AssetStructureTable assetStructureTable) throws SQLException {
         return metaData.getColumns(null, null, assetStructureTable.getTableName(), null);
     }
 
@@ -530,7 +534,8 @@ public class DataSourceServiceImpl extends ServiceImpl<DataSourceMapper, DataSou
      * @throws IllegalAccessException
      * @throws InstantiationException
      */
-    private AssetTableDetails processRow(ResultSet rs, DatabaseMetaData metaData, Connection conn, AssetStructureTable assetStructureTable) throws SQLException, ClassNotFoundException, IllegalAccessException, InstantiationException {
+    @Async
+    public AssetTableDetails processRow(ResultSet rs, DatabaseMetaData metaData, Connection conn, AssetStructureTable assetStructureTable) throws SQLException, ClassNotFoundException, IllegalAccessException, InstantiationException {
         String columnName = rs.getString("COLUMN_NAME");
         String remarks = rs.getString("REMARKS");
         boolean isPrimaryKey = isColumnPrimaryKey(metaData, assetStructureTable.getTableName(), columnName);
@@ -570,7 +575,8 @@ public class DataSourceServiceImpl extends ServiceImpl<DataSourceMapper, DataSou
      * @return
      * @throws SQLException
      */
-    private boolean isColumnPrimaryKey(DatabaseMetaData metaData, String tableName, String columnName) throws SQLException {
+    @Async
+    public boolean isColumnPrimaryKey(DatabaseMetaData metaData, String tableName, String columnName) throws SQLException {
         try (ResultSet rs = metaData.getPrimaryKeys(null, null, tableName)) {
             while (rs.next()) {
                 String primaryKeyColumnName = rs.getString("COLUMN_NAME");
@@ -594,7 +600,8 @@ public class DataSourceServiceImpl extends ServiceImpl<DataSourceMapper, DataSou
      * @throws IllegalAccessException
      * @throws InstantiationException
      */
-    private String getMappingType(Connection conn, String dataDetailsSQL, String columnName) throws SQLException, ClassNotFoundException, IllegalAccessException, InstantiationException {
+    @Async
+    public String getMappingType(Connection conn, String dataDetailsSQL, String columnName) throws SQLException, ClassNotFoundException, IllegalAccessException, InstantiationException {
         try (
                 PreparedStatement st = conn.prepareStatement(dataDetailsSQL);
                 ResultSet rs = st.executeQuery()) {