org.apache.cayenne.access.jdbc.ProcedureAction Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of cayenne-client-nodeps
Show all versions of cayenne-client-nodeps
Cayenne Object Persistence Framework
/*****************************************************************
* 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 org.apache.cayenne.access.jdbc;
import java.sql.CallableStatement;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.apache.cayenne.CayenneRuntimeException;
import org.apache.cayenne.access.OperationObserver;
import org.apache.cayenne.access.QueryLogger;
import org.apache.cayenne.access.trans.ProcedureTranslator;
import org.apache.cayenne.access.types.ExtendedType;
import org.apache.cayenne.dba.DbAdapter;
import org.apache.cayenne.map.EntityResolver;
import org.apache.cayenne.map.Procedure;
import org.apache.cayenne.map.ProcedureParameter;
import org.apache.cayenne.query.ProcedureQuery;
/**
* A SQLAction that runs a stored procedure. Note that ProcedureAction has internal state
* and is not thread-safe.
*
* @since 1.2
* @author Andrus Adamchik
*/
public class ProcedureAction extends BaseSQLAction {
protected ProcedureQuery query;
/**
* Holds a number of ResultSets processed by the action. This value is reset to zero
* on every "performAction" call.
*/
protected int processedResultSets;
public ProcedureAction(ProcedureQuery query, DbAdapter adapter,
EntityResolver entityResolver) {
super(adapter, entityResolver);
this.query = query;
}
public void performAction(Connection connection, OperationObserver observer)
throws SQLException, Exception {
processedResultSets = 0;
ProcedureTranslator transl = createTranslator(connection);
CallableStatement statement = (CallableStatement) transl.createStatement();
try {
// stored procedure may contain a mixture of update counts and result sets,
// and out parameters. Read out parameters first, then
// iterate until we exhaust all results
statement.execute();
// read out parameters
readProcedureOutParameters(statement, observer);
// read the rest of the query
while (true) {
if (statement.getMoreResults()) {
ResultSet rs = statement.getResultSet();
try {
RowDescriptor descriptor = describeResultSet(
rs,
processedResultSets++);
readResultSet(rs, descriptor, query, observer);
}
finally {
try {
rs.close();
}
catch (SQLException ex) {
}
}
}
else {
int updateCount = statement.getUpdateCount();
if (updateCount == -1) {
break;
}
QueryLogger.logUpdateCount(updateCount);
observer.nextCount(query, updateCount);
}
}
}
finally {
try {
statement.close();
}
catch (SQLException ex) {
}
}
}
/**
* Returns the ProcedureTranslator to use for this ProcedureAction.
*
* @param connection JDBC connection
*/
protected ProcedureTranslator createTranslator(Connection connection) {
ProcedureTranslator transl = new ProcedureTranslator();
transl.setAdapter(getAdapter());
transl.setQuery(query);
transl.setEntityResolver(getEntityResolver());
transl.setConnection(connection);
return transl;
}
/**
* Creates a RowDescriptor for result set.
*
* @param resultSet JDBC ResultSet
* @param setIndex a zero-based index of the ResultSet in the query results.
*/
protected RowDescriptor describeResultSet(ResultSet resultSet, int setIndex) {
if (setIndex < 0) {
throw new IllegalArgumentException(
"Expected a non-negative result set index. Got: " + setIndex);
}
List descriptors = query.getResultDescriptors();
if (descriptors.isEmpty()) {
// failover to default JDBC result descriptor
return new RowDescriptor(resultSet, getAdapter().getExtendedTypes());
}
// if one result is described, all of them must be present...
if (setIndex >= descriptors.size() || descriptors.get(setIndex) == null) {
throw new CayenneRuntimeException("No descriptor for result set at index '"
+ setIndex
+ "' configured.");
}
ColumnDescriptor[] columns = (ColumnDescriptor[]) descriptors.get(setIndex);
return new RowDescriptor(columns, getAdapter().getExtendedTypes());
}
/**
* Returns stored procedure for an internal query.
*/
protected Procedure getProcedure() {
return getEntityResolver().lookupProcedure(query);
}
/**
* Helper method that reads OUT parameters of a CallableStatement.
*/
protected void readProcedureOutParameters(
CallableStatement statement,
OperationObserver delegate) throws SQLException, Exception {
long t1 = System.currentTimeMillis();
// build result row...
Map result = null;
List parameters = getProcedure().getCallParameters();
for (int i = 0; i < parameters.size(); i++) {
ProcedureParameter parameter = (ProcedureParameter) parameters.get(i);
if (!parameter.isOutParam()) {
continue;
}
if (result == null) {
result = new HashMap();
}
ColumnDescriptor descriptor = new ColumnDescriptor(parameter);
ExtendedType type = getAdapter().getExtendedTypes().getRegisteredType(
descriptor.getJavaClass());
Object val = type.materializeObject(statement, i + 1, descriptor
.getJdbcType());
result.put(descriptor.getLabel(), val);
}
if (result != null && !result.isEmpty()) {
// treat out parameters as a separate data row set
QueryLogger.logSelectCount(1, System.currentTimeMillis() - t1);
delegate.nextDataRows(query, Collections.singletonList(result));
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy