/*
* JasperReports - Free Java Reporting Library.
* Copyright (C) 2001 - 2009 Jaspersoft Corporation. All rights reserved.
* http://www.jaspersoft.com
*
* Unless you have purchased a commercial license agreement from Jaspersoft,
* the following license terms apply:
*
* This program is part of JasperReports.
*
* JasperReports is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* JasperReports is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with JasperReports. If not, see <http://www.gnu.org/licenses/>.
*/
package net.sf.jasperreports.engine.query;
import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;
import net.sf.jasperreports.engine.JRDataSource;
import net.sf.jasperreports.engine.JRDataset;
import net.sf.jasperreports.engine.JRException;
import net.sf.jasperreports.engine.JRParameter;
import net.sf.jasperreports.engine.JRRuntimeException;
import net.sf.jasperreports.engine.JRValueParameter;
import net.sf.jasperreports.engine.data.JRHibernateIterateDataSource;
import net.sf.jasperreports.engine.data.JRHibernateListDataSource;
import net.sf.jasperreports.engine.data.JRHibernateScrollDataSource;
import net.sf.jasperreports.engine.util.JRProperties;
import net.sf.jasperreports.engine.util.JRStringUtil;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.hibernate.Hibernate;
import org.hibernate.Query;
import org.hibernate.ScrollMode;
import org.hibernate.ScrollableResults;
import org.hibernate.Session;
import org.hibernate.type.Type;
/**
* HQL query executer that uses Hibernate 3.
*
* @author Lucian Chirita (lucianc@users.sourceforge.net)
* @version $Id: JRHibernateQueryExecuter.java 4101 2010-12-20 12:56:51Z teodord $
*/
public class JRHibernateQueryExecuter extends JRAbstractQueryExecuter
{
private static final Log log = LogFactory.getLog(JRHibernateQueryExecuter.class);
private static final Map hibernateTypeMap;
static
{
hibernateTypeMap = new HashMap();
hibernateTypeMap.put(Boolean.class, Hibernate.BOOLEAN);
hibernateTypeMap.put(Byte.class, Hibernate.BYTE);
hibernateTypeMap.put(Double.class, Hibernate.DOUBLE);
hibernateTypeMap.put(Float.class, Hibernate.FLOAT);
hibernateTypeMap.put(Integer.class, Hibernate.INTEGER);
hibernateTypeMap.put(Long.class, Hibernate.LONG);
hibernateTypeMap.put(Short.class, Hibernate.SHORT);
hibernateTypeMap.put(java.math.BigDecimal.class, Hibernate.BIG_DECIMAL);
hibernateTypeMap.put(java.math.BigInteger.class, Hibernate.BIG_INTEGER);
hibernateTypeMap.put(Character.class, Hibernate.CHARACTER);
hibernateTypeMap.put(String.class, Hibernate.STRING);
hibernateTypeMap.put(java.util.Date.class, Hibernate.DATE);
hibernateTypeMap.put(java.sql.Timestamp.class, Hibernate.TIMESTAMP);
hibernateTypeMap.put(java.sql.Time.class, Hibernate.TIME);
}
private final Integer reportMaxCount;
private Session session;
private Query query;
private boolean queryRunning;
private ScrollableResults scrollableResults;
private boolean isClearCache;
public JRHibernateQueryExecuter(JRDataset dataset, Map parameters)
{
super(dataset, parameters);
session = (Session) getParameterValue(JRHibernateQueryExecuterFactory.PARAMETER_HIBERNATE_SESSION);
reportMaxCount = (Integer) getParameterValue(JRParameter.REPORT_MAX_COUNT);
isClearCache = JRProperties.getBooleanProperty(dataset,
JRHibernateQueryExecuterFactory.PROPERTY_HIBERNATE_CLEAR_CACHE,
false);
if (session == null)
{
log.warn("The supplied org.hibernate.Session object is null.");
}
parseQuery();
}
/**
* Creates an instance of {@link JRHibernateListDataSource JRHibernateListDataSource},
* {@link JRHibernateIterateDataSource JRHibernateIterateDataSource} or
* {@link JRHibernateScrollDataSource JRHibernateScrollDataSource}, depending on the
*/
public JRDataSource createDatasource() throws JRException
{
JRDataSource datasource = null;
String queryString = getQueryString();
if (session != null && queryString != null && queryString.trim().length() > 0)
{
createQuery(queryString);
datasource = createResultDatasource();
}
return datasource;
}
/**
* Creates a data source out of the query result.
*
* @return the data source
*/
protected JRDataSource createResultDatasource()
{
JRDataSource resDatasource;
String runType = JRProperties.getProperty(dataset,
JRHibernateQueryExecuterFactory.PROPERTY_HIBERNATE_QUERY_RUN_TYPE);
boolean useFieldDescriptions = JRProperties.getBooleanProperty(dataset,
JRHibernateQueryExecuterFactory.PROPERTY_HIBERNATE_FIELD_MAPPING_DESCRIPTIONS,
true);
if (runType == null || runType.equals(JRHibernateQueryExecuterFactory.VALUE_HIBERNATE_QUERY_RUN_TYPE_LIST))
{
try
{
int pageSize = JRProperties.getIntegerProperty(dataset,
JRHibernateQueryExecuterFactory.PROPERTY_HIBERNATE_QUERY_LIST_PAGE_SIZE,
0);
resDatasource = new JRHibernateListDataSource(this, useFieldDescriptions, pageSize);
}
catch (NumberFormatException e)
{
throw new JRRuntimeException("The " + JRHibernateQueryExecuterFactory.PROPERTY_HIBERNATE_QUERY_LIST_PAGE_SIZE +
" property must be numerical.", e);
}
}
else if (runType.equals(JRHibernateQueryExecuterFactory.VALUE_HIBERNATE_QUERY_RUN_TYPE_ITERATE))
{
resDatasource = new JRHibernateIterateDataSource(this, useFieldDescriptions);
}
else if (runType.equals(JRHibernateQueryExecuterFactory.VALUE_HIBERNATE_QUERY_RUN_TYPE_SCROLL))
{
resDatasource = new JRHibernateScrollDataSource(this, useFieldDescriptions);
}
else
{
throw new JRRuntimeException("Unknown value for the " + JRHibernateQueryExecuterFactory.PROPERTY_HIBERNATE_QUERY_RUN_TYPE +
" property. Possible values are " + JRHibernateQueryExecuterFactory.VALUE_HIBERNATE_QUERY_RUN_TYPE_LIST + ", " +
JRHibernateQueryExecuterFactory.VALUE_HIBERNATE_QUERY_RUN_TYPE_ITERATE + " and " +
JRHibernateQueryExecuterFactory.VALUE_HIBERNATE_QUERY_RUN_TYPE_SCROLL + ".");
}
return resDatasource;
}
/**
* Creates the Hibernate query object.
* <p/>
* If the value of the {@link JRHibernateQueryExecuterFactory#PARAMETER_HIBERNATE_FILTER_COLLECTION PARAMETER_HIBERNATE_FILTER_COLLECTION}
* is not null, then a filter query is created using the value of the parameter as the collection.
*
* @param queryString the query string
*/
protected synchronized void createQuery(String queryString)
{
if (log.isDebugEnabled())
{
log.debug("HQL query: " + queryString);
}
Object filterCollection = getParameterValue(JRHibernateQueryExecuterFactory.PARAMETER_HIBERNATE_FILTER_COLLECTION);
if (filterCollection == null)
{
query = session.createQuery(queryString);
}
else
{
query = session.createFilter(filterCollection, queryString);
}
query.setReadOnly(true);
int fetchSize = JRProperties.getIntegerProperty(dataset,
JRJdbcQueryExecuterFactory.PROPERTY_JDBC_FETCH_SIZE,
0);
if (fetchSize != 0)
{
query.setFetchSize(fetchSize);
}
setParameters();
}
/**
* Binds values for all the query parameters.
*/
protected void setParameters()
{
List parameterNames = getCollectedParameterNames();
if (!parameterNames.isEmpty())
{
Set namesSet = new HashSet();
for (Iterator iter = parameterNames.iterator(); iter.hasNext();)
{
String parameterName = (String) iter.next();
if (namesSet.add(parameterName))
{
JRValueParameter parameter = getValueParameter(parameterName);
setParameter(parameter);
}
}
}
}
/**
* Binds a parameter value to a query parameter.
*
* @param parameter the report parameter
*/
protected void setParameter(JRValueParameter parameter)
{
String hqlParamName = getHqlParameterName(parameter.getName());
Class clazz = parameter.getValueClass();
Object parameterValue = parameter.getValue();
if (log.isDebugEnabled())
{
log.debug("Parameter " + hqlParamName + " of type " + clazz.getName() + ": " + parameterValue);
}
Type type = (Type) hibernateTypeMap.get(clazz);
if (type != null)
{
query.setParameter(hqlParamName, parameterValue, type);
}
else if (Collection.class.isAssignableFrom(clazz))
{
query.setParameterList(hqlParamName, (Collection) parameterValue);
}
else
{
if (session.getSessionFactory().getClassMetadata(clazz) != null) //param type is a hibernate mapped entity
{
query.setEntity(hqlParamName, parameterValue);
}
else //let hibernate try to guess the type
{
query.setParameter(hqlParamName, parameterValue);
}
}
}
/**
* Closes the scrollable result when <em>scroll</em> execution type is used.
*/
public synchronized void close()
{
closeScrollableResults();
query = null;
}
/**
* Closes the scrollable results of the query.
*/
public void closeScrollableResults()
{
if (scrollableResults != null)
{
try
{
scrollableResults.close();
}
finally
{
scrollableResults = null;
}
}
}
public synchronized boolean cancelQuery() throws JRException
{
if (queryRunning)
{
session.cancelQuery();
return true;
}
return false;
}
protected String getParameterReplacement(String parameterName)
{
return ':' + getHqlParameterName(parameterName);
}
protected String getHqlParameterName(String parameterName)
{
return '_' + JRStringUtil.getJavaIdentifier(parameterName);
}
/**
* Returns the return types of the HQL query.
*
* @return the return types of the HQL query
*/
public Type[] getReturnTypes()
{
return query.getReturnTypes();
}
/**
* Returns the aliases of the HQL query.
*
* @return the aliases of the HQL query
*/
public String[] getReturnAliases()
{
return query.getReturnAliases();
}
/**
* Returns the dataset for which the query executer has been created.
*
* @return the dataset for which the query executer has been created
*/
public JRDataset getDataset()
{
return dataset;
}
/**
* Runs the query by calling <code>org.hibernate.Query.list()</code>.
* <p/>
* All the result rows are returned.
*
* @return the result of the query as a list
*/
public List list()
{
setMaxCount();
setQueryRunning(true);
try
{
return query.list();
}
finally
{
setQueryRunning(false);
}
}
protected synchronized void setQueryRunning(boolean queryRunning)
{
this.queryRunning = queryRunning;
}
private void setMaxCount()
{
if (reportMaxCount != null)
{
query.setMaxResults(reportMaxCount.intValue());
}
}
/**
* Returns a page of the query results by calling <code>org.hibernate.Query.iterate()</code>.
*
* @param firstIndex the index of the first row to return
* @param resultCount the number of rows to return
* @return result row list
*/
public List list(int firstIndex, int resultCount)
{
if (reportMaxCount != null && firstIndex + resultCount > reportMaxCount.intValue())
{
resultCount = reportMaxCount.intValue() - firstIndex;
}
query.setFirstResult(firstIndex);
query.setMaxResults(resultCount);
if (isClearCache)
{
clearCache();
}
return query.list();
}
/**
* Runs the query by calling <code>org.hibernate.Query.iterate()</code>.
*
* @return query iterator
*/
public Iterator iterate()
{
setMaxCount();
setQueryRunning(true);
try
{
return query.iterate();
}
finally
{
setQueryRunning(false);
}
}
/**
* Runs the query by calling <code>org.hibernate.Query.scroll()</code>.
*
* @return scrollable results of the query
*/
public ScrollableResults scroll()
{
setMaxCount();
setQueryRunning(true);
try
{
scrollableResults = query.scroll(ScrollMode.FORWARD_ONLY);
}
finally
{
setQueryRunning(false);
}
return scrollableResults;
}
public void clearCache()
{
session.flush();
session.clear();
}
}