center/dataease-dm
center/dataease-dm
13
0
Fork 0
forked from github/dataease
Code Pull Requests Activity

refactor(数据源): 优化数据源连接池

Browse Source
This commit is contained in:
junjun 2024-10-14 10:38:32 +08:00
parent 3a9973c68c
commit 57427955db
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
core/core-backend/src/main/java/io/dataease/datasource/provider/CalciteProvider.java
View File

@ -1345,13 +1345,18 @@ public class CalciteProvider extends Provider {
return connection; return connection;
} }
private Connection getConnectionFromPool(Long dsId) throws SQLException { private Connection getConnectionFromPool(Long dsId) {
try {
Connection connection = take(); Connection connection = take();
CalciteConnection calciteConnection = connection.unwrap(CalciteConnection.class); CalciteConnection calciteConnection = connection.unwrap(CalciteConnection.class);
SchemaPlus rootSchema = calciteConnection.getRootSchema(); SchemaPlus rootSchema = calciteConnection.getRootSchema();
JdbcSchema jdbcSchema = rootSchema.getSubSchema(String.format(SQLConstants.SCHEMA, dsId)).unwrap(JdbcSchema.class); JdbcSchema jdbcSchema = rootSchema.getSubSchema(String.format(SQLConstants.SCHEMA, dsId)).unwrap(JdbcSchema.class);
BasicDataSource basicDataSource = (BasicDataSource) jdbcSchema.getDataSource(); BasicDataSource basicDataSource = (BasicDataSource) jdbcSchema.getDataSource();
return basicDataSource.getConnection(); return basicDataSource.getConnection();
} catch (Exception e) {
DEException.throwException("连接无效");
}
return null;
} }
public void exec(EngineRequest engineRequest) throws Exception { public void exec(EngineRequest engineRequest) throws Exception {