Java tutorial
/** * Copyright 2017 Goldman Sachs. * 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.gs.obevo.db.impl.core.jdbc; import java.io.PrintWriter; import java.sql.Array; import java.sql.Blob; import java.sql.CallableStatement; import java.sql.Clob; import java.sql.Connection; import java.sql.DatabaseMetaData; import java.sql.NClob; import java.sql.PreparedStatement; import java.sql.SQLClientInfoException; import java.sql.SQLException; import java.sql.SQLFeatureNotSupportedException; import java.sql.SQLWarning; import java.sql.SQLXML; import java.sql.Savepoint; import java.sql.Statement; import java.sql.Struct; import java.util.Map; import java.util.Properties; import java.util.concurrent.Executor; import javax.sql.DataSource; import org.apache.commons.dbutils.DbUtils; /** * Data source that will keep a single connect open for its lifetime duration. Meant for classes to easily access the * database for testing purposes. */ public class SingleConnectionDataSource implements DataSource { private final DataSource ds; private Connection connection; private PrintWriter logWriter; private int loginTimeout; public SingleConnectionDataSource(DataSource ds) { this.ds = ds; } @Override public Connection getConnection() throws SQLException { if (this.connection == null) { this.connection = new ManagedDataSourceConnection(ds.getConnection()); } return this.connection; } @Override public Connection getConnection(String username, String password) throws SQLException { throw new UnsupportedOperationException( "getConnection call w/ username and password params is not supported"); } @Override public PrintWriter getLogWriter() throws SQLException { return logWriter; } @Override public void setLogWriter(PrintWriter logWriter) throws SQLException { this.logWriter = logWriter; } @Override public void setLoginTimeout(int loginTimeout) throws SQLException { this.loginTimeout = loginTimeout; } @Override public int getLoginTimeout() throws SQLException { return loginTimeout; } @Override public <T> T unwrap(Class<T> iface) throws SQLException { throw new SQLException(this.getClass().getName() + " is not a wrapper."); } @Override public boolean isWrapperFor(Class<?> iface) throws SQLException { return false; } @Override protected void finalize() throws Throwable { DbUtils.closeQuietly(connection); } public java.util.logging.Logger getParentLogger() throws SQLFeatureNotSupportedException { throw new SQLFeatureNotSupportedException( "To be implemented as getDataSource().getParentLogger() once default compile is 1.7"); } private static class ManagedDataSourceConnection implements Connection { private final Connection conn; public ManagedDataSourceConnection(Connection conn) { this.conn = conn; } @Override public void close() throws SQLException { // not closing - txn manager manages it } @Override public Statement createStatement() throws SQLException { return this.conn.createStatement(); } @Override public PreparedStatement prepareStatement(String sql) throws SQLException { return this.conn.prepareStatement(sql); } @Override public CallableStatement prepareCall(String sql) throws SQLException { return this.conn.prepareCall(sql); } @Override public String nativeSQL(String sql) throws SQLException { return this.conn.nativeSQL(sql); } @Override public void setAutoCommit(boolean autoCommit) throws SQLException { this.conn.setAutoCommit(autoCommit); } @Override public boolean getAutoCommit() throws SQLException { return this.conn.getAutoCommit(); } @Override public void commit() throws SQLException { this.conn.commit(); } @Override public void rollback() throws SQLException { this.conn.rollback(); } @Override public boolean isClosed() throws SQLException { return this.conn.isClosed(); } @Override public DatabaseMetaData getMetaData() throws SQLException { return this.conn.getMetaData(); } @Override public void setReadOnly(boolean readOnly) throws SQLException { this.conn.setReadOnly(readOnly); } @Override public boolean isReadOnly() throws SQLException { return this.conn.isReadOnly(); } @Override public void setCatalog(String catalog) throws SQLException { this.conn.setCatalog(catalog); } @Override public String getCatalog() throws SQLException { return this.conn.getCatalog(); } @Override public void setTransactionIsolation(int level) throws SQLException { this.conn.setTransactionIsolation(level); } @Override public int getTransactionIsolation() throws SQLException { return this.conn.getTransactionIsolation(); } @Override public SQLWarning getWarnings() throws SQLException { return this.conn.getWarnings(); } @Override public void clearWarnings() throws SQLException { this.conn.clearWarnings(); } @Override public Statement createStatement(int resultSetType, int resultSetConcurrency) throws SQLException { return this.conn.createStatement(resultSetType, resultSetConcurrency); } @Override public PreparedStatement prepareStatement(String sql, int resultSetType, int resultSetConcurrency) throws SQLException { return this.conn.prepareStatement(sql, resultSetType, resultSetConcurrency); } @Override public CallableStatement prepareCall(String sql, int resultSetType, int resultSetConcurrency) throws SQLException { return this.conn.prepareCall(sql, resultSetType, resultSetConcurrency); } @Override public Map<String, Class<?>> getTypeMap() throws SQLException { return this.conn.getTypeMap(); } @Override public void setTypeMap(Map<String, Class<?>> map) throws SQLException { this.conn.setTypeMap(map); } @Override public void setHoldability(int holdability) throws SQLException { this.conn.setHoldability(holdability); } @Override public int getHoldability() throws SQLException { return this.conn.getHoldability(); } @Override public Savepoint setSavepoint() throws SQLException { return this.conn.setSavepoint(); } @Override public Savepoint setSavepoint(String name) throws SQLException { return this.conn.setSavepoint(name); } @Override public void rollback(Savepoint savepoint) throws SQLException { this.conn.rollback(savepoint); } @Override public void releaseSavepoint(Savepoint savepoint) throws SQLException { this.conn.releaseSavepoint(savepoint); } @Override public Statement createStatement(int resultSetType, int resultSetConcurrency, int resultSetHoldability) throws SQLException { return this.conn.createStatement(resultSetType, resultSetConcurrency, resultSetHoldability); } @Override public PreparedStatement prepareStatement(String sql, int resultSetType, int resultSetConcurrency, int resultSetHoldability) throws SQLException { return this.conn.prepareStatement(sql, resultSetType, resultSetConcurrency, resultSetHoldability); } @Override public CallableStatement prepareCall(String sql, int resultSetType, int resultSetConcurrency, int resultSetHoldability) throws SQLException { return this.conn.prepareCall(sql, resultSetType, resultSetConcurrency, resultSetHoldability); } @Override public PreparedStatement prepareStatement(String sql, int autoGeneratedKeys) throws SQLException { return this.conn.prepareStatement(sql, autoGeneratedKeys); } @Override public PreparedStatement prepareStatement(String sql, int[] columnIndexes) throws SQLException { return this.conn.prepareStatement(sql, columnIndexes); } @Override public PreparedStatement prepareStatement(String sql, String[] columnNames) throws SQLException { return this.conn.prepareStatement(sql, columnNames); } @Override public Clob createClob() throws SQLException { return this.conn.createClob(); } @Override public Blob createBlob() throws SQLException { return this.conn.createBlob(); } @Override public NClob createNClob() throws SQLException { return this.conn.createNClob(); } @Override public SQLXML createSQLXML() throws SQLException { return this.conn.createSQLXML(); } @Override public boolean isValid(int timeout) throws SQLException { return this.conn.isValid(timeout); } @Override public void setClientInfo(String name, String value) throws SQLClientInfoException { this.conn.setClientInfo(name, value); } @Override public void setClientInfo(Properties properties) throws SQLClientInfoException { this.conn.setClientInfo(properties); } @Override public String getClientInfo(String name) throws SQLException { return this.conn.getClientInfo(name); } @Override public Properties getClientInfo() throws SQLException { return this.conn.getClientInfo(); } @Override public Array createArrayOf(String typeName, Object[] elements) throws SQLException { return this.conn.createArrayOf(typeName, elements); } @Override public Struct createStruct(String typeName, Object[] attributes) throws SQLException { return this.conn.createStruct(typeName, attributes); } @Override public <T> T unwrap(Class<T> iface) throws SQLException { return this.conn.unwrap(iface); } @Override public boolean isWrapperFor(Class<?> iface) throws SQLException { return this.conn.isWrapperFor(iface); } public void setSchema(String schema) throws SQLException { throw new UnsupportedOperationException( "Only implemented in Java 1.7; here for now to avoid compile errors"); } public String getSchema() throws SQLException { throw new UnsupportedOperationException( "Only implemented in Java 1.7; here for now to avoid compile errors"); } public void abort(Executor executor) throws SQLException { throw new UnsupportedOperationException( "Only implemented in Java 1.7; here for now to avoid compile errors"); } public void setNetworkTimeout(Executor executor, int milliseconds) throws SQLException { throw new UnsupportedOperationException( "Only implemented in Java 1.7; here for now to avoid compile errors"); } public int getNetworkTimeout() throws SQLException { throw new UnsupportedOperationException( "Only implemented in Java 1.7; here for now to avoid compile errors"); } } }