github/DataX
5
0
Fork 0
mirror of https://github.com/alibaba/DataX.git synced 2025-05-02 17:40:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #2074 from xieruiming/fix_conn

Browse Source 
postSql use DriverManager to get connection
This commit is contained in:
NJUEEXRM 2024-03-18 17:11:56 +08:00 committed by GitHub
commit 072b24f673
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 23 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
hologresjdbcwriter/src/main/java/com/alibaba/datax/plugin/writer/hologresjdbcwriter/BaseWriter.java
View File

@ -191,6 +191,7 @@ public class BaseWriter {
// 一般来说，是需要推迟到 task 中进行post 的执行（单表情况例外） // 一般来说，是需要推迟到 task 中进行post 的执行（单表情况例外）
public void post(Configuration originalConfig) { public void post(Configuration originalConfig) {
try {
String username = originalConfig.getString(Key.USERNAME); String username = originalConfig.getString(Key.USERNAME);
String password = originalConfig.getString(Key.PASSWORD); String password = originalConfig.getString(Key.PASSWORD);
@ -207,16 +208,17 @@ public class BaseWriter {
// 说明有 postSql 配置，则此处删除掉 // 说明有 postSql 配置，则此处删除掉
originalConfig.remove(Key.POST_SQL); originalConfig.remove(Key.POST_SQL);
String tempJdbcUrl = jdbcUrl.replace("postgresql", "hologres"); String tempJdbcUrl = jdbcUrl.replace("postgresql", "hologres");
Connection conn = DBUtil.getConnection(this.dataBaseType, try (Connection conn = DriverManager.getConnection(
tempJdbcUrl, username, password); tempJdbcUrl, username, password)) {
LOG.info( LOG.info(
"Begin to execute postSqls:[{}]. context info:{}.", "Begin to execute postSqls:[{}]. context info:{}.",
StringUtils.join(renderedPostSqls, ";"), tempJdbcUrl); StringUtils.join(renderedPostSqls, ";"), tempJdbcUrl);
WriterUtil.executeSqls(conn, renderedPostSqls, tempJdbcUrl, dataBaseType); WriterUtil.executeSqls(conn, renderedPostSqls, tempJdbcUrl, dataBaseType);
DBUtil.closeDBResources(null, null, conn);
} }
}
} catch (SQLException e) {
throw DataXException.asDataXException(DBUtilErrorCode.SQL_EXECUTE_FAIL, e);
}
} }
public void destroy(Configuration originalConfig) { public void destroy(Configuration originalConfig) {