Skip to content

Commit 470bb97

Browse files
authored
[Bugfix][TDengine] Fix the issue of losing the driver due to multiple calls to the submit job REST API #6581 (#6596)
1 parent d1599f8 commit 470bb97

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

Diff for: seatunnel-connectors-v2/connector-tdengine/src/main/java/org/apache/seatunnel/connectors/seatunnel/tdengine/source/TDengineSource.java

+3
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@
5656
import static org.apache.seatunnel.connectors.seatunnel.tdengine.config.TDengineSourceConfig.ConfigNames.URL;
5757
import static org.apache.seatunnel.connectors.seatunnel.tdengine.config.TDengineSourceConfig.ConfigNames.USERNAME;
5858
import static org.apache.seatunnel.connectors.seatunnel.tdengine.config.TDengineSourceConfig.buildSourceConfig;
59+
import static org.apache.seatunnel.connectors.seatunnel.tdengine.utils.TDengineUtil.checkDriverExist;
5960

6061
/**
6162
* TDengine source each split corresponds one subtable
@@ -135,6 +136,8 @@ private StableMetadata getStableMetadata(TDengineSourceConfig config) throws SQL
135136
config.getUsername(),
136137
"&password=",
137138
config.getPassword());
139+
// check td driver whether exist and if not, try to register
140+
checkDriverExist(jdbcUrl);
138141
try (Connection conn = DriverManager.getConnection(jdbcUrl)) {
139142
try (Statement statement = conn.createStatement()) {
140143
ResultSet metaResultSet =

Diff for: seatunnel-connectors-v2/connector-tdengine/src/main/java/org/apache/seatunnel/connectors/seatunnel/tdengine/source/TDengineSourceReader.java

+4
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,8 @@
4444
import java.util.Properties;
4545
import java.util.Set;
4646

47+
import static org.apache.seatunnel.connectors.seatunnel.tdengine.utils.TDengineUtil.checkDriverExist;
48+
4749
@Slf4j
4850
public class TDengineSourceReader implements SourceReader<SeaTunnelRow, TDengineSourceSplit> {
4951

@@ -108,6 +110,8 @@ public void open() {
108110
// @bobo (tdengine)
109111
connProps.setProperty(TSDBDriver.PROPERTY_KEY_BATCH_LOAD, "false");
110112
try {
113+
// check td driver whether exist and if not, try to register
114+
checkDriverExist(jdbcUrl);
111115
conn = DriverManager.getConnection(jdbcUrl, connProps);
112116
} catch (SQLException e) {
113117
throw new TDengineConnectorException(

0 commit comments

Comments
 (0)