/*** Copyright 2009-2016 the original author or authors.

*

* 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.*/

packageorg.apache.ibatis.executor.statement;importjava.sql.Connection;importjava.sql.SQLException;importjava.sql.Statement;importorg.apache.ibatis.executor.ErrorContext;importorg.apache.ibatis.executor.Executor;importorg.apache.ibatis.executor.ExecutorException;importorg.apache.ibatis.executor.keygen.KeyGenerator;importorg.apache.ibatis.executor.parameter.ParameterHandler;importorg.apache.ibatis.executor.resultset.ResultSetHandler;importorg.apache.ibatis.mapping.BoundSql;importorg.apache.ibatis.mapping.MappedStatement;importorg.apache.ibatis.reflection.factory.ObjectFactory;importorg.apache.ibatis.session.Configuration;importorg.apache.ibatis.session.ResultHandler;importorg.apache.ibatis.session.RowBounds;importorg.apache.ibatis.type.TypeHandlerRegistry;/***@authorClinton Begin*/

public abstract class BaseStatementHandler implementsStatementHandler {protected finalConfiguration configuration;protected finalObjectFactory objectFactory;protected finalTypeHandlerRegistry typeHandlerRegistry;protected finalResultSetHandler resultSetHandler;protected finalParameterHandler parameterHandler;protected finalExecutor executor;protected finalMappedStatement mappedStatement;protected finalRowBounds rowBounds;protectedBoundSql boundSql;protectedBaseStatementHandler(Executor executor, MappedStatement mappedStatement, Object parameterObject, RowBounds rowBounds, ResultHandler resultHandler, BoundSql boundSql) {this.configuration =mappedStatement.getConfiguration();this.executor =executor;this.mappedStatement =mappedStatement;this.rowBounds =rowBounds;this.typeHandlerRegistry =configuration.getTypeHandlerRegistry();this.objectFactory =configuration.getObjectFactory();if (boundSql == null) { //issue #435, get the key before calculating the statement

generateKeys(parameterObject);

boundSql=mappedStatement.getBoundSql(parameterObject);

}this.boundSql =boundSql;this.parameterHandler =configuration.newParameterHandler(mappedStatement, parameterObject, boundSql);this.resultSetHandler =configuration.newResultSetHandler(executor, mappedStatement, rowBounds, parameterHandler, resultHandler, boundSql);

}

@OverridepublicBoundSql getBoundSql() {returnboundSql;

}

@OverridepublicParameterHandler getParameterHandler() {returnparameterHandler;

}

@Overridepublic Statement prepare(Connection connection, Integer transactionTimeout) throwsSQLException {

ErrorContext.instance().sql(boundSql.getSql());

Statement statement= null;try{

statement=instantiateStatement(connection);

setStatementTimeout(statement, transactionTimeout);

setFetchSize(statement);returnstatement;

}catch(SQLException e) {

closeStatement(statement);throwe;

}catch(Exception e) {

closeStatement(statement);throw new ExecutorException("Error preparing statement. Cause: " +e, e);

}

}protected abstract Statement instantiateStatement(Connection connection) throwsSQLException;protected void setStatementTimeout(Statement stmt, Integer transactionTimeout) throwsSQLException {

Integer queryTimeout= null;if (mappedStatement.getTimeout() != null) {

queryTimeout=mappedStatement.getTimeout();

}else if (configuration.getDefaultStatementTimeout() != null) {

queryTimeout=configuration.getDefaultStatementTimeout();

}if (queryTimeout != null) {

stmt.setQueryTimeout(queryTimeout);

}

StatementUtil.applyTransactionTimeout(stmt, queryTimeout, transactionTimeout);

}protected void setFetchSize(Statement stmt) throwsSQLException {

Integer fetchSize=mappedStatement.getFetchSize();if (fetchSize != null) {

stmt.setFetchSize(fetchSize);return;

}

Integer defaultFetchSize=configuration.getDefaultFetchSize();if (defaultFetchSize != null) {

stmt.setFetchSize(defaultFetchSize);

}

}protected voidcloseStatement(Statement statement) {try{if (statement != null) {

statement.close();

}

}catch(SQLException e) {//ignore

}

}protected voidgenerateKeys(Object parameter) {

KeyGenerator keyGenerator=mappedStatement.getKeyGenerator();

ErrorContext.instance().store();

keyGenerator.processBefore(executor, mappedStatement,null, parameter);

ErrorContext.instance().recall();

}

}