com.alibaba.druid.pool.DruidPooledPreparedStatement Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of druid Show documentation
Show all versions of druid Show documentation
An JDBC datasource implementation.
/*
* Copyright 1999-2011 Alibaba Group Holding Ltd.
*
* 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.alibaba.druid.pool;
import java.io.InputStream;
import java.io.Reader;
import java.math.BigDecimal;
import java.net.URL;
import java.sql.Array;
import java.sql.Blob;
import java.sql.Clob;
import java.sql.Date;
import java.sql.NClob;
import java.sql.ParameterMetaData;
import java.sql.PreparedStatement;
import java.sql.Ref;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.RowId;
import java.sql.SQLException;
import java.sql.SQLXML;
import java.sql.Time;
import java.sql.Timestamp;
import java.util.Arrays;
import java.util.Calendar;
import com.alibaba.druid.pool.PreparedStatementPool.MethodType;
import com.alibaba.druid.proxy.jdbc.PreparedStatementProxy;
import com.alibaba.druid.support.logging.Log;
import com.alibaba.druid.support.logging.LogFactory;
import com.alibaba.druid.util.OracleUtils;
/**
* @author wenshao
*/
public class DruidPooledPreparedStatement extends DruidPooledStatement implements PreparedStatement {
private final static Log LOG = LogFactory.getLog(DruidPooledPreparedStatement.class);
private final PreparedStatementHolder holder;
private final PreparedStatement stmt;
private final String sql;
private int defaultMaxFieldSize;
private int defaultMaxRows;
private int defaultQueryTimeout;
private int defaultFetchDirection;
private int defaultFetchSize;
private int currentMaxFieldSize;
private int currentMaxRows;
private int currentQueryTimeout;
private int currentFetchDirection;
private int currentFetchSize;
public DruidPooledPreparedStatement(DruidPooledConnection conn, PreparedStatementHolder holder) throws SQLException{
super(conn, holder.getStatement());
this.stmt = holder.getStatement();
this.holder = holder;
this.sql = holder.getKey().sql;
// Remember the defaults
try {
defaultMaxFieldSize = stmt.getMaxFieldSize();
} catch (SQLException e) {
LOG.error("getMaxFieldSize error", e);
}
try {
defaultMaxRows = stmt.getMaxRows();
} catch (SQLException e) {
LOG.error("getMaxRows error", e);
}
try {
defaultQueryTimeout = stmt.getQueryTimeout();
} catch (SQLException e) {
LOG.error("getMaxRows error", e);
}
try {
defaultFetchDirection = stmt.getFetchDirection();
} catch (SQLException e) {
LOG.error("getFetchDirection error", e);
}
try {
defaultFetchSize = stmt.getFetchSize();
} catch (SQLException e) {
LOG.error("getFetchSize error", e);
}
currentMaxFieldSize = defaultMaxFieldSize;
currentMaxRows = defaultMaxRows;
currentQueryTimeout = defaultQueryTimeout;
currentFetchDirection = defaultFetchDirection;
currentFetchSize = defaultFetchSize;
}
public PreparedStatementHolder getPreparedStatementHolder() {
return holder;
}
public int getHitCount() {
return holder.getHitCount();
}
public void setFetchSize(int rows) throws SQLException {
currentFetchSize = rows;
super.setFetchSize(rows);
}
public void setFetchDirection(int direction) throws SQLException {
currentFetchDirection = direction;
super.setFetchDirection(direction);
}
public void setMaxFieldSize(int max) throws SQLException {
currentMaxFieldSize = max;
super.setMaxFieldSize(max);
}
public void setMaxRows(int max) throws SQLException {
currentMaxRows = max;
super.setMaxRows(max);
}
public void setQueryTimeout(int seconds) throws SQLException {
currentQueryTimeout = seconds;
super.setQueryTimeout(seconds);
}
public String getSql() {
return sql;
}
public PreparedStatementKey getKey() {
return holder.getKey();
}
public PreparedStatement getRawPreparedStatement() {
return stmt;
}
public PreparedStatement getRawStatement() {
return stmt;
}
@Override
public void close() throws SQLException {
if (isClosed()) {
return;
}
// Reset the defaults
if (defaultMaxFieldSize != currentMaxFieldSize) {
stmt.setMaxFieldSize(defaultMaxFieldSize);
currentMaxFieldSize = defaultMaxFieldSize;
}
if (defaultMaxRows != currentMaxRows) {
stmt.setMaxRows(defaultMaxRows);
currentMaxRows = defaultMaxRows;
}
if (defaultQueryTimeout != currentQueryTimeout) {
stmt.setQueryTimeout(defaultQueryTimeout);
currentQueryTimeout = defaultQueryTimeout;
}
if (defaultFetchDirection != currentFetchDirection) {
stmt.setFetchDirection(defaultFetchDirection);
currentFetchDirection = defaultFetchDirection;
}
if (defaultFetchSize != currentFetchSize) {
stmt.setFetchSize(defaultFetchSize);
currentFetchSize = defaultFetchSize;
}
conn.closePoolableStatement(this);
}
void closeInternal() throws SQLException {
super.close();
}
void setClosed(boolean value) {
this.closed = value;
}
@Override
public ResultSet executeQuery() throws SQLException {
checkOpen();
incrementExecuteCount();
transactionRecord(sql);
oracleSetRowPrefetch();
conn.beforeExecute();
try {
ResultSet rs = stmt.executeQuery();
if (rs == null) {
return null;
}
DruidPooledResultSet poolableResultSet = new DruidPooledResultSet(this, rs);
resultSetTrace.add(poolableResultSet);
return poolableResultSet;
} catch (Throwable t) {
throw checkException(t);
} finally {
conn.afterExecute();
}
}
@Override
public int executeUpdate() throws SQLException {
checkOpen();
incrementExecuteCount();
transactionRecord(sql);
conn.beforeExecute();
try {
return stmt.executeUpdate();
} catch (Throwable t) {
throw checkException(t);
} finally {
conn.afterExecute();
}
}
@Override
public void setNull(int parameterIndex, int sqlType) throws SQLException {
checkOpen();
try {
stmt.setNull(parameterIndex, sqlType);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setBoolean(int parameterIndex, boolean x) throws SQLException {
checkOpen();
try {
stmt.setBoolean(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setByte(int parameterIndex, byte x) throws SQLException {
checkOpen();
try {
stmt.setByte(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setShort(int parameterIndex, short x) throws SQLException {
checkOpen();
try {
stmt.setShort(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setInt(int parameterIndex, int x) throws SQLException {
checkOpen();
try {
stmt.setInt(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setLong(int parameterIndex, long x) throws SQLException {
checkOpen();
try {
stmt.setLong(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setFloat(int parameterIndex, float x) throws SQLException {
checkOpen();
try {
stmt.setFloat(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setDouble(int parameterIndex, double x) throws SQLException {
checkOpen();
try {
stmt.setDouble(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setBigDecimal(int parameterIndex, BigDecimal x) throws SQLException {
checkOpen();
try {
stmt.setBigDecimal(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setString(int parameterIndex, String x) throws SQLException {
checkOpen();
try {
stmt.setString(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setBytes(int parameterIndex, byte[] x) throws SQLException {
checkOpen();
try {
stmt.setBytes(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setDate(int parameterIndex, Date x) throws SQLException {
checkOpen();
try {
stmt.setDate(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setTime(int parameterIndex, Time x) throws SQLException {
checkOpen();
try {
stmt.setTime(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setTimestamp(int parameterIndex, Timestamp x) throws SQLException {
checkOpen();
try {
stmt.setTimestamp(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setAsciiStream(int parameterIndex, InputStream x, int length) throws SQLException {
checkOpen();
try {
stmt.setAsciiStream(parameterIndex, x, length);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
@Deprecated
public void setUnicodeStream(int parameterIndex, InputStream x, int length) throws SQLException {
checkOpen();
try {
stmt.setUnicodeStream(parameterIndex, x, length);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setBinaryStream(int parameterIndex, InputStream x, int length) throws SQLException {
checkOpen();
try {
stmt.setBinaryStream(parameterIndex, x, length);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void clearParameters() throws SQLException {
checkOpen();
try {
stmt.clearParameters();
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setObject(int parameterIndex, Object x, int targetSqlType) throws SQLException {
checkOpen();
try {
stmt.setObject(parameterIndex, x, targetSqlType);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setObject(int parameterIndex, Object x) throws SQLException {
checkOpen();
try {
stmt.setObject(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public boolean execute() throws SQLException {
checkOpen();
incrementExecuteCount();
transactionRecord(sql);
// oracleSetRowPrefetch();
conn.beforeExecute();
try {
return stmt.execute();
} catch (Throwable t) {
throw checkException(t);
} finally {
conn.afterExecute();
}
}
protected void oracleSetRowPrefetch() throws SQLException {
if (!conn.isOracle()) {
return;
}
if (holder.getHitCount() == 0) {
return;
}
int fetchRowPeak = holder.getFetchRowPeak();
if (fetchRowPeak < 0) {
return;
}
if (holder.getDefaultRowPretch() == -1) {
int defaultRowPretch = OracleUtils.getRowPrefetch(this);
holder.setDefaultRowPretch(defaultRowPretch);
holder.setRowPrefetch(defaultRowPretch);
}
int rowPrefetch;
if (fetchRowPeak <= 1) {
rowPrefetch = 2;
} else if (fetchRowPeak > holder.getDefaultRowPretch()) {
rowPrefetch = holder.getDefaultRowPretch();
} else {
rowPrefetch = fetchRowPeak + 1;
}
if (rowPrefetch != holder.getRowPrefetch()) {
OracleUtils.setRowPrefetch(this, rowPrefetch);
holder.setRowPrefetch(rowPrefetch);
}
}
@Override
public void addBatch() throws SQLException {
checkOpen();
try {
stmt.addBatch();
} catch (Throwable t) {
throw checkException(t);
}
}
public int[] executeBatch() throws SQLException {
checkOpen();
incrementExecuteCount();
transactionRecord(sql);
conn.beforeExecute();
try {
return stmt.executeBatch();
} catch (Throwable t) {
throw checkException(t);
} finally {
conn.afterExecute();
}
}
@Override
public void setCharacterStream(int parameterIndex, Reader reader, int length) throws SQLException {
checkOpen();
try {
stmt.setCharacterStream(parameterIndex, reader, length);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setRef(int parameterIndex, Ref x) throws SQLException {
checkOpen();
try {
stmt.setRef(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setBlob(int parameterIndex, Blob x) throws SQLException {
checkOpen();
try {
stmt.setBlob(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setClob(int parameterIndex, Clob x) throws SQLException {
checkOpen();
try {
stmt.setClob(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setArray(int parameterIndex, Array x) throws SQLException {
checkOpen();
try {
stmt.setArray(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public ResultSetMetaData getMetaData() throws SQLException {
checkOpen();
try {
return stmt.getMetaData();
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setDate(int parameterIndex, Date x, Calendar cal) throws SQLException {
checkOpen();
try {
stmt.setDate(parameterIndex, x, cal);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setTime(int parameterIndex, Time x, Calendar cal) throws SQLException {
checkOpen();
try {
stmt.setTime(parameterIndex, x, cal);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setTimestamp(int parameterIndex, Timestamp x, Calendar cal) throws SQLException {
checkOpen();
try {
stmt.setTimestamp(parameterIndex, x, cal);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setNull(int parameterIndex, int sqlType, String typeName) throws SQLException {
checkOpen();
try {
stmt.setNull(parameterIndex, sqlType, typeName);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setURL(int parameterIndex, URL x) throws SQLException {
checkOpen();
try {
stmt.setURL(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public ParameterMetaData getParameterMetaData() throws SQLException {
checkOpen();
try {
return stmt.getParameterMetaData();
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setRowId(int parameterIndex, RowId x) throws SQLException {
checkOpen();
try {
stmt.setRowId(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setNString(int parameterIndex, String value) throws SQLException {
checkOpen();
try {
stmt.setNString(parameterIndex, value);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setNCharacterStream(int parameterIndex, Reader value, long length) throws SQLException {
checkOpen();
try {
stmt.setNCharacterStream(parameterIndex, value, length);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setNClob(int parameterIndex, NClob value) throws SQLException {
checkOpen();
try {
stmt.setNClob(parameterIndex, value);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setClob(int parameterIndex, Reader reader, long length) throws SQLException {
checkOpen();
try {
stmt.setClob(parameterIndex, reader, length);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setBlob(int parameterIndex, InputStream inputStream, long length) throws SQLException {
checkOpen();
try {
stmt.setBlob(parameterIndex, inputStream, length);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setNClob(int parameterIndex, Reader reader, long length) throws SQLException {
checkOpen();
try {
stmt.setNClob(parameterIndex, reader, length);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setSQLXML(int parameterIndex, SQLXML xmlObject) throws SQLException {
checkOpen();
try {
stmt.setSQLXML(parameterIndex, xmlObject);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setObject(int parameterIndex, Object x, int targetSqlType, int scaleOrLength) throws SQLException {
checkOpen();
try {
stmt.setObject(parameterIndex, x, targetSqlType, scaleOrLength);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setAsciiStream(int parameterIndex, InputStream x, long length) throws SQLException {
checkOpen();
try {
stmt.setAsciiStream(parameterIndex, x, length);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setBinaryStream(int parameterIndex, InputStream x, long length) throws SQLException {
checkOpen();
try {
stmt.setBinaryStream(parameterIndex, x, length);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setCharacterStream(int parameterIndex, Reader reader, long length) throws SQLException {
checkOpen();
try {
stmt.setCharacterStream(parameterIndex, reader, length);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setAsciiStream(int parameterIndex, InputStream x) throws SQLException {
checkOpen();
try {
stmt.setAsciiStream(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setBinaryStream(int parameterIndex, InputStream x) throws SQLException {
checkOpen();
try {
stmt.setBinaryStream(parameterIndex, x);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setCharacterStream(int parameterIndex, Reader reader) throws SQLException {
checkOpen();
try {
stmt.setCharacterStream(parameterIndex, reader);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setNCharacterStream(int parameterIndex, Reader value) throws SQLException {
checkOpen();
try {
stmt.setNCharacterStream(parameterIndex, value);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setClob(int parameterIndex, Reader reader) throws SQLException {
checkOpen();
try {
stmt.setClob(parameterIndex, reader);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setBlob(int parameterIndex, InputStream inputStream) throws SQLException {
checkOpen();
try {
stmt.setBlob(parameterIndex, inputStream);
} catch (Throwable t) {
throw checkException(t);
}
}
@Override
public void setNClob(int parameterIndex, Reader reader) throws SQLException {
checkOpen();
try {
stmt.setNClob(parameterIndex, reader);
} catch (Throwable t) {
throw checkException(t);
}
}
public static class PreparedStatementKey {
private final String sql;
private final String catalog;
protected final MethodType methodType;
private final int resultSetType;
private final int resultSetConcurrency;
private final int resultSetHoldability;
private final int autoGeneratedKeys;
private final int[] columnIndexes;
private final String[] columnNames;
public PreparedStatementKey(String sql, String catalog, MethodType methodType) throws SQLException{
this(sql, catalog, methodType, 0, 0, 0, 0, null, null);
}
public PreparedStatementKey(String sql, String catalog, MethodType methodType, int resultSetType,
int resultSetConcurrency) throws SQLException{
this(sql, catalog, methodType, resultSetType, resultSetConcurrency, 0, 0, null, null);
}
public PreparedStatementKey(String sql, String catalog, MethodType methodType, int resultSetType,
int resultSetConcurrency, int resultSetHoldability) throws SQLException{
this(sql, catalog, methodType, resultSetType, resultSetConcurrency, resultSetHoldability, 0, null, null);
}
public PreparedStatementKey(String sql, String catalog, MethodType methodType, int autoGeneratedKeys)
throws SQLException{
this(sql, catalog, methodType, 0, 0, 0, autoGeneratedKeys, null, null);
}
public PreparedStatementKey(String sql, String catalog, MethodType methodType, int[] columnIndexes)
throws SQLException{
this(sql, catalog, methodType, 0, 0, 0, 0, columnIndexes, null);
}
public PreparedStatementKey(String sql, String catalog, MethodType methodType, String[] columnNames)
throws SQLException{
this(sql, catalog, methodType, 0, 0, 0, 0, null, columnNames);
}
public PreparedStatementKey(String sql, String catalog, MethodType methodType, int resultSetType,
int resultSetConcurrency, int resultSetHoldability, int autoGeneratedKeys,
int[] columnIndexes, String[] columnNames) throws SQLException{
if (sql == null) {
throw new SQLException("sql is null");
}
this.sql = sql;
this.catalog = catalog;
this.methodType = methodType;
this.resultSetType = resultSetType;
this.resultSetConcurrency = resultSetConcurrency;
this.resultSetHoldability = resultSetHoldability;
this.autoGeneratedKeys = autoGeneratedKeys;
this.columnIndexes = columnIndexes;
this.columnNames = columnNames;
}
public int getResultSetType() {
return resultSetType;
}
public int getResultSetConcurrency() {
return resultSetConcurrency;
}
public int getResultSetHoldability() {
return resultSetHoldability;
}
public boolean equals(Object object) {
PreparedStatementKey that = (PreparedStatementKey) object;
if (!this.sql.equals(that.sql)) {
return false;
}
if (this.catalog == null) {
if (that.catalog != null) {
return false;
}
} else {
if (!this.catalog.equals(that.catalog)) {
return false;
}
}
if (this.methodType != that.methodType) {
return false;
}
if (this.resultSetType != that.resultSetType) {
return false;
}
if (this.resultSetConcurrency != that.resultSetConcurrency) {
return false;
}
if (this.resultSetHoldability != that.resultSetHoldability) {
return false;
}
if (this.autoGeneratedKeys != that.autoGeneratedKeys) {
return false;
}
if (!Arrays.equals(columnIndexes, that.columnIndexes)) {
return false;
}
if (!Arrays.equals(columnNames, that.columnNames)) {
return false;
}
return true;
}
public int hashCode() {
final int prime = 31;
int result = 1;
result = prime * result + ((sql == null) ? 0 : sql.hashCode());
result = prime * result + ((catalog == null) ? 0 : catalog.hashCode());
result = prime * result + ((methodType == null) ? 0 : methodType.hashCode());
result = prime * result + resultSetConcurrency;
result = prime * result + resultSetHoldability;
result = prime * result + resultSetType;
result = prime * result + autoGeneratedKeys;
result = prime * result + Arrays.hashCode(columnIndexes);
result = prime * result + Arrays.hashCode(columnNames);
return result;
}
public String getSql() {
return sql;
}
}
@SuppressWarnings("unchecked")
public T unwrap(Class iface) throws SQLException {
if (iface == PreparedStatement.class) {
if (stmt instanceof PreparedStatementProxy) {
return stmt.unwrap(iface);
}
return (T) stmt;
}
if (iface == PreparedStatementHolder.class) {
return (T) this.holder;
}
return super.unwrap(iface);
}
}