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