diff --git a/dolphinscheduler-task-plugin/dolphinscheduler-task-datax/src/main/java/org/apache/dolphinscheduler/plugin/task/datax/DataxUtils.java b/dolphinscheduler-task-plugin/dolphinscheduler-task-datax/src/main/java/org/apache/dolphinscheduler/plugin/task/datax/DataxUtils.java index c377f88f33..d2c4b1d67e 100644 --- a/dolphinscheduler-task-plugin/dolphinscheduler-task-datax/src/main/java/org/apache/dolphinscheduler/plugin/task/datax/DataxUtils.java +++ b/dolphinscheduler-task-plugin/dolphinscheduler-task-datax/src/main/java/org/apache/dolphinscheduler/plugin/task/datax/DataxUtils.java @@ -17,6 +17,7 @@ package org.apache.dolphinscheduler.plugin.task.datax; +import com.alibaba.druid.sql.dialect.presto.parser.PrestoStatementParser; import org.apache.dolphinscheduler.spi.enums.DbType; import com.alibaba.druid.sql.dialect.clickhouse.parser.ClickhouseStatementParser; @@ -39,7 +40,7 @@ public class DataxUtils { public static final String DATAX_READER_PLUGIN_CLICKHOUSE = "clickhousereader"; - public static final String DATAX_READER_PLUGIN_HIVE = "rdbmsreader"; + public static final String DATAX_READER_PLUGIN_RDBMS = "rdbmsreader"; public static final String DATAX_WRITER_PLUGIN_MYSQL = "mysqlwriter"; @@ -51,7 +52,7 @@ public class DataxUtils { public static final String DATAX_WRITER_PLUGIN_CLICKHOUSE = "clickhousewriter"; - public static final String DATAX_WRITER_PLUGIN_HIVE = "rdbmswriter"; + public static final String DATAX_WRITER_PLUGIN_RDBMS = "rdbmswriter"; public static String getReaderPluginName(DbType dbType) { switch (dbType) { @@ -66,7 +67,9 @@ public class DataxUtils { case CLICKHOUSE: return DATAX_READER_PLUGIN_CLICKHOUSE; case HIVE: - return DATAX_READER_PLUGIN_HIVE; + return DATAX_READER_PLUGIN_RDBMS; + case PRESTO: + return DATAX_READER_PLUGIN_RDBMS; default: return null; } @@ -85,7 +88,9 @@ public class DataxUtils { case CLICKHOUSE: return DATAX_WRITER_PLUGIN_CLICKHOUSE; case HIVE: - return DATAX_WRITER_PLUGIN_HIVE; + return DATAX_WRITER_PLUGIN_RDBMS; + case PRESTO: + return DATAX_WRITER_PLUGIN_RDBMS; default: return null; } @@ -105,6 +110,8 @@ public class DataxUtils { return new ClickhouseStatementParser(sql); case HIVE: return new HiveStatementParser(sql); + case PRESTO: + return new PrestoStatementParser(sql); default: return null; } diff --git a/dolphinscheduler-ui/src/views/projects/task/components/node/fields/use-datax.ts b/dolphinscheduler-ui/src/views/projects/task/components/node/fields/use-datax.ts index 718e49f952..72952bb807 100644 --- a/dolphinscheduler-ui/src/views/projects/task/components/node/fields/use-datax.ts +++ b/dolphinscheduler-ui/src/views/projects/task/components/node/fields/use-datax.ts @@ -132,7 +132,8 @@ export function useDataX(model: { [field: string]: any }): IJsonItem[] { 'ORACLE', 'SQLSERVER', 'CLICKHOUSE', - 'HIVE' + 'HIVE', + 'PRESTO' ] onMounted(() => { initConstants()