org.squashtest.tm.internal.domain.report.common.hibernate.BelowDateCriterion.java Source code

Introduction

Here is the source code for org.squashtest.tm.internal.domain.report.common.hibernate.BelowDateCriterion.java

Source

/**
 *     This file is part of the Squashtest platform.
 *     Copyright (C) 2010 - 2016 Henix, henix.fr
 *
 *     See the NOTICE file distributed with this work for additional
 *     information regarding copyright ownership.
 *
 *     This is free software: you can redistribute it and/or modify
 *     it under the terms of the GNU Lesser General Public License as published by
 *     the Free Software Foundation, either version 3 of the License, or
 *     (at your option) any later version.
 *
 *     this software 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 Lesser General Public License for more details.
 *
 *     You should have received a copy of the GNU Lesser General Public License
 *     along with this software.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.squashtest.tm.internal.domain.report.common.hibernate;

import java.text.ParseException;
import java.util.Calendar;
import java.util.Date;
import java.util.GregorianCalendar;

import org.hibernate.criterion.Criterion;
import org.hibernate.criterion.Restrictions;
import org.squashtest.tm.internal.domain.report.query.QueryOperator;
import org.squashtest.tm.internal.domain.report.query.hibernate.ReportCriterion;

/*
 * this ReportCriterion is exactly wysiwyg
 *
 */
public class BelowDateCriterion extends ReportCriterion {

    public BelowDateCriterion() {

        setOperator(QueryOperator.COMPARATOR_LT);
        setParamClass(Date.class);
    }

    public BelowDateCriterion(String criterionName, String attributePath) {
        this();
        setCriterionName(criterionName);
        setAttributePath(attributePath);

    }

    private Date makeDate() throws ParseException {
        Object[] values = getParameters();
        if (values.length != 1) {
            throw new IllegalArgumentException(
                    "Criterion of type " + this.getClass().getSimpleName() + " cannot have more than one argument");
        }
        Date date = (Date) values[0];

        Calendar calendar = GregorianCalendar.getInstance();

        calendar.setTime(date);

        calendar.add(Calendar.DAY_OF_MONTH, 1);

        return calendar.getTime();

    }

    @Override
    public Criterion makeCriterion() {
        try {
            Criterion result = null;

            Date arg = makeDate();

            result = Restrictions.lt(getAttributePath(), arg);

            return result;
        } catch (Exception e) {
            return null;
        }
    }

}