_br-ode-1.X.svn/dao-jpa/src/main/java/org/apache/ode/dao/jpa/ProcessManagementDAOImpl.java

package org.apache.ode.dao.jpa;

import java.util.Collection;
import java.util.HashMap;
import java.util.Map;

import javax.persistence.EntityManager;
import javax.persistence.Query;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.ode.bpel.common.InstanceFilter;
import org.apache.ode.bpel.dao.BpelDAOConnection;
import org.apache.ode.bpel.dao.ProcessInstanceDAO;
import org.apache.ode.bpel.dao.ProcessManagementDAO;

public class ProcessManagementDAOImpl implements ProcessManagementDAO {
	private static final Log __log = LogFactory.getLog(ProcessManagementDAOImpl.class);

	private EntityManager em;
	
	public ProcessManagementDAOImpl(EntityManager em) {
		this.em = em;
	}
	
	public Object[] findFailedCountAndLastFailedDateForProcessId(BpelDAOConnection conn, String status, String processId) {
		Query query = em.createNamedQuery(ProcessInstanceDAOImpl.COUNT_FAILED_INSTANCES_BY_STATUS_AND_PROCESS_ID);
		query.setParameter("states", new InstanceFilter("status=" + status).convertFilterState());
		query.setParameter("processId", processId);
		
		return (Object[])query.getSingleResult();
	}
	
	public void prefetchActivityFailureCounts(Collection<ProcessInstanceDAO> instances) {
		if(__log.isTraceEnabled()) __log.trace("Prefetching activity failure counts for " + instances.size() + " instances.");
		
		if( instances.isEmpty() ) return;
		
		Query query = em.createNamedQuery(ActivityRecoveryDAOImpl.COUNT_ACTIVITY_RECOVERIES_BY_INSTANCES);
		query.setParameter("instances", instances);
		
		Map<Long, Long> countsByInstanceId = new HashMap<Long, Long>();
		for( Object instanceIdAndCount : query.getResultList() ) {
			Object instanceId = ((Object[])instanceIdAndCount)[0];
			Object count = ((Object[])instanceIdAndCount)[0];
			countsByInstanceId.put((Long)instanceId, (Long)count);
		}
		
		for( ProcessInstanceDAO instance : instances ) {
			Long count = countsByInstanceId.get(instance.getInstanceId());
			if( count != null ) {
				((ProcessInstanceDAOImpl)instance).setActivityFailureCount(count.intValue());
			}
		}
	}
}