Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.alipay.sofa.tracer.plugins.datasource.ExtendedConnection Maven / Gradle / Ivy
/*
* 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 com.alipay.sofa.tracer.plugins.datasource;
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.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;
/**
* @author shusong.yss
* @author qilong.zql
* @since 2.2.0
*/
public class ExtendedConnection implements Connection {
private final BaseDataSource dataSource;
private final Connection delegate;
private final Prop prop;
public ExtendedConnection(BaseDataSource dataSource, Connection delegate, Prop prop) {
this.dataSource = dataSource;
this.delegate = delegate;
this.prop = prop;
}
public Prop getProp() {
return prop;
}
public Connection getDelegate() {
return delegate;
}
public BaseDataSource getDataSource() {
return dataSource;
}
@Override
public Statement createStatement() throws SQLException {
Statement st = delegate.createStatement();
if (st != null) {
return new ExtendedStatement(this, st, prop);
}
return null;
}
@Override
public Statement createStatement(int resultSetType, int resultSetConcurrency)
throws SQLException {
Statement st = delegate.createStatement(resultSetType, resultSetConcurrency);
if (st != null) {
return new ExtendedStatement(this, st, prop);
}
return null;
}
@Override
public Statement createStatement(int resultSetType, int resultSetConcurrency,
int resultSetHoldability) throws SQLException {
Statement st = delegate.createStatement(resultSetType, resultSetConcurrency,
resultSetHoldability);
if (st != null) {
return new ExtendedStatement(this, st, prop);
}
return null;
}
@Override
public PreparedStatement prepareStatement(String sql) throws SQLException {
PreparedStatement delegatingPreparedStatement = null;
if (prop.isFastDelegate()) {
delegatingPreparedStatement = delegate.prepareStatement(sql);
} else {
delegatingPreparedStatement = new BasePreparedStatement(this) {
@Override
protected PreparedStatement doPrepareStatement(String sql) throws SQLException {
return delegate.prepareStatement(sql);
}
};
}
return new ExtendedPreparedStatement(this, delegatingPreparedStatement, sql, prop);
}
@Override
public PreparedStatement prepareStatement(String sql, final int resultSetType,
final int resultSetConcurrency) throws SQLException {
PreparedStatement delegatingPreparedStatement = null;
if (prop.isFastDelegate()) {
delegatingPreparedStatement = delegate.prepareStatement(sql, resultSetType,
resultSetConcurrency);
} else {
delegatingPreparedStatement = new BasePreparedStatement(this) {
@Override
protected PreparedStatement doPrepareStatement(String sql) throws SQLException {
return delegate.prepareStatement(sql, resultSetType, resultSetConcurrency);
}
};
}
return new ExtendedPreparedStatement(this, delegatingPreparedStatement, sql, prop);
}
@Override
public PreparedStatement prepareStatement(String sql, final int resultSetType,
final int resultSetConcurrency,
final int resultSetHoldability) throws SQLException {
PreparedStatement delegatingPreparedStatement = null;
if (prop.isFastDelegate()) {
delegatingPreparedStatement = delegate.prepareStatement(sql, resultSetType,
resultSetConcurrency, resultSetHoldability);
} else {
delegatingPreparedStatement = new BasePreparedStatement(this) {
@Override
protected PreparedStatement doPrepareStatement(String sql) throws SQLException {
return delegate.prepareStatement(sql, resultSetType, resultSetConcurrency,
resultSetHoldability);
}
};
}
return new ExtendedPreparedStatement(this, delegatingPreparedStatement, sql, prop);
}
@Override
public PreparedStatement prepareStatement(String sql, final int autoGeneratedKeys)
throws SQLException {
PreparedStatement delegatingPreparedStatement = null;
if (prop.isFastDelegate()) {
delegatingPreparedStatement = delegate.prepareStatement(sql, autoGeneratedKeys);
} else {
delegatingPreparedStatement = new BasePreparedStatement(this) {
@Override
protected PreparedStatement doPrepareStatement(String sql) throws SQLException {
return delegate.prepareStatement(sql, autoGeneratedKeys);
}
};
}
return new ExtendedPreparedStatement(this, delegatingPreparedStatement, sql, prop);
}
@Override
public PreparedStatement prepareStatement(String sql, final int[] columnIndexes)
throws SQLException {
PreparedStatement delegatingPreparedStatement = null;
if (prop.isFastDelegate()) {
delegatingPreparedStatement = delegate.prepareStatement(sql, columnIndexes);
} else {
delegatingPreparedStatement = new BasePreparedStatement(this) {
@Override
protected PreparedStatement doPrepareStatement(String sql) throws SQLException {
return delegate.prepareStatement(sql, columnIndexes);
}
};
}
return new ExtendedPreparedStatement(this, delegatingPreparedStatement, sql, prop);
}
@Override
public PreparedStatement prepareStatement(String sql, final String[] columnNames)
throws SQLException {
PreparedStatement delegatingPreparedStatement = null;
if (prop.isFastDelegate()) {
delegatingPreparedStatement = delegate.prepareStatement(sql, columnNames);
} else {
delegatingPreparedStatement = new BasePreparedStatement(this) {
@Override
protected PreparedStatement doPrepareStatement(String sql) throws SQLException {
return delegate.prepareStatement(sql, columnNames);
}
};
}
return new ExtendedPreparedStatement(this, delegatingPreparedStatement, sql, prop);
}
@Override
public CallableStatement prepareCall(String sql, int resultSetType, int resultSetConcurrency,
int resultSetHoldability) throws SQLException {
return delegate.prepareCall(sql, resultSetType, resultSetConcurrency, resultSetHoldability);
}
@Override
public CallableStatement prepareCall(String sql, int resultSetType, int resultSetConcurrency)
throws SQLException {
return delegate.prepareCall(sql, resultSetType, resultSetConcurrency);
}
@Override
public Map> getTypeMap() throws SQLException {
return delegate.getTypeMap();
}
@Override
public void setTypeMap(Map> map) throws SQLException {
delegate.setTypeMap(map);
}
@Override
public void setHoldability(int holdability) throws SQLException {
delegate.setHoldability(holdability);
}
@Override
public int getHoldability() throws SQLException {
return delegate.getHoldability();
}
@Override
public Savepoint setSavepoint() throws SQLException {
return delegate.setSavepoint();
}
@Override
public Savepoint setSavepoint(String name) throws SQLException {
return delegate.setSavepoint(name);
}
@Override
public void rollback(Savepoint savepoint) throws SQLException {
delegate.rollback();
}
@Override
public void releaseSavepoint(Savepoint savepoint) throws SQLException {
delegate.releaseSavepoint(savepoint);
}
@Override
public CallableStatement prepareCall(String sql) throws SQLException {
return delegate.prepareCall(sql);
}
@Override
public String nativeSQL(String sql) throws SQLException {
return delegate.nativeSQL(sql);
}
@Override
public void setAutoCommit(boolean autoCommit) throws SQLException {
delegate.setAutoCommit(autoCommit);
}
@Override
public boolean getAutoCommit() throws SQLException {
return delegate.getAutoCommit();
}
@Override
public void commit() throws SQLException {
delegate.commit();
}
@Override
public void rollback() throws SQLException {
delegate.rollback();
}
@Override
public void close() throws SQLException {
delegate.close();
}
@Override
public boolean isClosed() throws SQLException {
return delegate.isClosed();
}
@Override
public DatabaseMetaData getMetaData() throws SQLException {
return delegate.getMetaData();
}
@Override
public void setReadOnly(boolean readOnly) throws SQLException {
delegate.setReadOnly(readOnly);
}
@Override
public boolean isReadOnly() throws SQLException {
return delegate.isReadOnly();
}
@Override
public void setCatalog(String catalog) throws SQLException {
delegate.setCatalog(catalog);
}
@Override
public String getCatalog() throws SQLException {
return delegate.getCatalog();
}
@Override
public void setTransactionIsolation(int level) throws SQLException {
delegate.setTransactionIsolation(level);
}
@Override
public int getTransactionIsolation() throws SQLException {
return delegate.getTransactionIsolation();
}
@Override
public SQLWarning getWarnings() throws SQLException {
return delegate.getWarnings();
}
@Override
public void clearWarnings() throws SQLException {
delegate.clearWarnings();
}
@Override
public Clob createClob() throws SQLException {
return delegate.createClob();
}
@Override
public Blob createBlob() throws SQLException {
return delegate.createBlob();
}
@Override
public NClob createNClob() throws SQLException {
return delegate.createNClob();
}
@Override
public SQLXML createSQLXML() throws SQLException {
return delegate.createSQLXML();
}
@Override
public boolean isValid(int timeout) throws SQLException {
return delegate.isValid(timeout);
}
@Override
public void setClientInfo(String name, String value) throws SQLClientInfoException {
delegate.setClientInfo(name, value);
}
@Override
public void setClientInfo(Properties properties) throws SQLClientInfoException {
delegate.setClientInfo(properties);
}
@Override
public String getClientInfo(String name) throws SQLException {
return delegate.getClientInfo(name);
}
@Override
public Properties getClientInfo() throws SQLException {
return delegate.getClientInfo();
}
@Override
public Array createArrayOf(String typeName, Object[] elements) throws SQLException {
return delegate.createArrayOf(typeName, elements);
}
@Override
public Struct createStruct(String typeName, Object[] attributes) throws SQLException {
return delegate.createStruct(typeName, attributes);
}
@Override
public void setSchema(String schema) throws SQLException {
delegate.setSchema(schema);
}
@Override
public String getSchema() throws SQLException {
return delegate.getSchema();
}
@Override
public void abort(Executor executor) throws SQLException {
delegate.abort(executor);
}
@Override
public void setNetworkTimeout(Executor executor, int milliseconds) throws SQLException {
delegate.setNetworkTimeout(executor, milliseconds);
}
@Override
public int getNetworkTimeout() throws SQLException {
return delegate.getNetworkTimeout();
}
@Override
public T unwrap(Class iface) throws SQLException {
return delegate.unwrap(iface);
}
@Override
public boolean isWrapperFor(Class iface) throws SQLException {
return delegate.isWrapperFor(iface);
}
}
