org.openmrs.logic.db.hibernate.HibernateLogicPersonDAO.java Source code

Introduction

Here is the source code for org.openmrs.logic.db.hibernate.HibernateLogicPersonDAO.java

Source

/**
 * The contents of this file are subject to the OpenMRS Public License
 * Version 1.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://license.openmrs.org
 *
 * Software distributed under the License is distributed on an "AS IS"
 * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
 * License for the specific language governing rights and limitations
 * under the License.
 *
 * Copyright (C) OpenMRS, LLC.  All Rights Reserved.
 */
package org.openmrs.logic.db.hibernate;

import java.util.ArrayList;
import java.util.Calendar;
import java.util.Collection;
import java.util.Date;
import java.util.HashMap;
import java.util.List;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.hibernate.Criteria;
import org.hibernate.SessionFactory;
import org.hibernate.criterion.Criterion;
import org.hibernate.criterion.Restrictions;
import org.openmrs.Person;
import org.openmrs.logic.LogicCriteria;
import org.openmrs.logic.LogicException;
import org.openmrs.logic.LogicExpression;
import org.openmrs.logic.LogicTransform;
import org.openmrs.logic.db.LogicPersonDAO;
import org.openmrs.logic.op.Operator;
import org.openmrs.logic.util.LogicExpressionToCriterion;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Repository;

/**
 * 
 */
@Repository
public class HibernateLogicPersonDAO extends LogicExpressionToCriterion implements LogicPersonDAO {

    static {
        map.put("GENDER", "gender");
        map.put("BIRTHDATE", "birthdate");
        map.put("BIRTHDATE ESTIMATED", "birthdateEstimated");
        map.put("DEAD", "dead");
        map.put("DEATH DATE", "deathDate");
        map.put("CAUSE OF DEATH", "causeOfDeath");
        map.put("GIVEN NAME", "name.givenName");
        map.put("MIDDLE NAME", "name.middleName");
        map.put("FAMILY NAME", "name.familyName");
        map.put("FAMILY NAME2", "name.familyName2");
    }

    protected final Log log = LogFactory.getLog(getClass());

    /**
     * Hibernate session factory
     */
    @Autowired
    private SessionFactory sessionFactory;

    /**
     * Set session factory
     * 
     * @param sessionFactory
     */
    public void setSessionFactory(SessionFactory sessionFactory) {
        this.sessionFactory = sessionFactory;
    }

    // Helper function, converts logic service's criteria into Hibernate's
    // criteria
    @SuppressWarnings("unchecked")
    private List<Person> logicToHibernate(LogicExpression expression, Collection<Integer> personIds) {
        Criteria criteria = sessionFactory.getCurrentSession().createCriteria(Person.class);
        criteria.createAlias("names", "name");

        Date indexDate = Calendar.getInstance().getTime();
        Operator transformOperator = null;
        LogicTransform transform = expression.getTransform();
        Integer numResults = null;

        if (transform != null) {
            transformOperator = transform.getTransformOperator();
            numResults = transform.getNumResults();
        }

        if (numResults == null) {
            numResults = 1;
        }
        // set the transform and evaluate the right criteria
        // if there is any
        if (transformOperator == Operator.DISTINCT) {
            criteria.setResultTransformer(Criteria.DISTINCT_ROOT_ENTITY);
        }
        Criterion c = this.getCriterion(expression, indexDate, criteria);
        if (c != null) {
            criteria.add(c);
        }
        List<Person> results = new ArrayList<Person>();

        criteria.add(Restrictions.in("personId", personIds));
        results.addAll(criteria.list());

        //return a single result per patient for these operators
        //I don't see an easy way to do this in hibernate so I am
        //doing some postprocessing
        if (transformOperator == Operator.FIRST || transformOperator == Operator.LAST) {
            HashMap<Integer, ArrayList<Person>> nResultMap = new HashMap<Integer, ArrayList<Person>>();

            for (Person currResult : results) {
                Integer currPersonId = currResult.getPersonId();
                ArrayList<Person> prevResults = nResultMap.get(currPersonId);
                if (prevResults == null) {
                    prevResults = new ArrayList<Person>();
                    nResultMap.put(currPersonId, prevResults);
                }

                if (prevResults.size() < numResults) {
                    prevResults.add(currResult);
                }
            }

            if (nResultMap.values().size() > 0) {
                results.clear();

                for (ArrayList<Person> currPatientPerson : nResultMap.values()) {
                    results.addAll(currPatientPerson);
                }
            }
        }
        return results;
    }

    /**
     * @throws LogicException
     * @see org.openmrs.api.db.PersonDAO#getPeople(String, Boolean)
     */
    public List<Person> getPersons(Collection<Integer> personIds, LogicCriteria logicCriteria)
            throws LogicException {
        return logicToHibernate(logicCriteria.getExpression(), personIds);
    }

}