/******************************************************************************* | |
* Copyright (c) 1998, 2013 Oracle and/or its affiliates. All rights reserved. | |
* This program and the accompanying materials are made available under the | |
* terms of the Eclipse Public License v1.0 and Eclipse Distribution License v. 1.0 | |
* which accompanies this distribution. | |
* The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html | |
* and the Eclipse Distribution License is available at | |
* http://www.eclipse.org/org/documents/edl-v10.php. | |
* | |
* Contributors: | |
* Oracle - initial API and implementation from Oracle TopLink | |
******************************************************************************/ | |
package org.eclipse.persistence.testing.tests.feature; | |
import java.util.ArrayList; | |
import java.util.List; | |
import org.eclipse.persistence.descriptors.DescriptorQueryManager; | |
import org.eclipse.persistence.exceptions.DatabaseException; | |
import org.eclipse.persistence.queries.InsertObjectQuery; | |
import org.eclipse.persistence.sessions.UnitOfWork; | |
import org.eclipse.persistence.testing.models.employee.domain.Address; | |
import org.eclipse.persistence.testing.models.employee.domain.Employee; | |
/** | |
* Bug 214910: Add query timeout support to batched update queries (Oracle DB 9.0.1+)</p> | |
* Test the query timeout feature in batch queries. | |
* For data queries , a queryTimeout on the largest DatabaseQuery of the batch will be used. | |
* For object queries, a queryTimeout on the largest DescriptorQueryManager (parent) or DatabaseQuery | |
* of the batch will be used. | |
*/ | |
public abstract class QueryTimeoutBatchDescriptorQueryManagerTest extends QueryTimeoutBatchTestCase { | |
protected int getParentQueryTimeout() { return 2; } | |
protected int getChildQueryTimeout() { return -1; } | |
protected String getQuerySQLPostfix() { | |
return ", SUM(e.address_id) as version from address e, address b, address b, address c, address c, address c, address b"; | |
} | |
/** | |
* Test that the queryTimeout passed by the client to the QueryManager is | |
* picked up by the BatchWritingMechanism and set on the Statement. | |
*/ | |
public void test() { | |
UnitOfWork uow = null; | |
try { | |
uow = setupPlatform(); | |
initializeDatabase(uow); | |
// Get new UOW | |
uow = getSession().acquireUnitOfWork(); | |
registerObjects(uow); | |
verifyBefore(uow); | |
uow.commit(); | |
} catch (Exception e) { | |
/** Throws | |
* Internal Exception: java.sql.SQLException: ORA-01013: user requested cancel of current operation | |
*/ | |
//e.printStackTrace(); | |
//System.err.print(e.getMessage()); | |
if (e instanceof DatabaseException) { | |
limitExceeded = true; | |
vendorErrorCodeEncountered = ((DatabaseException)e).getDatabaseErrorCode(); | |
} | |
// Release transaction mutex | |
//((AbstractSession)uow).rollbackTransaction(); | |
} finally { | |
resetPlatform(); | |
} | |
} | |
/** | |
* Iterate and register a number of objects in the unitOfWork | |
* @param uow | |
*/ | |
protected List<Employee> registerObjects(UnitOfWork uow) { | |
List<Employee> objectListForEditing = new ArrayList<Employee>(); | |
Address address = null; | |
Employee employee = null; | |
for (int i = 0; i < getNumberOfInserts(); i++) { | |
address = new Address(); | |
address.setCity(new StringBuffer("city").append(i).toString()); | |
address.setProvince(new StringBuffer("province").append(i).toString()); | |
employee = new Employee(); | |
employee.setFirstName(new StringBuffer("first").append(i).toString()); | |
employee.setLastName(new StringBuffer("last").append(i).toString()); | |
//employee.setId(new java.math.BigDecimal(i)); | |
employee.setAddress(address); | |
uow.registerObject(employee); | |
// Use the return clone for editing | |
objectListForEditing.add(employee); | |
// Test framework callback | |
// Set the timeout at the descriptor level for each object | |
setDescriptorLevelQueryTimeout(uow.getDescriptor(employee).getDescriptorQueryManager()); | |
setDescriptorLevelQueryTimeout(uow.getDescriptor(address).getDescriptorQueryManager()); | |
// Test framework callback | |
// Set the timeout on each query object (already defined) | |
setQueryLevelQueryTimeout(uow, employee); | |
setQueryLevelQueryTimeout(uow, address); | |
// Add custom insertQuery | |
InsertObjectQuery query = new InsertObjectQuery(); | |
//queryCount++; | |
// insert into employee (emp_id, version) SELECT 40003, SUM(e.address_id) as version from address e, address b, address b, address c, address c; | |
StringBuffer sBuffer = new StringBuffer(getQuerySQLPrefix()); | |
sBuffer.append(getCurrentIDSequence() + i); | |
//sBuffer.append(i); | |
sBuffer.append(getQuerySQLPostfix()); | |
query.setSQLString(sBuffer.toString()); | |
// Override parent | |
query.setQueryTimeout(getChildQueryTimeout()); | |
// We don't set the f_name parameter so Ignore: WARNING: Missing Query parameter for named argument: 1 null will be substituted. | |
uow.getDescriptor(employee).getDescriptorQueryManager().setInsertQuery(query); | |
} | |
return objectListForEditing; | |
} | |
/** | |
* This is a callback from the object loop in registerObjects that allows the test | |
* to set a timeout globally on the DescriptorQueryManager | |
*/ | |
public void setDescriptorLevelQueryTimeout(DescriptorQueryManager queryManager) { | |
// Set timeout globally on the root object using the last call appended | |
queryManager.setQueryTimeout(getParentQueryTimeout()); | |
} | |
/** | |
* This is a callback from the object loop in registerObjects that allows the test | |
* to set a timeout on individual queries | |
*/ | |
public void setQueryLevelQueryTimeout(UnitOfWork uow, Object object) { | |
} | |
} |