com.amazonaws.athena.connectors.jdbc.connection.GenericJdbcConnectionFactory Maven / Gradle / Ivy
The newest version!
/*-
* #%L
* athena-jdbc
* %%
* Copyright (C) 2019 Amazon Web Services
* %%
* Licensed 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.
* #L%
*/
package com.amazonaws.athena.connectors.jdbc.connection;
import org.apache.commons.lang3.Validate;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.sql.Connection;
import java.sql.DriverManager;
import java.util.Map;
import java.util.Properties;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
/**
* Provides a generic jdbc connection factory that can be used to connect to standard databases. Configures following
* defaults if not present:
*
* - Default ports will be used for the engine if not present.
*
*/
public class GenericJdbcConnectionFactory
implements JdbcConnectionFactory
{
private static final Logger LOGGER = LoggerFactory.getLogger(GenericJdbcConnectionFactory.class);
private static final String SECRET_NAME_PATTERN_STRING = "(\\$\\{[a-zA-Z0-9:/_+=.@-]+})";
public static final Pattern SECRET_NAME_PATTERN = Pattern.compile(SECRET_NAME_PATTERN_STRING);
private final DatabaseConnectionInfo databaseConnectionInfo;
private final DatabaseConnectionConfig databaseConnectionConfig;
private final Properties jdbcProperties;
/**
* @param databaseConnectionConfig database connection configuration {@link DatabaseConnectionConfig}
* @param properties JDBC connection properties.
*/
public GenericJdbcConnectionFactory(final DatabaseConnectionConfig databaseConnectionConfig, final Map properties, final DatabaseConnectionInfo databaseConnectionInfo)
{
this.databaseConnectionInfo = Validate.notNull(databaseConnectionInfo, "databaseConnectionInfo must not be null");
this.databaseConnectionConfig = Validate.notNull(databaseConnectionConfig, "databaseEngine must not be null");
this.jdbcProperties = new Properties();
if (properties != null) {
this.jdbcProperties.putAll(properties);
}
}
@Override
public Connection getConnection(final JdbcCredentialProvider jdbcCredentialProvider)
throws Exception
{
final String derivedJdbcString;
if (jdbcCredentialProvider != null) {
Matcher secretMatcher = SECRET_NAME_PATTERN.matcher(databaseConnectionConfig.getJdbcConnectionString());
derivedJdbcString = secretMatcher.replaceAll(Matcher.quoteReplacement(""));
jdbcProperties.put("user", jdbcCredentialProvider.getCredential().getUser());
jdbcProperties.put("password", jdbcCredentialProvider.getCredential().getPassword());
}
else {
derivedJdbcString = databaseConnectionConfig.getJdbcConnectionString();
}
// register driver
Class.forName(databaseConnectionInfo.getDriverClassName()).newInstance();
// create connection
return DriverManager.getConnection(derivedJdbcString, this.jdbcProperties);
}
private String encodeValue(String value)
{
try {
return URLEncoder.encode(value, StandardCharsets.UTF_8.toString());
}
catch (UnsupportedEncodingException ex) {
throw new RuntimeException(ex);
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy