Skip to content

Commit 986e60b

Browse files
authored
[Feature][Connector] Support parse sql scripts to lineage (#578)
1 parent 9620680 commit 986e60b

File tree

14 files changed

+81
-75
lines changed

14 files changed

+81
-75
lines changed

datavines-connector/datavines-connector-plugins/datavines-connector-clickhouse/src/main/java/io/datavines/connector/plugin/ClickHouseConnectorFactory.java

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
*/
1717
package io.datavines.connector.plugin;
1818

19+
import com.alibaba.druid.DbType;
1920
import io.datavines.connector.api.*;
2021

2122
public class ClickHouseConnectorFactory extends AbstractJdbcConnectorFactory {
@@ -55,6 +56,11 @@ public TypeConverter getTypeConverter() {
5556
return new ClickHouseTypeConverter();
5657
}
5758

59+
@Override
60+
public StatementParser getStatementParser() {
61+
return new DefaultStatementParser(DbType.clickhouse);
62+
}
63+
5864
@Override
5965
public String getPluginName() {
6066
return "clickhouse";

datavines-connector/datavines-connector-plugins/datavines-connector-databend/src/main/java/io/datavines/connector/plugin/DatabendConnectorFactory.java

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
*/
1717
package io.datavines.connector.plugin;
1818

19+
import com.alibaba.druid.DbType;
1920
import io.datavines.connector.api.*;
2021

2122
public class DatabendConnectorFactory extends AbstractJdbcConnectorFactory {
@@ -45,6 +46,11 @@ public ConfigBuilder getConfigBuilder() {
4546
return new DatabendConfigBuilder();
4647
}
4748

49+
@Override
50+
public StatementParser getStatementParser() {
51+
return new DefaultStatementParser(DbType.mysql);
52+
}
53+
4854
@Override
4955
public String getPluginName() {
5056
return "databend";

datavines-connector/datavines-connector-plugins/datavines-connector-dm/src/main/java/io/datavines/connector/plugin/DmConnectorFactory.java

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
*/
1717
package io.datavines.connector.plugin;
1818

19+
import com.alibaba.druid.DbType;
1920
import io.datavines.connector.api.*;
2021

2122
public class DmConnectorFactory extends AbstractJdbcConnectorFactory {
@@ -45,6 +46,11 @@ public ConfigBuilder getConfigBuilder() {
4546
return new DmConfigBuilder();
4647
}
4748

49+
@Override
50+
public StatementParser getStatementParser() {
51+
return new DefaultStatementParser(DbType.dm);
52+
}
53+
4854
@Override
4955
public String getPluginName() {
5056
return "dm";

datavines-connector/datavines-connector-plugins/datavines-connector-doris/src/main/java/io/datavines/connector/plugin/DorisConnectorFactory.java

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
*/
1717
package io.datavines.connector.plugin;
1818

19+
import com.alibaba.druid.DbType;
1920
import io.datavines.connector.api.*;
2021

2122
public class DorisConnectorFactory extends AbstractJdbcConnectorFactory {
@@ -45,6 +46,11 @@ public ConfigBuilder getConfigBuilder() {
4546
return new DorisConfigBuilder();
4647
}
4748

49+
@Override
50+
public StatementParser getStatementParser() {
51+
return new DefaultStatementParser(DbType.mysql);
52+
}
53+
4854
@Override
4955
public String getPluginName() {
5056
return "doris";

datavines-connector/datavines-connector-plugins/datavines-connector-hive/src/main/java/io/datavines/connector/plugin/HiveConnectorFactory.java

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
*/
1717
package io.datavines.connector.plugin;
1818

19+
import com.alibaba.druid.DbType;
1920
import io.datavines.connector.api.*;
2021

2122
public class HiveConnectorFactory extends AbstractJdbcConnectorFactory {
@@ -55,6 +56,11 @@ public DataSourceClient getDataSourceClient() {
5556
return new HiveDataSourceClient();
5657
}
5758

59+
@Override
60+
public StatementParser getStatementParser() {
61+
return new DefaultStatementParser(DbType.hive);
62+
}
63+
5864
@Override
5965
public String getPluginName() {
6066
return "hive";

datavines-connector/datavines-connector-plugins/datavines-connector-impala/src/main/java/io/datavines/connector/plugin/ImpalaConnectorFactory.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,10 +16,12 @@
1616
*/
1717
package io.datavines.connector.plugin;
1818

19+
import com.alibaba.druid.DbType;
1920
import io.datavines.connector.api.Connector;
2021
import io.datavines.connector.api.ParameterConverter;
2122
import io.datavines.connector.api.Dialect;
2223
import io.datavines.connector.api.Executor;
24+
import io.datavines.connector.api.StatementParser;
2325

2426
public class ImpalaConnectorFactory extends AbstractJdbcConnectorFactory {
2527

@@ -43,6 +45,11 @@ public Executor getExecutor() {
4345
return new ImpalaExecutor(getDataSourceClient());
4446
}
4547

48+
@Override
49+
public StatementParser getStatementParser() {
50+
return new DefaultStatementParser(DbType.hive);
51+
}
52+
4653
@Override
4754
public String getPluginName() {
4855
return "impala";

datavines-connector/datavines-connector-plugins/datavines-connector-mysql/src/main/java/io/datavines/connector/plugin/MysqlConnectorFactory.java

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
*/
1717
package io.datavines.connector.plugin;
1818

19+
import com.alibaba.druid.DbType;
1920
import io.datavines.connector.api.*;
2021

2122
public class MysqlConnectorFactory extends AbstractJdbcConnectorFactory {
@@ -50,6 +51,11 @@ public MetricScript getMetricScript() {
5051
return new MysqlMetricScript();
5152
}
5253

54+
@Override
55+
public StatementParser getStatementParser() {
56+
return new DefaultStatementParser(DbType.mysql);
57+
}
58+
5359
@Override
5460
public String getPluginName() {
5561
return "mysql";

datavines-connector/datavines-connector-plugins/datavines-connector-oracle/src/main/java/io/datavines/connector/plugin/OracleConnectorFactory.java

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
*/
1717
package io.datavines.connector.plugin;
1818

19+
import com.alibaba.druid.DbType;
1920
import io.datavines.connector.api.*;
2021

2122
public class OracleConnectorFactory extends AbstractJdbcConnectorFactory{
@@ -55,6 +56,11 @@ public MetricScript getMetricScript() {
5556
return new OracleMetricScript();
5657
}
5758

59+
@Override
60+
public StatementParser getStatementParser() {
61+
return new DefaultStatementParser(DbType.oracle);
62+
}
63+
5864
@Override
5965
public String getPluginName() {
6066
return "oracle";

datavines-connector/datavines-connector-plugins/datavines-connector-postgresql/src/main/java/io/datavines/connector/plugin/PostgreSqlConnectorFactory.java

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
*/
1717
package io.datavines.connector.plugin;
1818

19+
import com.alibaba.druid.DbType;
1920
import io.datavines.connector.api.*;
2021

2122
public class PostgreSqlConnectorFactory extends AbstractJdbcConnectorFactory {
@@ -55,6 +56,11 @@ public MetricScript getMetricScript() {
5556
return new PostgreSqlMetricScript();
5657
}
5758

59+
@Override
60+
public StatementParser getStatementParser() {
61+
return new DefaultStatementParser(DbType.postgresql);
62+
}
63+
5864
@Override
5965
public String getPluginName() {
6066
return "postgresql";

datavines-connector/datavines-connector-plugins/datavines-connector-presto/src/main/java/io/datavines/connector/plugin/PrestoConnectorFactory.java

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
*/
1717
package io.datavines.connector.plugin;
1818

19+
import com.alibaba.druid.DbType;
1920
import io.datavines.connector.api.*;
2021

2122
public class PrestoConnectorFactory extends AbstractJdbcConnectorFactory {
@@ -50,6 +51,11 @@ public DataSourceClient getDataSourceClient() {
5051
return new PrestoDataSourceClient();
5152
}
5253

54+
@Override
55+
public StatementParser getStatementParser() {
56+
return new DefaultStatementParser(DbType.presto);
57+
}
58+
5359
@Override
5460
public String getPluginName() {
5561
return "presto";

0 commit comments

Comments
 (0)