[Improvement-15919][datasource] Improvement datasource get name (#15920)

This commit is contained in:
小可耐 2024-04-28 11:02:44 +08:00 committed by GitHub
parent b29965bdce
commit 7a55adeae9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
36 changed files with 95 additions and 66 deletions

View File

@ -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());
} }

View File

@ -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);
} }

View File

@ -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();
} }
} }

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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));
} }

View File

@ -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

View File

@ -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();
} }
} }

View File

@ -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

View File

@ -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

View File

@ -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();
} }
} }

View File

@ -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());
} }

View File

@ -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

View File

@ -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));
} }

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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();
} }
} }

View File

@ -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();
} }
} }

View File

@ -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()));

View File

@ -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

View File

@ -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);
} }

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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()));
} }

View File

@ -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();
} }
} }

View File

@ -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

View File

@ -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

View File

@ -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();
} }
} }

View File

@ -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()));
} }

View File

@ -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;
} }