org.openmrs.module.webservices.rest.web.api.impl.RestHelperServiceImpl.java Source code

Introduction

Here is the source code for org.openmrs.module.webservices.rest.web.api.impl.RestHelperServiceImpl.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.module.webservices.rest.web.api.impl;

import java.io.Serializable;
import java.lang.reflect.Method;
import java.util.List;

import org.hibernate.Criteria;
import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.hibernate.criterion.Restrictions;
import org.openmrs.api.impl.BaseOpenmrsService;
import org.openmrs.module.webservices.rest.web.api.RestHelperService;
import org.springframework.transaction.annotation.Transactional;

/**
 * REST helper service, which must not be used outside of the REST module.
 */
public class RestHelperServiceImpl extends BaseOpenmrsService implements RestHelperService {

    SessionFactory sessionFactory;

    Method method;

    public void setSessionFactory(SessionFactory sessionFactory) {
        this.sessionFactory = sessionFactory;
    }

    /**
     * @see org.openmrs.module.webservices.rest.web.api.RestHelperService#getConceptMapByUuid(java.lang.String)
     */
    @Override
    @Transactional(readOnly = true)
    public <T> T getObjectByUuid(Class<? extends T> type, String uuid) {
        return type.cast(getSession().createCriteria(type).add(Restrictions.eq("uuid", uuid)).uniqueResult());
    }

    private Session getSession() {
        if (method == null) {
            try {
                return sessionFactory.getCurrentSession();
            } catch (NoSuchMethodError error) {
                //Supports Hibernate 3 by casting org.hibernate.classic.Session to org.hibernate.Session
                try {
                    method = sessionFactory.getClass().getMethod("getCurrentSession");
                    return (Session) method.invoke(sessionFactory);
                } catch (Exception e) {
                    throw new IllegalStateException(e);
                }
            }
        } else {
            try {
                return (Session) method.invoke(sessionFactory);
            } catch (Exception e) {
                throw new IllegalStateException(e);
            }
        }
    }

    /**
     * @see org.openmrs.module.webservices.rest.web.api.RestHelperService#getObjectById(java.lang.Class,
     *      java.io.Serializable)
     */
    @Override
    public <T> T getObjectById(Class<? extends T> type, Serializable id) {
        return type.cast(getSession().get(type, id));
    }

    /**
     * @see org.openmrs.module.webservices.rest.web.api.RestHelperService#getObjectByField(java.lang.Class,
     *      java.lang.String, java.lang.Object)
     */
    @SuppressWarnings("unchecked")
    @Override
    public <T> List<T> getObjectsByFields(Class<? extends T> type, Field... fields) {
        Criteria criteria = getSession().createCriteria(type);
        for (Field field : fields) {
            if (field != null) {
                criteria.add(Restrictions.eq(field.getName(), field.getValue()));
            }
        }
        return criteria.list();
    }
}