[Improvement-15919][datasource] Improvement datasource get name (#15920)
This commit is contained in:
parent
b29965bdce
commit
7a55adeae9
|
|
@ -118,7 +118,7 @@ public abstract class AbstractDataSourceProcessor implements DataSourceProcessor
|
||||||
@Override
|
@Override
|
||||||
public String getDatasourceUniqueId(ConnectionParam connectionParam, DbType dbType) {
|
public String getDatasourceUniqueId(ConnectionParam connectionParam, DbType dbType) {
|
||||||
BaseConnectionParam baseConnectionParam = (BaseConnectionParam) connectionParam;
|
BaseConnectionParam baseConnectionParam = (BaseConnectionParam) connectionParam;
|
||||||
return MessageFormat.format("{0}@{1}@{2}@{3}", dbType.getDescp(), baseConnectionParam.getUser(),
|
return MessageFormat.format("{0}@{1}@{2}@{3}", dbType.getName(), baseConnectionParam.getUser(),
|
||||||
PasswordUtils.encodePassword(baseConnectionParam.getPassword()), baseConnectionParam.getJdbcUrl());
|
PasswordUtils.encodePassword(baseConnectionParam.getPassword()), baseConnectionParam.getJdbcUrl());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -69,9 +69,9 @@ public class DataSourceClientProvider {
|
||||||
String datasourceUniqueId = DataSourceUtils.getDatasourceUniqueId(baseConnectionParam, dbType);
|
String datasourceUniqueId = DataSourceUtils.getDatasourceUniqueId(baseConnectionParam, dbType);
|
||||||
return POOLED_DATASOURCE_CLIENT_CACHE.get(datasourceUniqueId, () -> {
|
return POOLED_DATASOURCE_CLIENT_CACHE.get(datasourceUniqueId, () -> {
|
||||||
Map<String, DataSourceChannel> dataSourceChannelMap = dataSourcePluginManager.getDataSourceChannelMap();
|
Map<String, DataSourceChannel> dataSourceChannelMap = dataSourcePluginManager.getDataSourceChannelMap();
|
||||||
DataSourceChannel dataSourceChannel = dataSourceChannelMap.get(dbType.getDescp());
|
DataSourceChannel dataSourceChannel = dataSourceChannelMap.get(dbType.getName());
|
||||||
if (null == dataSourceChannel) {
|
if (null == dataSourceChannel) {
|
||||||
throw new RuntimeException(String.format("datasource plugin '%s' is not found", dbType.getDescp()));
|
throw new RuntimeException(String.format("datasource plugin '%s' is not found", dbType.getName()));
|
||||||
}
|
}
|
||||||
return dataSourceChannel.createPooledDataSourceClient(baseConnectionParam, dbType);
|
return dataSourceChannel.createPooledDataSourceClient(baseConnectionParam, dbType);
|
||||||
});
|
});
|
||||||
|
|
@ -85,9 +85,9 @@ public class DataSourceClientProvider {
|
||||||
public static AdHocDataSourceClient getAdHocDataSourceClient(DbType dbType, ConnectionParam connectionParam) {
|
public static AdHocDataSourceClient getAdHocDataSourceClient(DbType dbType, ConnectionParam connectionParam) {
|
||||||
BaseConnectionParam baseConnectionParam = (BaseConnectionParam) connectionParam;
|
BaseConnectionParam baseConnectionParam = (BaseConnectionParam) connectionParam;
|
||||||
Map<String, DataSourceChannel> dataSourceChannelMap = dataSourcePluginManager.getDataSourceChannelMap();
|
Map<String, DataSourceChannel> dataSourceChannelMap = dataSourcePluginManager.getDataSourceChannelMap();
|
||||||
DataSourceChannel dataSourceChannel = dataSourceChannelMap.get(dbType.getDescp());
|
DataSourceChannel dataSourceChannel = dataSourceChannelMap.get(dbType.getName());
|
||||||
if (null == dataSourceChannel) {
|
if (null == dataSourceChannel) {
|
||||||
throw new RuntimeException(String.format("datasource plugin '%s' is not found", dbType.getDescp()));
|
throw new RuntimeException(String.format("datasource plugin '%s' is not found", dbType.getName()));
|
||||||
}
|
}
|
||||||
return dataSourceChannel.createAdHocDataSourceClient(baseConnectionParam, dbType);
|
return dataSourceChannel.createAdHocDataSourceClient(baseConnectionParam, dbType);
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.athena;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -32,6 +33,6 @@ public class AthenaDataSourceChannelFactory implements DataSourceChannelFactory
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "athena";
|
return DbType.ATHENA.getName();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.azuresql;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -27,7 +28,7 @@ public class AzureSQLDataSourceChannelFactory implements DataSourceChannelFactor
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "azuresql";
|
return DbType.AZURESQL.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.clickhouse;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -27,7 +28,7 @@ public class ClickHouseDataSourceChannelFactory implements DataSourceChannelFact
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "clickhouse";
|
return DbType.CLICKHOUSE.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
||||||
|
|
@ -28,7 +28,7 @@ public class DamengDataSourceChannelFactory implements DataSourceChannelFactory
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return DbType.DAMENG.getDescp();
|
return DbType.DAMENG.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.databend;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -27,7 +28,7 @@ public class DatabendDataSourceChannelFactory implements DataSourceChannelFactor
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "databend";
|
return DbType.DATABEND.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
||||||
|
|
@ -151,7 +151,7 @@ public class DatabendDataSourceProcessorTest {
|
||||||
@Test
|
@Test
|
||||||
public void testDbType() {
|
public void testDbType() {
|
||||||
Assertions.assertEquals(19, DbType.DATABEND.getCode());
|
Assertions.assertEquals(19, DbType.DATABEND.getCode());
|
||||||
Assertions.assertEquals("databend", DbType.DATABEND.getDescp());
|
Assertions.assertEquals("databend", DbType.DATABEND.getName());
|
||||||
Assertions.assertEquals(DbType.DATABEND, DbType.of(19));
|
Assertions.assertEquals(DbType.DATABEND, DbType.of(19));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.db2;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -27,7 +28,7 @@ public class DB2DataSourceChannelFactory implements DataSourceChannelFactory {
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "db2";
|
return DbType.DB2.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
||||||
|
|
@ -32,6 +32,6 @@ public class DorisDataSourceChannelFactory implements DataSourceChannelFactory {
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return DbType.DORIS.getDescp();
|
return DbType.DORIS.getName();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.hana;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -27,7 +28,7 @@ public class HanaDataSourceChannelFactory implements DataSourceChannelFactory {
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "hana";
|
return DbType.HANA.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.hive;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -27,7 +28,7 @@ public class HiveDataSourceChannelFactory implements DataSourceChannelFactory {
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "hive";
|
return DbType.HIVE.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.k8s;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -32,7 +33,7 @@ public class K8sDataSourceChannelFactory implements DataSourceChannelFactory {
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "k8s";
|
return DbType.K8S.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -58,7 +58,7 @@ public class K8sDataSourceProcessor extends AbstractDataSourceProcessor {
|
||||||
@Override
|
@Override
|
||||||
public String getDatasourceUniqueId(ConnectionParam connectionParam, DbType dbType) {
|
public String getDatasourceUniqueId(ConnectionParam connectionParam, DbType dbType) {
|
||||||
K8sConnectionParam baseConnectionParam = (K8sConnectionParam) connectionParam;
|
K8sConnectionParam baseConnectionParam = (K8sConnectionParam) connectionParam;
|
||||||
return MessageFormat.format("{0}@{1}@{2}", dbType.getDescp(),
|
return MessageFormat.format("{0}@{1}@{2}", dbType.getName(),
|
||||||
PasswordUtils.encodePassword(baseConnectionParam.getKubeConfig()), baseConnectionParam.getNamespace());
|
PasswordUtils.encodePassword(baseConnectionParam.getKubeConfig()), baseConnectionParam.getNamespace());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.kyuubi;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -27,7 +28,7 @@ public class KyuubiDataSourceChannelFactory implements DataSourceChannelFactory
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "kyuubi";
|
return DbType.KYUUBI.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
||||||
|
|
@ -143,7 +143,7 @@ public class KyuubiDataSourceProcessorTest {
|
||||||
@Test
|
@Test
|
||||||
public void testDbType() {
|
public void testDbType() {
|
||||||
Assertions.assertEquals(18, DbType.KYUUBI.getCode());
|
Assertions.assertEquals(18, DbType.KYUUBI.getCode());
|
||||||
Assertions.assertEquals("kyuubi", DbType.KYUUBI.getDescp());
|
Assertions.assertEquals("kyuubi", DbType.KYUUBI.getName());
|
||||||
Assertions.assertEquals(DbType.KYUUBI, DbType.of(18));
|
Assertions.assertEquals(DbType.KYUUBI, DbType.of(18));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.mysql;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -27,7 +28,7 @@ public class MySQLDataSourceChannelFactory implements DataSourceChannelFactory {
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "mysql";
|
return DbType.MYSQL.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.oceanbase;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -27,7 +28,7 @@ public class OceanBaseDataSourceChannelFactory implements DataSourceChannelFacto
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "oceanbase";
|
return DbType.OCEANBASE.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.oracle;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -27,7 +28,7 @@ public class OracleDataSourceChannelFactory implements DataSourceChannelFactory
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "oracle";
|
return DbType.ORACLE.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.postgresql;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -27,7 +28,7 @@ public class PostgreSQLDataSourceChannelFactory implements DataSourceChannelFact
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "postgresql";
|
return DbType.POSTGRESQL.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.presto;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -27,7 +28,7 @@ public class PrestoDataSourceChannelFactory implements DataSourceChannelFactory
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "presto";
|
return DbType.PRESTO.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.redshift;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -32,6 +33,6 @@ public class RedshiftDataSourceChannelFactory implements DataSourceChannelFactor
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "redshift";
|
return DbType.REDSHIFT.getName();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.sagemaker;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -32,7 +33,7 @@ public class SagemakerDataSourceChannelFactory implements DataSourceChannelFacto
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "sagemaker";
|
return DbType.SAGEMAKER.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -57,7 +57,7 @@ public class SagemakerDataSourceProcessor extends AbstractDataSourceProcessor {
|
||||||
@Override
|
@Override
|
||||||
public String getDatasourceUniqueId(ConnectionParam connectionParam, DbType dbType) {
|
public String getDatasourceUniqueId(ConnectionParam connectionParam, DbType dbType) {
|
||||||
SagemakerConnectionParam baseConnectionParam = (SagemakerConnectionParam) connectionParam;
|
SagemakerConnectionParam baseConnectionParam = (SagemakerConnectionParam) connectionParam;
|
||||||
return MessageFormat.format("{0}@{1}@{2}@{3}", dbType.getDescp(),
|
return MessageFormat.format("{0}@{1}@{2}@{3}", dbType.getName(),
|
||||||
PasswordUtils.encodePassword(baseConnectionParam.getUserName()),
|
PasswordUtils.encodePassword(baseConnectionParam.getUserName()),
|
||||||
PasswordUtils.encodePassword(baseConnectionParam.getPassword()),
|
PasswordUtils.encodePassword(baseConnectionParam.getPassword()),
|
||||||
PasswordUtils.encodePassword(baseConnectionParam.getAwsRegion()));
|
PasswordUtils.encodePassword(baseConnectionParam.getAwsRegion()));
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.snowflake;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -27,7 +28,7 @@ public class SnowflakeDataSourceChannelFactory implements DataSourceChannelFacto
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "snowflake";
|
return DbType.SNOWFLAKE.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
||||||
|
|
@ -169,7 +169,7 @@ public class SnowflakeDataSourceProcessorTest {
|
||||||
@Test
|
@Test
|
||||||
public void testDbType() {
|
public void testDbType() {
|
||||||
Assertions.assertEquals(20, DbType.SNOWFLAKE.getCode());
|
Assertions.assertEquals(20, DbType.SNOWFLAKE.getCode());
|
||||||
Assertions.assertEquals("snowflake", DbType.SNOWFLAKE.getDescp());
|
Assertions.assertEquals("snowflake", DbType.SNOWFLAKE.getName());
|
||||||
Assertions.assertEquals(DbType.of(20), DbType.SNOWFLAKE);
|
Assertions.assertEquals(DbType.of(20), DbType.SNOWFLAKE);
|
||||||
Assertions.assertEquals(DbType.ofName("SNOWFLAKE"), DbType.SNOWFLAKE);
|
Assertions.assertEquals(DbType.ofName("SNOWFLAKE"), DbType.SNOWFLAKE);
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.spark;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -27,7 +28,7 @@ public class SparkDataSourceChannelFactory implements DataSourceChannelFactory {
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "spark";
|
return DbType.SPARK.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.sqlserver;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -27,7 +28,7 @@ public class SQLServerDataSourceChannelFactory implements DataSourceChannelFacto
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "sqlserver";
|
return DbType.SQLSERVER.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.ssh;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -27,7 +28,7 @@ public class SSHDataSourceChannelFactory implements DataSourceChannelFactory {
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "ssh";
|
return DbType.SSH.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
||||||
|
|
@ -55,7 +55,7 @@ public class SSHDataSourceProcessor extends AbstractDataSourceProcessor {
|
||||||
@Override
|
@Override
|
||||||
public String getDatasourceUniqueId(ConnectionParam connectionParam, DbType dbType) {
|
public String getDatasourceUniqueId(ConnectionParam connectionParam, DbType dbType) {
|
||||||
SSHConnectionParam baseConnectionParam = (SSHConnectionParam) connectionParam;
|
SSHConnectionParam baseConnectionParam = (SSHConnectionParam) connectionParam;
|
||||||
return MessageFormat.format("{0}@{1}@{2}@{3}", dbType.getDescp(), baseConnectionParam.getHost(),
|
return MessageFormat.format("{0}@{1}@{2}@{3}", dbType.getName(), baseConnectionParam.getHost(),
|
||||||
baseConnectionParam.getUser(),
|
baseConnectionParam.getUser(),
|
||||||
PasswordUtils.encodePassword(baseConnectionParam.getPassword()));
|
PasswordUtils.encodePassword(baseConnectionParam.getPassword()));
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -33,6 +33,6 @@ public class StarRocksDataSourceChannelFactory implements DataSourceChannelFacto
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return DbType.STARROCKS.getDescp();
|
return DbType.STARROCKS.getName();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.trino;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -27,7 +28,7 @@ public class TrinoDataSourceChannelFactory implements DataSourceChannelFactory {
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "trino";
|
return DbType.TRINO.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.vertica;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -27,7 +28,7 @@ public class VerticaDataSourceChannelFactory implements DataSourceChannelFactory
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "vertica";
|
return DbType.VERTICA.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
||||||
|
|
@ -19,6 +19,7 @@ package org.apache.dolphinscheduler.plugin.datasource.zeppelin;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel;
|
||||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory;
|
||||||
|
import org.apache.dolphinscheduler.spi.enums.DbType;
|
||||||
|
|
||||||
import com.google.auto.service.AutoService;
|
import com.google.auto.service.AutoService;
|
||||||
|
|
||||||
|
|
@ -32,7 +33,7 @@ public class ZeppelinDataSourceChannelFactory implements DataSourceChannelFactor
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getName() {
|
public String getName() {
|
||||||
return "zeppelin";
|
return DbType.ZEPPELIN.getName();
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -56,7 +56,7 @@ public class ZeppelinDataSourceProcessor extends AbstractDataSourceProcessor {
|
||||||
@Override
|
@Override
|
||||||
public String getDatasourceUniqueId(ConnectionParam connectionParam, DbType dbType) {
|
public String getDatasourceUniqueId(ConnectionParam connectionParam, DbType dbType) {
|
||||||
ZeppelinConnectionParam baseConnectionParam = (ZeppelinConnectionParam) connectionParam;
|
ZeppelinConnectionParam baseConnectionParam = (ZeppelinConnectionParam) connectionParam;
|
||||||
return MessageFormat.format("{0}@{1}@{2}@{3}", dbType.getDescp(), baseConnectionParam.getRestEndpoint(),
|
return MessageFormat.format("{0}@{1}@{2}@{3}", dbType.getName(), baseConnectionParam.getRestEndpoint(),
|
||||||
baseConnectionParam.getUsername(), PasswordUtils.encodePassword(baseConnectionParam.getPassword()));
|
baseConnectionParam.getUsername(), PasswordUtils.encodePassword(baseConnectionParam.getPassword()));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -28,42 +28,44 @@ import com.google.common.base.Functions;
|
||||||
|
|
||||||
public enum DbType {
|
public enum DbType {
|
||||||
|
|
||||||
MYSQL(0, "mysql"),
|
MYSQL(0, "mysql", "mysql"),
|
||||||
POSTGRESQL(1, "postgresql"),
|
POSTGRESQL(1, "postgresql", "postgresql"),
|
||||||
HIVE(2, "hive"),
|
HIVE(2, "hive", "hive"),
|
||||||
SPARK(3, "spark"),
|
SPARK(3, "spark", "spark"),
|
||||||
CLICKHOUSE(4, "clickhouse"),
|
CLICKHOUSE(4, "clickhouse", "clickhouse"),
|
||||||
ORACLE(5, "oracle"),
|
ORACLE(5, "oracle", "oracle"),
|
||||||
SQLSERVER(6, "sqlserver"),
|
SQLSERVER(6, "sqlserver", "sqlserver"),
|
||||||
DB2(7, "db2"),
|
DB2(7, "db2", "db2"),
|
||||||
PRESTO(8, "presto"),
|
PRESTO(8, "presto", "presto"),
|
||||||
H2(9, "h2"),
|
H2(9, "h2", "h2"),
|
||||||
REDSHIFT(10, "redshift"),
|
REDSHIFT(10, "redshift", "redshift"),
|
||||||
ATHENA(11, "athena"),
|
ATHENA(11, "athena", "athena"),
|
||||||
TRINO(12, "trino"),
|
TRINO(12, "trino", "trino"),
|
||||||
STARROCKS(13, "starrocks"),
|
STARROCKS(13, "starrocks", "starrocks"),
|
||||||
AZURESQL(14, "azuresql"),
|
AZURESQL(14, "azuresql", "azuresql"),
|
||||||
DAMENG(15, "dameng"),
|
DAMENG(15, "dameng", "dameng"),
|
||||||
OCEANBASE(16, "oceanbase"),
|
OCEANBASE(16, "oceanbase", "oceanbase"),
|
||||||
SSH(17, "ssh"),
|
SSH(17, "ssh", "ssh"),
|
||||||
KYUUBI(18, "kyuubi"),
|
KYUUBI(18, "kyuubi", "kyuubi"),
|
||||||
DATABEND(19, "databend"),
|
DATABEND(19, "databend", "databend"),
|
||||||
SNOWFLAKE(20, "snowflake"),
|
SNOWFLAKE(20, "snowflake", "snowflake"),
|
||||||
VERTICA(21, "vertica"),
|
VERTICA(21, "vertica", "vertica"),
|
||||||
HANA(22, "hana"),
|
HANA(22, "hana", "hana"),
|
||||||
DORIS(23, "doris"),
|
DORIS(23, "doris", "doris"),
|
||||||
ZEPPELIN(24, "zeppelin"),
|
ZEPPELIN(24, "zeppelin", "zeppelin"),
|
||||||
SAGEMAKER(25, "sagemaker"),
|
SAGEMAKER(25, "sagemaker", "sagemaker"),
|
||||||
|
|
||||||
K8S(26, "k8s");
|
K8S(26, "k8s", "k8s");
|
||||||
private static final Map<Integer, DbType> DB_TYPE_MAP =
|
private static final Map<Integer, DbType> DB_TYPE_MAP =
|
||||||
Arrays.stream(DbType.values()).collect(toMap(DbType::getCode, Functions.identity()));
|
Arrays.stream(DbType.values()).collect(toMap(DbType::getCode, Functions.identity()));
|
||||||
@EnumValue
|
@EnumValue
|
||||||
private final int code;
|
private final int code;
|
||||||
|
private final String name;
|
||||||
private final String descp;
|
private final String descp;
|
||||||
|
|
||||||
DbType(int code, String descp) {
|
DbType(int code, String name, String descp) {
|
||||||
this.code = code;
|
this.code = code;
|
||||||
|
this.name = name;
|
||||||
this.descp = descp;
|
this.descp = descp;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -83,6 +85,10 @@ public enum DbType {
|
||||||
return code;
|
return code;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public String getName() {
|
||||||
|
return name;
|
||||||
|
}
|
||||||
|
|
||||||
public String getDescp() {
|
public String getDescp() {
|
||||||
return descp;
|
return descp;
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue