DriverDataSourceImpl.java

/**
 * Copyright (C) 2008 Mathieu Carbou <mathieu.carbou@gmail.com>
 *
 * 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.
 */
package com.mycila.testing.plugin.db;

import com.mycila.testing.plugin.db.api.DbDataSource;
import com.mycila.testing.plugin.db.api.DbProp;
import com.mycila.testing.plugin.db.api.Isolation;

import javax.sql.DataSource;
import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.Properties;

/**
 * @author Mathieu Carbou (mathieu.carbou@gmail.com)
 */
final class DriverDataSourceImpl implements DataSource {

    private final String url;
    private String username;
    private String password;
    private Properties connectionProperties = new Properties();
    private Isolation defaultIsolation = Isolation.DEFAULT;

    private DriverDataSourceImpl(String url) {
        this.url = url;
    }

    private DriverDataSourceImpl withUsername(String username) {
        this.username = username;
        return this;
    }

    private DriverDataSourceImpl withPassword(String password) {
        this.password = password;
        return this;
    }

    private DriverDataSourceImpl withProperty(String property, Object value) {
        connectionProperties.put(property, value);
        return this;
    }

    private DriverDataSourceImpl withDefaultIsolation(Isolation isolation) {
        this.defaultIsolation = isolation;
        return this;
    }

    public Connection getConnection() throws SQLException {
        return getConnectionFromDriverManager(username, password);
    }

    public Connection getConnection(String username, String password) throws SQLException {
        return getConnectionFromDriverManager(username, password);
    }

    private Connection getConnectionFromDriverManager(String username, String password) throws SQLException {
        if (username != null) {
            connectionProperties.setProperty("user", username);
        }
        if (password != null) {
            connectionProperties.setProperty("password", password);
        }
        Connection connection = DriverManager.getConnection(url, connectionProperties);
        if (defaultIsolation != Isolation.DEFAULT && connection.getTransactionIsolation() != defaultIsolation.value()) {
            connection.setTransactionIsolation(defaultIsolation.value());
        }
        return connection;
    }

    public PrintWriter getLogWriter() throws SQLException {
        throw new UnsupportedOperationException("getLogWriter");
    }

    public void setLogWriter(PrintWriter out) throws SQLException {
        throw new UnsupportedOperationException("setLogWriter");
    }

    public void setLoginTimeout(int seconds) throws SQLException {
        throw new UnsupportedOperationException("setLoginTimeout");
    }

    public int getLoginTimeout() throws SQLException {
        return 0;
    }

    //---------------------------------------------------------------------
    // Implementation of JDBC 4.0's Wrapper interface
    //---------------------------------------------------------------------

    public Object unwrap(Class iface) throws SQLException {
        if (!DataSource.class.equals(iface)) {
            throw new SQLException("DataSource of type [" + getClass().getName() + "] can only be unwrapped as [javax.sql.DataSource], not as [" + iface.getName());
        }
        return this;
    }

    public boolean isWrapperFor(Class iface) throws SQLException {
        return DataSource.class.equals(iface);
    }

    static DriverDataSourceImpl from(DbDataSource driverDataSource) {
        try {
            Class.forName(driverDataSource.driver().getName());
        } catch (ClassNotFoundException e) {
            throw new RuntimeException(e.getMessage(), e);
        }
        DriverDataSourceImpl dataSource = new DriverDataSourceImpl(driverDataSource.url())
                .withUsername(driverDataSource.username())
                .withPassword(driverDataSource.password())
                .withDefaultIsolation(driverDataSource.defaultIsolation());
        for (DbProp property : driverDataSource.properties()) {
            dataSource.withProperty(property.name(), property.value());
        }
        return dataSource;
    }

}