org.emonocot.persistence.dao.hibernate.ResourceDaoImpl.java Source code

Introduction

Here is the source code for org.emonocot.persistence.dao.hibernate.ResourceDaoImpl.java

Source

/*
 * This is eMonocot, a global online biodiversity information resource.
 *
 * Copyright  20112015 The Board of Trustees of the Royal Botanic Gardens, Kew and The University of Oxford
 *
 * eMonocot is free software: you can redistribute it and/or modify it under the terms of the
 * GNU Affero General Public License as published by the Free Software Foundation, either version 3
 * of the License, or (at your option) any later version.
 *
 * eMonocot 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 Affero General Public License for more details.
 *
 * The complete text of the GNU Affero General Public License is in the source repository as the file
 * COPYING.  It is also available from <http://www.gnu.org/licenses/>.
 */
package org.emonocot.persistence.dao.hibernate;

import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.emonocot.model.hibernate.Fetch;
import org.emonocot.model.registry.Resource;
import org.emonocot.persistence.dao.ResourceDao;
import org.hibernate.Criteria;
import org.hibernate.FetchMode;
import org.hibernate.criterion.Projections;
import org.hibernate.criterion.Property;
import org.hibernate.criterion.Restrictions;
import org.joda.time.DateTime;
import org.springframework.batch.core.BatchStatus;
import org.springframework.stereotype.Repository;

/**
 *
 * @author ben
 *
 */
@Repository
public class ResourceDaoImpl extends SearchableDaoImpl<Resource> implements ResourceDao {

    /**
     *
     */
    private static Map<String, Fetch[]> FETCH_PROFILES;

    static {
        FETCH_PROFILES = new HashMap<String, Fetch[]>();
        FETCH_PROFILES.put("job-with-source", new Fetch[] { new Fetch("organisation", FetchMode.JOIN),
                new Fetch("parameters", FetchMode.SELECT) });
    }

    /**
     *
     */
    public ResourceDaoImpl() {
        super(Resource.class);
    }

    /**
     * @param profile Set the profile name
     * @return the fetch profile
     */
    @Override
    public final Fetch[] getProfile(final String profile) {
        return ResourceDaoImpl.FETCH_PROFILES.get(profile);
    }

    /**
     * @param sourceId Set the source identifier
     * @return the total number of jobs for a given source
     */
    public final Long count(final String sourceId) {
        Criteria criteria = getSession().createCriteria(type);
        criteria.createAlias("source", "src").add(Restrictions.eq("src.identifier", sourceId));
        criteria.setProjection(Projections.rowCount());
        return (Long) criteria.uniqueResult();
    }

    /**
     * @param sourceId Set the source identifier
     * @param page Set the offset (in size chunks, 0-based), optional
     * @param size Set the page size
     * @return A list of jobs
     */
    public final List<Resource> list(final String sourceId, final Integer page, final Integer size) {
        Criteria criteria = getSession().createCriteria(type);
        criteria.createAlias("organisation", "org").add(Restrictions.eq("org.identifier", sourceId));

        if (size != null) {
            criteria.setMaxResults(size);
            if (page != null) {
                criteria.setFirstResult(page * size);
            }
        }
        return (List<Resource>) criteria.list();
    }

    /**
     * @param id Set the job id
     * @return the job
     */
    public final Resource findByJobId(final Long id) {
        Criteria criteria = getSession().createCriteria(type);
        criteria.add(Restrictions.eq("jobId", id));
        return (Resource) criteria.uniqueResult();
    }

    @Override
    public boolean isHarvesting() {
        Criteria criteria = getSession().createCriteria(type);
        criteria.add(Restrictions.isNotNull("resourceType"));
        criteria.add(Restrictions.not(Restrictions.in("status", Arrays.asList(new BatchStatus[] {
                BatchStatus.COMPLETED, BatchStatus.FAILED, BatchStatus.ABANDONED, BatchStatus.STOPPED }))));
        criteria.setProjection(Projections.rowCount());
        return (Long) criteria.uniqueResult() > 0;
    }

    @Override
    public List<Resource> listResourcesToHarvest(Integer limit, DateTime now, String fetch) {
        Criteria criteria = getSession().createCriteria(type);
        criteria.add(Restrictions.isNotNull("resourceType"));
        criteria.add(Restrictions.in("status", Arrays.asList(new BatchStatus[] { BatchStatus.COMPLETED,
                BatchStatus.FAILED, BatchStatus.ABANDONED, BatchStatus.STOPPED })));
        criteria.add(Restrictions.eq("scheduled", Boolean.TRUE));
        criteria.add(Restrictions.disjunction().add(Restrictions.lt("nextAvailableDate", now))
                .add(Restrictions.isNull("nextAvailableDate")));

        if (limit != null) {
            criteria.setMaxResults(limit);
        }
        enableProfilePreQuery(criteria, fetch);
        criteria.addOrder(Property.forName("nextAvailableDate").asc());
        List<Resource> result = (List<Resource>) criteria.list();
        for (Resource t : result) {
            enableProfilePostQuery(t, fetch);
        }
        return result;
    }

    @Override
    protected boolean isSearchableObject() {
        return false;
    }

    @Override
    public Resource findResourceByUri(String identifier) {
        Criteria criteria = getSession().createCriteria(type);
        criteria.add(Restrictions.eq("uri", identifier));
        return (Resource) criteria.uniqueResult();
    }

}