ZhaoGuodong
2 years ago
committed by
GitHub
23 changed files with 619 additions and 9 deletions
@ -0,0 +1,22 @@
|
||||
# AWS Athena |
||||
|
||||
 |
||||
|
||||
## Datasource Parameters |
||||
|
||||
| **Datasource** | **Description** | |
||||
| --- | --- | |
||||
| Datasource | Select ATHENA. | |
||||
| Datasource name | Enter the name of the DataSource. | |
||||
| Description | Enter a description of the DataSource. | |
||||
| Username | Set the AWS access key. | |
||||
| Password | Set the AWS secret access key. | |
||||
| AwsRegion | Set the AWS region. | |
||||
| Database name | Enter the database name of the ATHENA connection. | |
||||
| Jdbc connection parameters | Parameter settings for ATHENA connection, in JSON format. | |
||||
|
||||
## Native Supported |
||||
|
||||
- No, read section example in [datasource-setting](../howto/datasource-setting.md) `DataSource Center` section to activate this datasource. |
||||
- JDBC driver configuration reference document [athena-connect-with-jdbc](https://docs.amazonaws.cn/athena/latest/ug/connect-with-jdbc.html) |
||||
- Driver download link [SimbaAthenaJDBC-2.0.31.1000/AthenaJDBC42.jar](https://s3.cn-north-1.amazonaws.com.cn/athena-downloads-cn/drivers/JDBC/SimbaAthenaJDBC-2.0.31.1000/AthenaJDBC42.jar) |
||||
@ -0,0 +1,19 @@
|
||||
# AWS Athena 数据源 |
||||
|
||||
 |
||||
|
||||
|
||||
- 数据源：选择 ATHENA |
||||
- 数据源名称：输入数据源的名称 |
||||
- 描述：输入数据源的描述 |
||||
- 用户名：设置连接 AWS 的 access key |
||||
- 密码：设置连接 AWS 的 secret access key |
||||
- AwsRegion: 设置连接 AWS 的区域 |
||||
- 数据库名：输入连接 ATHENA 的数据库名称 |
||||
- Jdbc 连接参数：用于 ATHENA 连接的参数设置，以 JSON 形式填写 |
||||
|
||||
## 是否原生支持 |
||||
|
||||
- 否，使用前需请参考 [数据源配置](../howto/datasource-setting.md) 中的 "数据源中心" 章节激活数据源。 |
||||
- JDBC驱动配置参考文档 [athena-connect-with-jdbc](https://docs.amazonaws.cn/athena/latest/ug/connect-with-jdbc.html) |
||||
- 驱动下载链接 [SimbaAthenaJDBC-2.0.31.1000/AthenaJDBC42.jar](https://s3.cn-north-1.amazonaws.com.cn/athena-downloads-cn/drivers/JDBC/SimbaAthenaJDBC-2.0.31.1000/AthenaJDBC42.jar) |
||||
|
After Width: | Height: | Size: 424 KiB |
@ -0,0 +1,44 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?> |
||||
<!-- |
||||
~ Licensed to the Apache Software Foundation (ASF) under one or more |
||||
~ contributor license agreements. See the NOTICE file distributed with |
||||
~ this work for additional information regarding copyright ownership. |
||||
~ The ASF licenses this file to You under the Apache License, Version 2.0 |
||||
~ (the "License"); you may not use this file except in compliance with |
||||
~ the License. You may obtain a copy of the License at |
||||
~ |
||||
~ http://www.apache.org/licenses/LICENSE-2.0 |
||||
~ |
||||
~ Unless required by applicable law or agreed to in writing, software |
||||
~ distributed under the License is distributed on an "AS IS" BASIS, |
||||
~ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
~ See the License for the specific language governing permissions and |
||||
~ limitations under the License. |
||||
--> |
||||
|
||||
<project xmlns="http://maven.apache.org/POM/4.0.0" |
||||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" |
||||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd"> |
||||
<parent> |
||||
<artifactId>dolphinscheduler-datasource-plugin</artifactId> |
||||
<groupId>org.apache.dolphinscheduler</groupId> |
||||
<version>dev-SNAPSHOT</version> |
||||
</parent> |
||||
<modelVersion>4.0.0</modelVersion> |
||||
|
||||
<artifactId>dolphinscheduler-datasource-athena</artifactId> |
||||
<packaging>jar</packaging> |
||||
|
||||
<dependencies> |
||||
<dependency> |
||||
<groupId>org.apache.dolphinscheduler</groupId> |
||||
<artifactId>dolphinscheduler-spi</artifactId> |
||||
<scope>provided</scope> |
||||
</dependency> |
||||
<dependency> |
||||
<groupId>org.apache.dolphinscheduler</groupId> |
||||
<artifactId>dolphinscheduler-datasource-api</artifactId> |
||||
</dependency> |
||||
|
||||
</dependencies> |
||||
</project> |
||||
@ -0,0 +1,30 @@
|
||||
/* |
||||
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||
* contributor license agreements. See the NOTICE file distributed with |
||||
* this work for additional information regarding copyright ownership. |
||||
* The ASF licenses this file to You under the Apache License, Version 2.0 |
||||
* (the "License"); you may not use this file except in compliance with |
||||
* the License. You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
package org.apache.dolphinscheduler.plugin.datasource.athena; |
||||
|
||||
import org.apache.dolphinscheduler.spi.datasource.BaseConnectionParam; |
||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel; |
||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceClient; |
||||
import org.apache.dolphinscheduler.spi.enums.DbType; |
||||
|
||||
public class AthenaDataSourceChannel implements DataSourceChannel { |
||||
@Override |
||||
public DataSourceClient createDataSourceClient(BaseConnectionParam baseConnectionParam, DbType dbType) { |
||||
return new AthenaDataSourceClient(baseConnectionParam,dbType); |
||||
} |
||||
} |
||||
@ -0,0 +1,36 @@
|
||||
/* |
||||
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||
* contributor license agreements. See the NOTICE file distributed with |
||||
* this work for additional information regarding copyright ownership. |
||||
* The ASF licenses this file to You under the Apache License, Version 2.0 |
||||
* (the "License"); you may not use this file except in compliance with |
||||
* the License. You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
package org.apache.dolphinscheduler.plugin.datasource.athena; |
||||
|
||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannel; |
||||
import org.apache.dolphinscheduler.spi.datasource.DataSourceChannelFactory; |
||||
|
||||
import com.google.auto.service.AutoService; |
||||
|
||||
@AutoService(DataSourceChannelFactory.class) |
||||
public class AthenaDataSourceChannelFactory implements DataSourceChannelFactory { |
||||
@Override |
||||
public DataSourceChannel create() { |
||||
return new AthenaDataSourceChannel(); |
||||
} |
||||
|
||||
@Override |
||||
public String getName() { |
||||
return "athena"; |
||||
} |
||||
} |
||||
@ -0,0 +1,28 @@
|
||||
/* |
||||
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||
* contributor license agreements. See the NOTICE file distributed with |
||||
* this work for additional information regarding copyright ownership. |
||||
* The ASF licenses this file to You under the Apache License, Version 2.0 |
||||
* (the "License"); you may not use this file except in compliance with |
||||
* the License. You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
package org.apache.dolphinscheduler.plugin.datasource.athena; |
||||
|
||||
import org.apache.dolphinscheduler.plugin.datasource.api.client.CommonDataSourceClient; |
||||
import org.apache.dolphinscheduler.spi.datasource.BaseConnectionParam; |
||||
import org.apache.dolphinscheduler.spi.enums.DbType; |
||||
|
||||
public class AthenaDataSourceClient extends CommonDataSourceClient { |
||||
public AthenaDataSourceClient(BaseConnectionParam baseConnectionParam, DbType dbType) { |
||||
super(baseConnectionParam, dbType); |
||||
} |
||||
} |
||||
@ -0,0 +1,49 @@
|
||||
/* |
||||
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||
* contributor license agreements. See the NOTICE file distributed with |
||||
* this work for additional information regarding copyright ownership. |
||||
* The ASF licenses this file to You under the Apache License, Version 2.0 |
||||
* (the "License"); you may not use this file except in compliance with |
||||
* the License. You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
package org.apache.dolphinscheduler.plugin.datasource.athena.param; |
||||
|
||||
import org.apache.dolphinscheduler.spi.datasource.BaseConnectionParam; |
||||
|
||||
public class AthenaConnectionParam extends BaseConnectionParam { |
||||
|
||||
protected String awsRegion; |
||||
|
||||
public String getAwsRegion() { |
||||
return awsRegion; |
||||
} |
||||
|
||||
public void setAwsRegion(String awsRegion) { |
||||
this.awsRegion = awsRegion; |
||||
} |
||||
|
||||
@Override |
||||
public String toString() { |
||||
return "AthenaConnectionParam{" |
||||
+ "user='" + user + '\'' |
||||
+ ", password='" + password + '\'' |
||||
+ ", address='" + address + '\'' |
||||
+ ", database='" + database + '\'' |
||||
+ ", jdbcUrl='" + jdbcUrl + '\'' |
||||
+ ", driverLocation='" + driverLocation + '\'' |
||||
+ ", driverClassName='" + driverClassName + '\'' |
||||
+ ", validationQuery='" + validationQuery + '\'' |
||||
+ ", other='" + other + '\'' |
||||
+ ", awsRegion='" + awsRegion + '\'' |
||||
+ '}'; |
||||
} |
||||
} |
||||
@ -0,0 +1,35 @@
|
||||
/* |
||||
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||
* contributor license agreements. See the NOTICE file distributed with |
||||
* this work for additional information regarding copyright ownership. |
||||
* The ASF licenses this file to You under the Apache License, Version 2.0 |
||||
* (the "License"); you may not use this file except in compliance with |
||||
* the License. You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
package org.apache.dolphinscheduler.plugin.datasource.athena.param; |
||||
|
||||
import org.apache.dolphinscheduler.plugin.datasource.api.datasource.BaseDataSourceParamDTO; |
||||
import org.apache.dolphinscheduler.spi.enums.DbType; |
||||
|
||||
import lombok.Data; |
||||
import lombok.EqualsAndHashCode; |
||||
|
||||
@Data |
||||
@EqualsAndHashCode(callSuper = true) |
||||
public class AthenaDataSourceParamDTO extends BaseDataSourceParamDTO { |
||||
protected String awsRegion; |
||||
|
||||
@Override |
||||
public DbType getType() { |
||||
return DbType.ATHENA; |
||||
} |
||||
} |
||||
@ -0,0 +1,158 @@
|
||||
/* |
||||
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||
* contributor license agreements. See the NOTICE file distributed with |
||||
* this work for additional information regarding copyright ownership. |
||||
* The ASF licenses this file to You under the Apache License, Version 2.0 |
||||
* (the "License"); you may not use this file except in compliance with |
||||
* the License. You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
package org.apache.dolphinscheduler.plugin.datasource.athena.param; |
||||
|
||||
import com.google.auto.service.AutoService; |
||||
import org.apache.commons.collections4.MapUtils; |
||||
import org.apache.dolphinscheduler.plugin.datasource.api.datasource.AbstractDataSourceProcessor; |
||||
import org.apache.dolphinscheduler.plugin.datasource.api.datasource.BaseDataSourceParamDTO; |
||||
import org.apache.dolphinscheduler.plugin.datasource.api.datasource.DataSourceProcessor; |
||||
import org.apache.dolphinscheduler.plugin.datasource.api.utils.PasswordUtils; |
||||
import org.apache.dolphinscheduler.spi.datasource.BaseConnectionParam; |
||||
import org.apache.dolphinscheduler.spi.datasource.ConnectionParam; |
||||
import org.apache.dolphinscheduler.spi.enums.DbType; |
||||
import org.apache.dolphinscheduler.spi.utils.Constants; |
||||
import org.apache.dolphinscheduler.spi.utils.JSONUtils; |
||||
import org.apache.dolphinscheduler.spi.utils.StringUtils; |
||||
|
||||
import java.sql.Connection; |
||||
import java.sql.DriverManager; |
||||
import java.sql.SQLException; |
||||
import java.util.ArrayList; |
||||
import java.util.LinkedHashMap; |
||||
import java.util.List; |
||||
import java.util.Map; |
||||
|
||||
@AutoService(DataSourceProcessor.class) |
||||
public class AthenaDataSourceProcessor extends AbstractDataSourceProcessor { |
||||
|
||||
@Override |
||||
public BaseDataSourceParamDTO castDatasourceParamDTO(String paramJson) { |
||||
return JSONUtils.parseObject(paramJson, AthenaDataSourceParamDTO.class); |
||||
} |
||||
|
||||
@Override |
||||
public BaseDataSourceParamDTO createDatasourceParamDTO(String connectionJson) { |
||||
AthenaConnectionParam |
||||
connectionParams = (AthenaConnectionParam) this.createConnectionParams(connectionJson); |
||||
|
||||
AthenaDataSourceParamDTO |
||||
athenaDatasourceParamDTO = new AthenaDataSourceParamDTO(); |
||||
athenaDatasourceParamDTO.setAwsRegion(connectionParams.getAwsRegion()); |
||||
athenaDatasourceParamDTO.setDatabase(connectionParams.getDatabase()); |
||||
athenaDatasourceParamDTO.setUserName(connectionParams.getUser()); |
||||
athenaDatasourceParamDTO.setOther(this.parseOther(connectionParams.getOther())); |
||||
|
||||
return athenaDatasourceParamDTO; |
||||
} |
||||
|
||||
@Override |
||||
public BaseConnectionParam createConnectionParams(BaseDataSourceParamDTO datasourceParam) { |
||||
AthenaDataSourceParamDTO athenaParam = (AthenaDataSourceParamDTO) datasourceParam; |
||||
String address = String.format("%s%s=%s;", Constants.JDBC_ATHENA, "AwsRegion", athenaParam.getAwsRegion()); |
||||
|
||||
AthenaConnectionParam |
||||
athenaConnectionParam = new AthenaConnectionParam(); |
||||
athenaConnectionParam.setUser(athenaParam.getUserName()); |
||||
athenaConnectionParam.setPassword(PasswordUtils.encodePassword(athenaParam.getPassword())); |
||||
athenaConnectionParam.setAwsRegion(athenaParam.getAwsRegion()); |
||||
athenaConnectionParam.setOther(this.transformOther(athenaParam.getOther())); |
||||
athenaConnectionParam.setAddress(address); |
||||
athenaConnectionParam.setJdbcUrl(address); |
||||
athenaConnectionParam.setDatabase(athenaParam.getDatabase()); |
||||
athenaConnectionParam.setDriverClassName(this.getDatasourceDriver()); |
||||
athenaConnectionParam.setValidationQuery(this.getValidationQuery()); |
||||
athenaConnectionParam.setProps(athenaParam.getOther()); |
||||
|
||||
return athenaConnectionParam; |
||||
} |
||||
|
||||
@Override |
||||
public ConnectionParam createConnectionParams(String connectionJson) { |
||||
return JSONUtils.parseObject(connectionJson, AthenaConnectionParam.class); |
||||
} |
||||
|
||||
@Override |
||||
public String getDatasourceDriver() { |
||||
return Constants.COM_ATHENA_JDBC_DRIVER; |
||||
} |
||||
|
||||
@Override |
||||
public String getValidationQuery() { |
||||
return Constants.ATHENA_VALIDATION_QUERY; |
||||
} |
||||
|
||||
@Override |
||||
public String getJdbcUrl(ConnectionParam connectionParam) { |
||||
AthenaConnectionParam |
||||
athenaConnectionParam = (AthenaConnectionParam) connectionParam; |
||||
if (!StringUtils.isEmpty(athenaConnectionParam.getOther())) { |
||||
return String.format("%s%s", athenaConnectionParam.getJdbcUrl(), athenaConnectionParam.getOther()); |
||||
} |
||||
return athenaConnectionParam.getJdbcUrl(); |
||||
} |
||||
|
||||
@Override |
||||
public Connection getConnection(ConnectionParam connectionParam) throws ClassNotFoundException, SQLException { |
||||
AthenaConnectionParam athenaConnectionParam = (AthenaConnectionParam) connectionParam; |
||||
Class.forName(this.getDatasourceDriver()); |
||||
return DriverManager.getConnection(this.getJdbcUrl(connectionParam), |
||||
athenaConnectionParam.getUser(), PasswordUtils.decodePassword(athenaConnectionParam.getPassword())); |
||||
} |
||||
|
||||
@Override |
||||
public DbType getDbType() { |
||||
return DbType.ATHENA; |
||||
} |
||||
|
||||
@Override |
||||
public DataSourceProcessor create() { |
||||
return new AthenaDataSourceProcessor(); |
||||
} |
||||
|
||||
private String transformOther(Map<String, String> otherMap) { |
||||
if (MapUtils.isNotEmpty(otherMap)) { |
||||
List<String> list = new ArrayList<>(otherMap.size()); |
||||
otherMap.forEach((key, value) -> list.add(String.format("%s=%s", key, value))); |
||||
return String.join(Constants.SEMICOLON, list); |
||||
} |
||||
return null; |
||||
} |
||||
|
||||
private Map<String, String> parseOther(String other) { |
||||
Map<String, String> otherMap = new LinkedHashMap<>(); |
||||
if (StringUtils.isEmpty(other)) { |
||||
return otherMap; |
||||
} |
||||
String[] configs = other.split(Constants.SEMICOLON); |
||||
for (String config : configs) { |
||||
otherMap.put(config.split(Constants.EQUAL_SIGN)[0], config.split(Constants.EQUAL_SIGN)[1]); |
||||
} |
||||
return otherMap; |
||||
} |
||||
|
||||
@Override |
||||
protected void checkHost(String host) { |
||||
// Do not need to set the host, nothing to do
|
||||
} |
||||
|
||||
@Override |
||||
protected void checkDatabasePatter(String database) { |
||||
// Do not need to set the database, nothing to do
|
||||
} |
||||
} |
||||
@ -0,0 +1,120 @@
|
||||
/* |
||||
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||
* contributor license agreements. See the NOTICE file distributed with |
||||
* this work for additional information regarding copyright ownership. |
||||
* The ASF licenses this file to You under the Apache License, Version 2.0 |
||||
* (the "License"); you may not use this file except in compliance with |
||||
* the License. You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
package org.apache.dolphinscheduler.plugin.datasource.athena.param; |
||||
|
||||
import org.apache.dolphinscheduler.plugin.datasource.api.plugin.DataSourceClientProvider; |
||||
import org.apache.dolphinscheduler.plugin.datasource.api.utils.CommonUtils; |
||||
import org.apache.dolphinscheduler.plugin.datasource.api.utils.DataSourceUtils; |
||||
import org.apache.dolphinscheduler.plugin.datasource.api.utils.PasswordUtils; |
||||
import org.apache.dolphinscheduler.spi.enums.DbType; |
||||
import org.apache.dolphinscheduler.spi.utils.Constants; |
||||
|
||||
import java.sql.DriverManager; |
||||
import java.util.HashMap; |
||||
import java.util.Map; |
||||
|
||||
import org.junit.Assert; |
||||
import org.junit.Test; |
||||
import org.junit.runner.RunWith; |
||||
import org.mockito.Mockito; |
||||
import org.powermock.api.mockito.PowerMockito; |
||||
import org.powermock.core.classloader.annotations.PrepareForTest; |
||||
import org.powermock.modules.junit4.PowerMockRunner; |
||||
|
||||
@RunWith(PowerMockRunner.class) |
||||
@PrepareForTest({Class.class, DriverManager.class, DataSourceUtils.class, CommonUtils.class, DataSourceClientProvider.class, PasswordUtils.class}) |
||||
public class AthenaDataSourceProcessorTest { |
||||
|
||||
private final AthenaDataSourceProcessor athenaDataSourceProcessor = new AthenaDataSourceProcessor(); |
||||
|
||||
@Test |
||||
public void testCreateConnectionParams() { |
||||
Map<String, String> props = new HashMap<>(); |
||||
props.put("LogLevel", "6"); |
||||
AthenaDataSourceParamDTO athenaDataSourceParamDTO = new AthenaDataSourceParamDTO(); |
||||
athenaDataSourceParamDTO.setDatabase(""); |
||||
athenaDataSourceParamDTO.setUserName("awsuser"); |
||||
athenaDataSourceParamDTO.setPassword("123456"); |
||||
athenaDataSourceParamDTO.setAwsRegion("cn-north-1"); |
||||
athenaDataSourceParamDTO.setOther(props); |
||||
PowerMockito.mockStatic(PasswordUtils.class); |
||||
PowerMockito.when(PasswordUtils.encodePassword(Mockito.anyString())).thenReturn("test"); |
||||
AthenaConnectionParam connectionParams = (AthenaConnectionParam) this.athenaDataSourceProcessor |
||||
.createConnectionParams(athenaDataSourceParamDTO); |
||||
Assert.assertEquals("jdbc:awsathena://AwsRegion=cn-north-1;", connectionParams.getAddress()); |
||||
Assert.assertEquals("jdbc:awsathena://AwsRegion=cn-north-1;", connectionParams.getJdbcUrl()); |
||||
} |
||||
|
||||
@Test |
||||
public void testCreateConnectionParams2() { |
||||
String connectionJson = "{\"user\":\"awsuser\",\"password\":\"123456\",\"address\":\"jdbc:awsathena://AwsRegion=cn-north-1;\"" |
||||
+ ",\"database\":\"\",\"jdbcUrl\":\"jdbc:awsathena://AwsRegion=cn-north-1;\", \"awsRegion\":\"cn-north-1\"}"; |
||||
AthenaConnectionParam connectionParams = (AthenaConnectionParam) this.athenaDataSourceProcessor |
||||
.createConnectionParams(connectionJson); |
||||
Assert.assertNotNull(connectionParams); |
||||
Assert.assertEquals("awsuser", connectionParams.getUser()); |
||||
Assert.assertEquals("cn-north-1", connectionParams.getAwsRegion()); |
||||
} |
||||
|
||||
@Test |
||||
public void testGetDatasourceDriver() { |
||||
Assert.assertEquals(Constants.COM_ATHENA_JDBC_DRIVER, this.athenaDataSourceProcessor.getDatasourceDriver()); |
||||
} |
||||
|
||||
@Test |
||||
public void testGetJdbcUrl() { |
||||
AthenaConnectionParam athenaConnectionParam = new AthenaConnectionParam(); |
||||
athenaConnectionParam.setJdbcUrl("jdbc:awsathena://AwsRegion=cn-north-1;"); |
||||
athenaConnectionParam.setOther("LogLevel=6;LogPath=/tmp;"); |
||||
Assert.assertEquals("jdbc:awsathena://AwsRegion=cn-north-1;LogLevel=6;LogPath=/tmp;", |
||||
this.athenaDataSourceProcessor.getJdbcUrl(athenaConnectionParam)); |
||||
|
||||
} |
||||
|
||||
@Test |
||||
public void testGetJdbcUrlNoOther() { |
||||
AthenaConnectionParam athenaConnectionParam = new AthenaConnectionParam(); |
||||
athenaConnectionParam.setJdbcUrl("jdbc:awsathena://AwsRegion=cn-north-1;"); |
||||
athenaConnectionParam.setOther(""); |
||||
Assert.assertEquals("jdbc:awsathena://AwsRegion=cn-north-1;", |
||||
this.athenaDataSourceProcessor.getJdbcUrl(athenaConnectionParam)); |
||||
|
||||
} |
||||
|
||||
@Test |
||||
public void testGetDbType() { |
||||
Assert.assertEquals(DbType.ATHENA, this.athenaDataSourceProcessor.getDbType()); |
||||
} |
||||
|
||||
@Test |
||||
public void testGetValidationQuery() { |
||||
Assert.assertEquals(Constants.ATHENA_VALIDATION_QUERY, this.athenaDataSourceProcessor.getValidationQuery()); |
||||
|
||||
} |
||||
|
||||
@Test |
||||
public void testCreateDatasourceParamDTO() { |
||||
String connectionJson = "{\"user\":\"awsuser\",\"password\":\"123456\",\"address\":\"jdbc:awsathena://AwsRegion=cn-north-1;\"" |
||||
+ ",\"database\":\"\",\"jdbcUrl\":\"jdbc:awsathena://AwsRegion=cn-north-1;\", \"awsRegion\":\"cn-north-1\"}"; |
||||
AthenaDataSourceParamDTO athenaDataSourceParamDTO = (AthenaDataSourceParamDTO) this.athenaDataSourceProcessor |
||||
.createDatasourceParamDTO(connectionJson); |
||||
Assert.assertEquals("awsuser", athenaDataSourceParamDTO.getUserName()); |
||||
Assert.assertEquals("cn-north-1", athenaDataSourceParamDTO.getAwsRegion()); |
||||
Assert.assertEquals("", athenaDataSourceParamDTO.getDatabase()); |
||||
} |
||||
} |
||||
Loading…
Reference in new issue