com.pastdev.jsch.tunnel.TunneledDataSourceWrapper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jsch-extension Show documentation

A set of extensions on top of JSch providing a full SCP protocol implementation, tunneling including multi-hop, a Java 7 FileSystem like implementation for Java 6 and remote command execution

The newest version!

package com.pastdev.jsch.tunnel;


import java.io.Closeable;
import java.io.IOException;
import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.SQLException;
import java.sql.SQLFeatureNotSupportedException;


import javax.sql.DataSource;


import org.slf4j.Logger;
import org.slf4j.LoggerFactory;


/**
 * Wraps a {@link javax.sql.DataSource DataSource}, adding the ability to
 * tunnel over ssh for the connection.
 */
public class TunneledDataSourceWrapper implements DataSource, Closeable {
    private static Logger log = LoggerFactory.getLogger( TunneledDataSourceWrapper.class );
    private DataSource dataSource;
    private TunnelConnectionManager tunnel;

    /**
     * Creates a DataSource that will ensure all of the tunnels specified by
     * tunnel are opened when datasource connections are created, 
     * and closed when the datasource is closed.
     * 
     * @param tunnel The tunnel manager
     * @param dataSource The datasource that requires tunneled connections
     * 
     * @see com.pastdev.jsch.tunnel.TunnelConnectionManager
     */
    public TunneledDataSourceWrapper( TunnelConnectionManager tunnel, DataSource dataSource ) {
        this.tunnel = tunnel;
        this.dataSource = dataSource;
    }

    @Override
    public void close() throws IOException {
        log.info( "closing tunnel" );
        tunnel.close();
    }

    private void ensureTunnelIsOpen() throws SQLException {
        try {
            tunnel.ensureOpen();
        }
        catch ( Exception e ) {
            throw new SQLException( "unable to open tunnel", e );
        }
    }

    @Override
    public Connection getConnection() throws SQLException {
        ensureTunnelIsOpen();
        return dataSource.getConnection();
    }

    @Override
    public Connection getConnection( String username, String password ) throws SQLException {
        ensureTunnelIsOpen();
        return dataSource.getConnection( username, password );
    }

    @Override
    public int getLoginTimeout() throws SQLException {
        return dataSource.getLoginTimeout();
    }

    @Override
    public PrintWriter getLogWriter() throws SQLException {
        return dataSource.getLogWriter();
    }

    @Override
    public java.util.logging.Logger getParentLogger() throws SQLFeatureNotSupportedException {
    	return dataSource.getParentLogger();
    }

    @Override
    public boolean isWrapperFor( Class iface ) throws SQLException {
        if ( dataSource.getClass().equals( iface.getClass() ) ) return true;
        return dataSource.isWrapperFor( iface );
    }

    @Override
    public void setLoginTimeout( int seconds ) throws SQLException {
        dataSource.setLoginTimeout( seconds );
    }

    @Override
    public void setLogWriter( PrintWriter out ) throws SQLException {
        dataSource.setLogWriter( out );
    }

    @SuppressWarnings( "unchecked" )
    @Override
    public  T unwrap( Class iface ) throws SQLException {
        if ( dataSource.getClass().equals( iface.getClass() ) ) return (T) dataSource;
        return dataSource.unwrap( iface );
    }
}