org.libreplan.business.workreports.daos.WorkReportDAO.java Source code

Introduction

Here is the source code for org.libreplan.business.workreports.daos.WorkReportDAO.java

Source

/*
 * This file is part of LibrePlan
 *
 * Copyright (C) 2009-2010 Fundacin para o Fomento da Calidade Industrial e
 *                         Desenvolvemento Tecnolxico de Galicia
 * Copyright (C) 2010-2011 Igalia, S.L.
 *
 * This program 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.
 *
 * This program 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.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.libreplan.business.workreports.daos;

import java.util.ArrayList;
import java.util.Calendar;
import java.util.Date;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import org.hibernate.Criteria;
import org.hibernate.NonUniqueResultException;
import org.hibernate.Query;
import org.hibernate.criterion.Restrictions;
import org.joda.time.LocalDate;
import org.libreplan.business.common.IAdHocTransactionService;
import org.libreplan.business.common.daos.IntegrationEntityDAO;
import org.libreplan.business.common.entities.PersonalTimesheetsPeriodicityEnum;
import org.libreplan.business.common.exceptions.InstanceNotFoundException;
import org.libreplan.business.orders.daos.IOrderDAO;
import org.libreplan.business.orders.entities.OrderElement;
import org.libreplan.business.resources.entities.Resource;
import org.libreplan.business.workreports.entities.PredefinedWorkReportTypes;
import org.libreplan.business.workreports.entities.WorkReport;
import org.libreplan.business.workreports.entities.WorkReportLine;
import org.libreplan.business.workreports.entities.WorkReportType;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Repository;

/**
 * Dao for {@link WorkReportDAO}.
 *
 * @author Diego Pino Garca <dpino@igalia.com>
 */
@Repository
@Scope(BeanDefinition.SCOPE_SINGLETON)
public class WorkReportDAO extends IntegrationEntityDAO<WorkReport> implements IWorkReportDAO {

    @Autowired
    private IAdHocTransactionService adHocTransactionService;

    @Autowired
    private IOrderDAO orderDAO;

    @Autowired
    private IWorkReportTypeDAO workReportTypeDAO;

    private final String WORK_REPORT_TYPE_COLUMN = "workReportType";

    @SuppressWarnings("unchecked")
    @Override
    public List<WorkReport> getAllByWorkReportType(WorkReportType workReportType) {
        return getSession().createCriteria(WorkReport.class)
                .add(Restrictions.eq(WORK_REPORT_TYPE_COLUMN, workReportType)).list();
    }

    @Override
    public List<WorkReport> getAll() {
        return list(WorkReport.class);
    }

    @Override
    public List<WorkReport> allWorkReportsWithAssociatedOrdersUnproxied() {
        forceOrdersUnproxied();
        return list(WorkReport.class);
    }

    private void forceOrdersUnproxied() {
        List<OrderElement> elements = adHocTransactionService
                .runOnAnotherReadOnlyTransaction(() -> getOrderElementsAssociatedWithWorkReports());

        orderDAO.loadOrdersAvoidingProxyFor(elements);
    }

    private List<OrderElement> getOrderElementsAssociatedWithWorkReports() {
        Set<OrderElement> result = new HashSet<>();

        result.addAll(elementsFrom(getSession().createQuery("select w.orderElement from WorkReport w")));

        result.addAll(elementsFrom(getSession()
                .createQuery("select line.orderElement from WorkReport w JOIN w.workReportLines line")));

        return new ArrayList<>(result);
    }

    @SuppressWarnings("unchecked")
    private List<OrderElement> elementsFrom(Query orderElementsQuery) {
        return orderElementsQuery.list();
    }

    @Override
    public int getFirstReportYear() {
        final String query = "select min(w.date) from WorkReportLine w";
        Date minDate = (Date) getSession().createQuery(query).uniqueResult();

        if (minDate != null) {
            Calendar calendar = Calendar.getInstance();
            calendar.setTime(minDate);

            return calendar.get(Calendar.YEAR);
        }
        return 0;
    }

    @Override
    public int getLastReportYear() {
        final String query = "select max(w.date) from WorkReportLine w";
        Date maxDate = (Date) getSession().createQuery(query).uniqueResult();

        if (maxDate != null) {
            Calendar calendar = Calendar.getInstance();
            calendar.setTime(maxDate);

            return calendar.get(Calendar.YEAR);
        }
        return 0;
    }

    @Override
    @SuppressWarnings("unchecked")
    public WorkReport getPersonalTimesheetWorkReport(Resource resource, LocalDate date,
            PersonalTimesheetsPeriodicityEnum periodicity) {

        Criteria criteria = getSession().createCriteria(WorkReport.class);
        criteria.add(Restrictions.eq(WORK_REPORT_TYPE_COLUMN, getPersonalTimesheetsWorkReportType()));
        List<WorkReport> personalTimesheets = criteria.add(Restrictions.eq("resource", resource)).list();

        LocalDate start = periodicity.getStart(date);
        LocalDate end = periodicity.getEnd(date);

        for (WorkReport workReport : personalTimesheets) {
            Set<WorkReportLine> workReportLines = workReport.getWorkReportLines();

            if (!workReportLines.isEmpty()) {
                LocalDate workReportDate = LocalDate.fromDateFields(workReportLines.iterator().next().getDate());

                if (workReportDate.compareTo(start) >= 0 && workReportDate.compareTo(end) <= 0) {
                    return workReport;
                }
            }
        }

        return null;
    }

    private WorkReportType getPersonalTimesheetsWorkReportType() {
        WorkReportType workReportType;
        try {
            workReportType = workReportTypeDAO
                    .findUniqueByName(PredefinedWorkReportTypes.PERSONAL_TIMESHEETS.getName());

        } catch (NonUniqueResultException | InstanceNotFoundException e) {
            throw new RuntimeException(e);
        }

        return workReportType;
    }

    @Override
    public boolean isAnyPersonalTimesheetAlreadySaved() {
        return getSession().createCriteria(WorkReport.class)
                .add(Restrictions.eq(WORK_REPORT_TYPE_COLUMN, getPersonalTimesheetsWorkReportType())).list()
                .isEmpty();
    }

    @Override
    @SuppressWarnings("unchecked")
    public List<WorkReport> findPersonalTimesheetsByResourceAndOrderElement(Resource resource) {
        return getSession().createCriteria(WorkReport.class)
                .add(Restrictions.eq(WORK_REPORT_TYPE_COLUMN, getPersonalTimesheetsWorkReportType()))
                .add(Restrictions.eq("resource", resource)).list();
    }

}