ru.apertum.qsystem.server.model.calendar.CalendarTableModel.java Source code

Introduction

Here is the source code for ru.apertum.qsystem.server.model.calendar.CalendarTableModel.java

Source

/*
 *  Copyright (C) 2010 {Apertum}Projects. web: www.apertum.ru email: info@apertum.ru
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU 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 General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package ru.apertum.qsystem.server.model.calendar;

import java.util.ArrayList;
import java.util.Date;
import java.util.GregorianCalendar;
import java.util.LinkedList;
import java.util.List;
import javax.swing.table.AbstractTableModel;
import org.hibernate.criterion.DetachedCriteria;
import org.hibernate.criterion.Property;
import org.springframework.transaction.TransactionDefinition;
import org.springframework.transaction.TransactionStatus;
import org.springframework.transaction.support.DefaultTransactionDefinition;
import ru.apertum.qsystem.common.QLog;
import ru.apertum.qsystem.common.exceptions.ClientException;
import ru.apertum.qsystem.server.Spring;

/**
 *  ? ? ? ?
 * @author Evgeniy Egorov
 */
public class CalendarTableModel extends AbstractTableModel {

    private final List<FreeDay> days;
    private List<FreeDay> days_del;

    public CalendarTableModel(long calcId) {
        QLog.l().logger().debug("  ? ?");
        this.calcId = calcId;
        days = getFreeDays(calcId);
        days_del = new ArrayList<>(days);
    }

    /**
     *    ?? 
     */
    final private long calcId;

    /**
     *     .
     * @param calcId id ?
     * @return ??    ?
     */
    public synchronized static LinkedList<FreeDay> getFreeDays(final Long calcId) {
        final DetachedCriteria criteria = DetachedCriteria.forClass(FreeDay.class);
        criteria.add(Property.forName("calendarId").eq(calcId));
        return new LinkedList<>(Spring.getInstance().getHt().findByCriteria(criteria));
    }

    /**
     * ? . ? ?  ?  
     * @param date
     * @param noInvert true -    ??  
     * @return   ??   ?/ true = ??
     */
    public boolean addDay(Date date, boolean noInvert) {
        final FreeDay day = isFreeDate(date);
        if (day != null) {
            if (noInvert) {
                return true;
            } else {
                days.remove(day);
            }
            return false;
        } else {
            days.add(new FreeDay(date, calcId));
            return true;
        }
    }

    /**
     * ?     
     * @param date
     * @return
     */
    public FreeDay isFreeDate(Date date) {
        for (FreeDay day : days) {
            if (day.equals(date)) {
                return day;
            }
        }
        return null;
    }

    @Override
    public int getRowCount() {
        return 12;
    }

    @Override
    public int getColumnCount() {
        return 32;
    }

    @Override
    public Object getValueAt(int rowIndex, int columnIndex) {
        return "X";
    }

    @Override
    public Class<?> getColumnClass(int columnIndex) {
        return columnIndex == 0 ? super.getColumnClass(columnIndex) : FreeDay.class;
    }

    @Override
    public String getColumnName(int column) {
        return column == 0 ? "" : Integer.toString(column);
    }

    /**
     * ?    
     * @param year
     */
    public void dropCalendar(int year) {
        QLog.l().logger().debug("? ");
        final ArrayList<FreeDay> del = new ArrayList<>();
        final GregorianCalendar gc = new GregorianCalendar();
        for (FreeDay freeDay : days) {
            gc.setTime(freeDay.getDate());
            if (gc.get(GregorianCalendar.YEAR) == year) {
                del.add(freeDay);
            }
        }
        days.removeAll(del);
        fireTableDataChanged();
    }

    /**
     *  ? ? 
     * @param year
     */
    public void checkSaturday(int year) {
        QLog.l().logger().debug(" ? ?");
        final GregorianCalendar gc = new GregorianCalendar();
        gc.set(GregorianCalendar.YEAR, year);
        final int ye = year % 4 == 0 ? 366 : 365;
        for (int d = 1; d <= ye; d++) {
            gc.set(GregorianCalendar.DAY_OF_YEAR, d);
            if (gc.get(GregorianCalendar.DAY_OF_WEEK) == 7) {
                addDay(gc.getTime(), true);
            }
        }
        fireTableDataChanged();
    }

    /**
     *  ? ??? 
     * @param year
     */
    public void checkSunday(int year) {
        QLog.l().logger().debug(" ? ???");
        final GregorianCalendar gc = new GregorianCalendar();
        gc.set(GregorianCalendar.YEAR, year);
        final int ye = year % 4 == 0 ? 366 : 365;
        for (int d = 1; d <= ye; d++) {
            gc.set(GregorianCalendar.DAY_OF_YEAR, d);
            if (gc.get(GregorianCalendar.DAY_OF_WEEK) == 1) {
                addDay(gc.getTime(), true);
            }
        }
        fireTableDataChanged();
    }

    /**
     *  .
     */
    public void save() {
        QLog.l().logger().info("?  ID = " + calcId);

        final DefaultTransactionDefinition def = new DefaultTransactionDefinition();
        def.setName("SomeTxName");
        def.setPropagationBehavior(TransactionDefinition.PROPAGATION_REQUIRED);
        TransactionStatus status = Spring.getInstance().getTxManager().getTransaction(def);
        try {
            final LinkedList<FreeDay> dels = new LinkedList<>();
            for (FreeDay bad : days_del) {
                boolean f = true;
                for (FreeDay good : days) {
                    if (good.equals(bad)) {
                        f = false;
                    }
                }
                if (f) {
                    dels.add(bad);
                }
            }
            Spring.getInstance().getHt().deleteAll(dels);
            Spring.getInstance().getHt().saveOrUpdateAll(days);
        } catch (Exception ex) {
            Spring.getInstance().getTxManager().rollback(status);
            throw new ClientException(
                    " ?  ?   (JDBC).\n    ,  , ? ? ? ?,     ?.\n ? (Ctrl + S)    ? ? ? ?.\n\n["
                            + ex.getLocalizedMessage() + "]\n(" + ex.toString() + ")");
        }
        Spring.getInstance().getTxManager().commit(status);
        QLog.l().logger().debug("  ");
        //     
        days_del = new ArrayList<>(days);
    }

    /**
     *   ?? ?
     * @return
     */
    public boolean isSaved() {
        for (FreeDay day : days) {
            if (day.getId() == null) {
                return false;
            }
        }
        if (days_del.size() != days.size()) {
            return false;
        }
        return true;
    }
}