From c056a3fdf42f0579ec4a5dcfbe39d5684063dbd0 Mon Sep 17 00:00:00 2001 From: taojinlong Date: Sat, 12 Oct 2024 15:31:21 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=20=E6=95=B0=E6=8D=AE=E7=9B=B4=E8=BF=9E?= =?UTF-8?q?=E6=97=B6=E4=BD=BF=E7=94=A8=E8=BF=9E=E6=8E=A5=E6=B1=A0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../manage/DatasourceSyncManage.java | 19 +-- .../datasource/manage/EngineManage.java | 7 +- .../datasource/provider/CalciteProvider.java | 135 +++++++++++------- .../datasource/provider/EngineProvider.java | 5 +- .../datasource/provider/H2EngineProvider.java | 16 +-- .../provider/MysqlEngineProvider.java | 15 -- .../datasource/provider/Provider.java | 7 +- 7 files changed, 98 insertions(+), 106 deletions(-) diff --git a/core/core-backend/src/main/java/io/dataease/datasource/manage/DatasourceSyncManage.java b/core/core-backend/src/main/java/io/dataease/datasource/manage/DatasourceSyncManage.java index bb9b1ff249..3bc4e5c64d 100644 --- a/core/core-backend/src/main/java/io/dataease/datasource/manage/DatasourceSyncManage.java +++ b/core/core-backend/src/main/java/io/dataease/datasource/manage/DatasourceSyncManage.java @@ -8,10 +8,7 @@ import io.dataease.datasource.dao.auto.entity.CoreDatasourceTask; import io.dataease.datasource.dao.auto.entity.CoreDatasourceTaskLog; import io.dataease.datasource.dao.auto.entity.CoreDeEngine; import io.dataease.datasource.dao.auto.mapper.CoreDatasourceMapper; -import io.dataease.datasource.provider.ApiUtils; -import io.dataease.datasource.provider.EngineProvider; -import io.dataease.datasource.provider.ExcelUtils; -import io.dataease.datasource.provider.ProviderUtil; +import io.dataease.datasource.provider.*; import io.dataease.datasource.request.EngineRequest; import io.dataease.datasource.server.DatasourceServer; import io.dataease.datasource.server.DatasourceTaskServer; @@ -20,6 +17,8 @@ import io.dataease.extensions.datasource.dto.DatasetTableDTO; import io.dataease.extensions.datasource.dto.DatasourceDTO; import io.dataease.extensions.datasource.dto.DatasourceRequest; import io.dataease.extensions.datasource.dto.TableField; +import io.dataease.extensions.datasource.factory.ProviderFactory; +import io.dataease.extensions.datasource.provider.Provider; import io.dataease.job.schedule.ExtractDataJob; import io.dataease.job.schedule.ScheduleManager; import io.dataease.utils.BeanUtils; @@ -49,6 +48,8 @@ public class DatasourceSyncManage { private DatasourceTaskServer datasourceTaskServer; @Resource private ScheduleManager scheduleManager; + @Resource + private CalciteProvider calciteProvider; public void extractExcelData(CoreDatasource coreDatasource, String type) { if (coreDatasource == null) { @@ -248,7 +249,7 @@ public class DatasourceSyncManage { for (int page = 1; page <= totalPage; page++) { engineRequest.setQuery(engineProvider.insertSql(engineTableName, dataList, page, pageNumber)); - engineProvider.exec(engineRequest); + calciteProvider.exec(engineRequest); } } @@ -278,7 +279,7 @@ public class DatasourceSyncManage { } for (int page = 1; page <= totalPage; page++) { engineRequest.setQuery(engineProvider.insertSql(engineTableName, dataList, page, pageNumber)); - engineProvider.exec(engineRequest); + calciteProvider.exec(engineRequest); } } @@ -291,7 +292,7 @@ public class DatasourceSyncManage { for (int i = 0; i < replaceTableSql.length; i++) { if (StringUtils.isNotEmpty(replaceTableSql[i])) { engineRequest.setQuery(replaceTableSql[i]); - engineProvider.exec(engineRequest); + calciteProvider.exec(engineRequest); } } } @@ -302,7 +303,7 @@ public class DatasourceSyncManage { engineRequest.setEngine(engine); EngineProvider engineProvider = ProviderUtil.getEngineProvider(engine.getType()); engineRequest.setQuery(engineProvider.createTableSql(tableName, tableFields, engine)); - engineProvider.exec(engineRequest); + calciteProvider.exec(engineRequest); } public void dropEngineTable(String tableName) throws Exception { @@ -311,7 +312,7 @@ public class DatasourceSyncManage { engineRequest.setEngine(engine); EngineProvider engineProvider = ProviderUtil.getEngineProvider(engine.getType()); engineRequest.setQuery(engineProvider.dropTable(tableName)); - engineProvider.exec(engineRequest); + calciteProvider.exec(engineRequest); } public void addSchedule(CoreDatasourceTask datasourceTask) throws DEException { diff --git a/core/core-backend/src/main/java/io/dataease/datasource/manage/EngineManage.java b/core/core-backend/src/main/java/io/dataease/datasource/manage/EngineManage.java index 32fd5cd302..18949d1dce 100644 --- a/core/core-backend/src/main/java/io/dataease/datasource/manage/EngineManage.java +++ b/core/core-backend/src/main/java/io/dataease/datasource/manage/EngineManage.java @@ -5,13 +5,12 @@ import io.dataease.datasource.dao.auto.entity.CoreDatasource; import io.dataease.datasource.dao.auto.entity.CoreDeEngine; import io.dataease.datasource.dao.auto.mapper.CoreDatasourceMapper; import io.dataease.datasource.dao.auto.mapper.CoreDeEngineMapper; -import io.dataease.datasource.provider.EngineProvider; -import io.dataease.datasource.provider.ProviderUtil; import io.dataease.datasource.type.H2; import io.dataease.datasource.type.Mysql; import io.dataease.exception.DEException; import io.dataease.extensions.datasource.dto.DatasourceDTO; import io.dataease.extensions.datasource.dto.DatasourceRequest; +import io.dataease.extensions.datasource.factory.ProviderFactory; import io.dataease.result.ResultMessage; import io.dataease.utils.BeanUtils; import io.dataease.utils.JsonUtil; @@ -75,12 +74,12 @@ public class EngineManage { throw new Exception("未完整设置数据引擎"); } try { - EngineProvider provider = ProviderUtil.getEngineProvider(engine.getType()); + DatasourceRequest datasourceRequest = new DatasourceRequest(); DatasourceDTO datasource = new DatasourceDTO(); BeanUtils.copyBean(datasource, engine); datasourceRequest.setDatasource(datasource); - provider.checkStatus(datasourceRequest); + ProviderFactory.getProvider(engine.getType()).checkStatus(datasourceRequest); } catch (Exception e) { DEException.throwException("校验失败：" + e.getMessage()); } diff --git a/core/core-backend/src/main/java/io/dataease/datasource/provider/CalciteProvider.java b/core/core-backend/src/main/java/io/dataease/datasource/provider/CalciteProvider.java index ef4d3f9a49..02d993a42f 100644 --- a/core/core-backend/src/main/java/io/dataease/datasource/provider/CalciteProvider.java +++ b/core/core-backend/src/main/java/io/dataease/datasource/provider/CalciteProvider.java @@ -6,6 +6,7 @@ import io.dataease.datasource.dao.auto.entity.CoreDatasource; import io.dataease.datasource.dao.auto.entity.CoreDriver; import io.dataease.datasource.dao.auto.mapper.CoreDatasourceMapper; import io.dataease.datasource.manage.EngineManage; +import io.dataease.datasource.request.EngineRequest; import io.dataease.datasource.type.*; import io.dataease.engine.constant.SQLConstants; import io.dataease.exception.DEException; @@ -20,10 +21,13 @@ import jakarta.annotation.PostConstruct; import jakarta.annotation.Resource; import org.apache.calcite.adapter.jdbc.JdbcSchema; import org.apache.calcite.config.CalciteConnectionProperty; +import org.apache.calcite.config.Lex; import org.apache.calcite.config.NullCollation; import org.apache.calcite.jdbc.CalciteConnection; import org.apache.calcite.schema.Schema; import org.apache.calcite.schema.SchemaPlus; +import org.apache.calcite.sql.SqlNode; +import org.apache.calcite.sql.parser.SqlParser; import org.apache.commons.collections4.CollectionUtils; import org.apache.commons.dbcp2.BasicDataSource; import org.apache.commons.lang3.StringUtils; @@ -94,26 +98,6 @@ public class CalciteProvider extends Provider { return schemas; } - @Override - public List getTables(DatasourceRequest datasourceRequest) { - List tables = new ArrayList<>(); - - try (ConnectionObj con = getConnection(datasourceRequest.getDatasource()); Statement statement = getStatement(con.getConnection(), 30)) { - datasourceRequest.setDsVersion(con.getConnection().getMetaData().getDatabaseMajorVersion()); - List tablesSqls = getTablesSql(datasourceRequest); - for (String tablesSql : tablesSqls) { - ResultSet resultSet = statement.executeQuery(tablesSql); - while (resultSet.next()) { - tables.add(getTableDesc(datasourceRequest, resultSet)); - } - } - } catch (Exception e) { - DEException.throwException(e.getMessage()); - } - return tables; - } - - @Override public String checkStatus(DatasourceRequest datasourceRequest) throws Exception { DatasourceConfiguration.DatasourceType datasourceType = DatasourceConfiguration.DatasourceType.valueOf(datasourceRequest.getDatasource().getType()); @@ -146,6 +130,24 @@ public class CalciteProvider extends Provider { return "Success"; } + @Override + public List getTables(DatasourceRequest datasourceRequest) { + List tables = new ArrayList<>(); + try (Connection con = getConnectionFromPool(datasourceRequest.getDatasource().getId()); Statement statement = getStatement(con, 30)) { + datasourceRequest.setDsVersion(con.getMetaData().getDatabaseMajorVersion()); + List tablesSqls = getTablesSql(datasourceRequest); + for (String tablesSql : tablesSqls) { + ResultSet resultSet = statement.executeQuery(tablesSql); + while (resultSet.next()) { + tables.add(getTableDesc(datasourceRequest, resultSet)); + } + } + } catch (Exception e) { + DEException.throwException(e.getMessage()); + } + return tables; + } + @Override public Map fetchResultField(DatasourceRequest datasourceRequest) throws DEException { // 不跨数据源 @@ -197,6 +199,22 @@ public class CalciteProvider extends Provider { return map; } + @Override + public String transSqlDialect(String sql, Map dsMap) throws DEException { + DatasourceSchemaDTO value = dsMap.entrySet().iterator().next().getValue(); + try (Connection connection = getConnectionFromPool(value.getId());) { + // 获取数据库version + if (connection != null) { + value.setDsVersion(connection.getMetaData().getDatabaseMajorVersion()); + } + SqlParser parser = SqlParser.create(sql, SqlParser.Config.DEFAULT.withLex(Lex.JAVA)); + SqlNode sqlNode = parser.parseStmt(); + return sqlNode.toSqlString(getDialect(value)).toString(); + } catch (Exception e) { + DEException.throwException(e.getMessage()); + } + return null; + } private List fetchResultField(ResultSet rs) throws Exception { List fieldList = new ArrayList<>(); @@ -229,7 +247,7 @@ public class CalciteProvider extends Provider { String table = datasourceRequest.getTable(); if (StringUtils.isEmpty(table)) { ResultSet resultSet = null; - try (ConnectionObj con = getConnection(datasourceRequest.getDatasource()); Statement statement = getStatement(con.getConnection(), 30)) { + try (Connection con = getConnectionFromPool(datasourceRequest.getDatasource().getId()); Statement statement = getStatement(con, 30)) { if (DatasourceConfiguration.DatasourceType.valueOf(datasourceSchemaDTO.getType()) == DatasourceConfiguration.DatasourceType.oracle) { statement.executeUpdate("ALTER SESSION SET CURRENT_SCHEMA = " + datasourceConfiguration.getSchema()); } @@ -248,8 +266,8 @@ public class CalciteProvider extends Provider { } } else { ResultSet resultSet = null; - try (ConnectionObj con = getConnection(datasourceRequest.getDatasource()); Statement statement = getStatement(con.getConnection(), 30)) { - datasourceRequest.setDsVersion(con.getConnection().getMetaData().getDatabaseMajorVersion()); + try (Connection con = getConnectionFromPool(datasourceRequest.getDatasource().getId()); Statement statement = getStatement(con, 30)) { + datasourceRequest.setDsVersion(con.getMetaData().getDatabaseMajorVersion()); if (datasourceRequest.getDatasource().getType().equalsIgnoreCase("mongo") || isDorisCatalog(datasourceRequest)) { resultSet = statement.executeQuery("select * from " + table + " limit 0 offset 0 "); return fetchResultField(resultSet); @@ -394,7 +412,7 @@ public class CalciteProvider extends Provider { // schema ResultSet resultSet = null; - try (ConnectionObj con = getConnection(datasourceRequest.getDatasource()); Statement statement = getPreparedStatement(con.getConnection(), datasourceConfiguration.getQueryTimeout(), datasourceRequest.getQuery(), datasourceRequest.getTableFieldWithValues())) { + try (Connection con = getConnectionFromPool(datasourceRequest.getDatasource().getId()); Statement statement = getPreparedStatement(con, datasourceConfiguration.getQueryTimeout(), datasourceRequest.getQuery(), datasourceRequest.getTableFieldWithValues())) { if (DatasourceConfiguration.DatasourceType.valueOf(value.getType()) == DatasourceConfiguration.DatasourceType.oracle) { statement.executeUpdate("ALTER SESSION SET CURRENT_SCHEMA = " + datasourceConfiguration.getSchema()); } @@ -434,16 +452,13 @@ public class CalciteProvider extends Provider { public void exec(DatasourceRequest datasourceRequest) throws DEException { DatasourceSchemaDTO value = datasourceRequest.getDsList().entrySet().iterator().next().getValue(); datasourceRequest.setDatasource(value); - DatasourceConfiguration datasourceConfiguration = JsonUtil.parseObject(datasourceRequest.getDatasource().getConfiguration(), DatasourceConfiguration.class); - // schema ResultSet resultSet = null; - try (ConnectionObj con = getConnection(datasourceRequest.getDatasource()); Statement statement = getPreparedStatement(con.getConnection(), datasourceConfiguration.getQueryTimeout(), datasourceRequest.getQuery(), datasourceRequest.getTableFieldWithValues())) { + try (Connection con = getConnectionFromPool(datasourceRequest.getDatasource().getId()); Statement statement = getPreparedStatement(con, datasourceConfiguration.getQueryTimeout(), datasourceRequest.getQuery(), datasourceRequest.getTableFieldWithValues())) { if (DatasourceConfiguration.DatasourceType.valueOf(value.getType()) == DatasourceConfiguration.DatasourceType.oracle) { statement.executeUpdate("ALTER SESSION SET CURRENT_SCHEMA = " + datasourceConfiguration.getSchema()); } - if (CollectionUtils.isNotEmpty(datasourceRequest.getTableFieldWithValues())) { LogUtil.info("execWithPreparedStatement sql: " + datasourceRequest.getQuery()); for (int i = 0; i < datasourceRequest.getTableFieldWithValues().size(); i++) { @@ -474,12 +489,10 @@ public class CalciteProvider extends Provider { public int executeUpdate(DatasourceRequest datasourceRequest) throws DEException { DatasourceSchemaDTO value = datasourceRequest.getDsList().entrySet().iterator().next().getValue(); datasourceRequest.setDatasource(value); - DatasourceConfiguration datasourceConfiguration = JsonUtil.parseObject(datasourceRequest.getDatasource().getConfiguration(), DatasourceConfiguration.class); - // schema ResultSet resultSet = null; - try (ConnectionObj con = getConnection(datasourceRequest.getDatasource()); Statement statement = getPreparedStatement(con.getConnection(), datasourceConfiguration.getQueryTimeout(), datasourceRequest.getQuery(), datasourceRequest.getTableFieldWithValues())) { + try (Connection con = getConnectionFromPool(datasourceRequest.getDatasource().getId()); Statement statement = getPreparedStatement(con, datasourceConfiguration.getQueryTimeout(), datasourceRequest.getQuery(), datasourceRequest.getTableFieldWithValues())) { if (DatasourceConfiguration.DatasourceType.valueOf(value.getType()) == DatasourceConfiguration.DatasourceType.oracle) { statement.executeUpdate("ALTER SESSION SET CURRENT_SCHEMA = " + datasourceConfiguration.getSchema()); } @@ -697,17 +710,13 @@ public class CalciteProvider extends Provider { return tableField; } - public Connection initConnection(Map dsMap) { - Connection connection = getCalciteConnection(); + public Connection initConnection(Map dsMap) throws SQLException { + Connection connection = take(); CalciteConnection calciteConnection = null; - try { - calciteConnection = connection.unwrap(CalciteConnection.class); - } catch (Exception e) { - DEException.throwException(e); - } + calciteConnection = connection.unwrap(CalciteConnection.class); DatasourceRequest datasourceRequest = new DatasourceRequest(); datasourceRequest.setDsList(dsMap); - SchemaPlus rootSchema = buildSchema(datasourceRequest, calciteConnection); + buildSchema(datasourceRequest, calciteConnection); return connection; } @@ -897,10 +906,8 @@ public class CalciteProvider extends Provider { rootSchema.add(ds.getSchemaAlias(), schema); } } catch (Exception e) { - e.printStackTrace(); } } catch (Exception e) { - e.printStackTrace(); } }); } @@ -1245,11 +1252,8 @@ public class CalciteProvider extends Provider { return null; } - private Connection connection = null; - public static int capacity = 10; - public void initConnectionPool() { LogUtil.info("Begin to init datasource pool..."); QueryWrapper datasourceQueryWrapper = new QueryWrapper(); @@ -1258,25 +1262,22 @@ public class CalciteProvider extends Provider { if (engine != null) { coreDatasources.add(engine); } - Map dsMap = new HashMap<>(); + for (CoreDatasource coreDatasource : coreDatasources) { + Map dsMap = new HashMap<>(); DatasourceSchemaDTO datasourceSchemaDTO = new DatasourceSchemaDTO(); BeanUtils.copyBean(datasourceSchemaDTO, coreDatasource); datasourceSchemaDTO.setSchemaAlias(String.format(SQLConstants.SCHEMA, datasourceSchemaDTO.getId())); dsMap.put(datasourceSchemaDTO.getId(), datasourceSchemaDTO); - } - LogUtil.info("dsMap size..." + dsMap.keySet().size()); - try { commonThreadPool.addTask(() -> { try { connection = initConnection(dsMap); - } catch (Exception e) { + } catch (Exception ignore) { } }); - - } catch (Exception e) { - } + LogUtil.info("dsMap size..." + coreDatasources.size()); + } public void update(DatasourceDTO datasourceDTO) throws DEException { @@ -1333,13 +1334,37 @@ public class CalciteProvider extends Provider { Provider.getSessions().remove(datasource.getId()); } - public Connection take() { - if (connection == null) { - DEException.throwException("初始化连接池失败!"); + if (connection == null) { // 第一次检查，无需锁 + synchronized (Connection.class) { // 同步块 + if (connection == null) { // 第二次检查，需要锁 + connection = getCalciteConnection(); + } + } } return connection; } + private Connection getConnectionFromPool(Long dsId) throws SQLException { + Connection connection = take(); + CalciteConnection calciteConnection = connection.unwrap(CalciteConnection.class); + SchemaPlus rootSchema = calciteConnection.getRootSchema(); + JdbcSchema jdbcSchema = rootSchema.getSubSchema(String.format(SQLConstants.SCHEMA, dsId)).unwrap(JdbcSchema.class); + BasicDataSource basicDataSource = (BasicDataSource) jdbcSchema.getDataSource(); + return basicDataSource.getConnection(); + } + public void exec(EngineRequest engineRequest) throws Exception { + DatasourceConfiguration configuration = JsonUtil.parseObject(engineRequest.getEngine().getConfiguration(), DatasourceConfiguration.class); + int queryTimeout = configuration.getQueryTimeout(); + DatasourceDTO datasource = new DatasourceDTO(); + BeanUtils.copyBean(datasource, engineRequest.getEngine()); + try (Connection connection = getConnectionFromPool(datasource.getId()); Statement stat = getStatement(connection, queryTimeout)) { + PreparedStatement preparedStatement = connection.prepareStatement(engineRequest.getQuery()); + preparedStatement.setQueryTimeout(queryTimeout); + Boolean result = preparedStatement.execute(); + } catch (Exception e) { + throw e; + } + } } diff --git a/core/core-backend/src/main/java/io/dataease/datasource/provider/EngineProvider.java b/core/core-backend/src/main/java/io/dataease/datasource/provider/EngineProvider.java index cecadb0745..695c9decc7 100644 --- a/core/core-backend/src/main/java/io/dataease/datasource/provider/EngineProvider.java +++ b/core/core-backend/src/main/java/io/dataease/datasource/provider/EngineProvider.java @@ -10,7 +10,7 @@ import java.util.List; * @Author gin * @Date 2021/5/17 4:19 下午 */ -public abstract class EngineProvider extends CalciteProvider { +public abstract class EngineProvider { public abstract String createView(String name, String viewSQL); public abstract String dropTable(String name); @@ -23,6 +23,5 @@ public abstract class EngineProvider extends CalciteProvider { public abstract String insertSql(String name, List dataList, int page, int pageNumber); - public void exec(EngineRequest datasourceRequest) throws Exception { - } + } diff --git a/core/core-backend/src/main/java/io/dataease/datasource/provider/H2EngineProvider.java b/core/core-backend/src/main/java/io/dataease/datasource/provider/H2EngineProvider.java index 5cdd5dd85c..9a4d737cba 100644 --- a/core/core-backend/src/main/java/io/dataease/datasource/provider/H2EngineProvider.java +++ b/core/core-backend/src/main/java/io/dataease/datasource/provider/H2EngineProvider.java @@ -5,6 +5,7 @@ import io.dataease.dataset.utils.TableUtils; import io.dataease.datasource.dao.auto.entity.CoreDeEngine; import io.dataease.datasource.request.EngineRequest; import io.dataease.datasource.type.H2; +import io.dataease.datasource.type.Mysql; import io.dataease.extensions.datasource.dto.ConnectionObj; import io.dataease.extensions.datasource.dto.DatasourceDTO; import io.dataease.extensions.datasource.dto.TableField; @@ -22,21 +23,6 @@ import java.util.List; @Service("h2Engine") public class H2EngineProvider extends EngineProvider { - - public void exec(EngineRequest engineRequest) throws Exception { - DatasourceConfiguration configuration = JsonUtil.parseObject(engineRequest.getEngine().getConfiguration(), H2.class); - int queryTimeout = configuration.getQueryTimeout(); - DatasourceDTO datasource = new DatasourceDTO(); - BeanUtils.copyBean(datasource, engineRequest.getEngine()); - try (ConnectionObj connection = getConnection(datasource); Statement stat = getStatement(connection.getConnection(), queryTimeout)) { - PreparedStatement preparedStatement = connection.getConnection().prepareStatement(engineRequest.getQuery()); - preparedStatement.setQueryTimeout(queryTimeout); - Boolean result = preparedStatement.execute(); - } catch (Exception e) { - throw e; - } - } - private static final String creatTableSql = "CREATE TABLE IF NOT EXISTS `TABLE_NAME`" + "Column_Fields;"; diff --git a/core/core-backend/src/main/java/io/dataease/datasource/provider/MysqlEngineProvider.java b/core/core-backend/src/main/java/io/dataease/datasource/provider/MysqlEngineProvider.java index b42875e7a4..e3f454f5d2 100644 --- a/core/core-backend/src/main/java/io/dataease/datasource/provider/MysqlEngineProvider.java +++ b/core/core-backend/src/main/java/io/dataease/datasource/provider/MysqlEngineProvider.java @@ -26,21 +26,6 @@ import java.util.List; @Service("mysqlEngine") public class MysqlEngineProvider extends EngineProvider { - - public void exec(EngineRequest engineRequest) throws Exception { - DatasourceConfiguration configuration = JsonUtil.parseObject(engineRequest.getEngine().getConfiguration(), Mysql.class); - int queryTimeout = configuration.getQueryTimeout(); - DatasourceDTO datasource = new DatasourceDTO(); - BeanUtils.copyBean(datasource, engineRequest.getEngine()); - try (ConnectionObj connection = getConnection(datasource); Statement stat = getStatement(connection.getConnection(), queryTimeout)) { - PreparedStatement preparedStatement = connection.getConnection().prepareStatement(engineRequest.getQuery()); - preparedStatement.setQueryTimeout(queryTimeout); - Boolean result = preparedStatement.execute(); - } catch (Exception e) { - throw e; - } - } - private static final String creatTableSql = "CREATE TABLE IF NOT EXISTS `TABLE_NAME`" + "Column_Fields;"; diff --git a/sdk/extensions/extensions-datasource/src/main/java/io/dataease/extensions/datasource/provider/Provider.java b/sdk/extensions/extensions-datasource/src/main/java/io/dataease/extensions/datasource/provider/Provider.java index c8af998b90..0d4308f86b 100644 --- a/sdk/extensions/extensions-datasource/src/main/java/io/dataease/extensions/datasource/provider/Provider.java +++ b/sdk/extensions/extensions-datasource/src/main/java/io/dataease/extensions/datasource/provider/Provider.java @@ -133,15 +133,12 @@ public abstract class Provider { } public String transSqlDialect(String sql, Map dsMap) throws DEException { - try { - DatasourceSchemaDTO value = dsMap.entrySet().iterator().next().getValue(); - + DatasourceSchemaDTO value = dsMap.entrySet().iterator().next().getValue(); + try (ConnectionObj connection = getConnection(value)) { // 获取数据库version - ConnectionObj connection = getConnection(value); if (connection != null) { value.setDsVersion(connection.getConnection().getMetaData().getDatabaseMajorVersion()); } - SqlParser parser = SqlParser.create(sql, SqlParser.Config.DEFAULT.withLex(Lex.JAVA)); SqlNode sqlNode = parser.parseStmt(); return sqlNode.toSqlString(getDialect(value)).toString();