Merge pull request #12676 from dataease/pr@dev-v2@refactor_ds

refactor(数据源): 优化数据源连接池
This commit is contained in:
Junjun 2024-10-14 10:40:11 +08:00 committed by GitHub
commit 3760e58166
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1345,13 +1345,18 @@ public class CalciteProvider extends Provider {
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();
private Connection getConnectionFromPool(Long dsId) {
try {
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();
} catch (Exception e) {
DEException.throwException("连接无效");
}
return null;
}
public void exec(EngineRequest engineRequest) throws Exception {