TrainingAttendeeDao.java

/////////////////////////////////////////////////////////////////////////////
//
// Project ProjectForge Community Edition
//         www.projectforge.org
//
// Copyright (C) 2001-2014 Kai Reinhard (k.reinhard@micromata.de)
//
// ProjectForge is dual-licensed.
//
// This community edition 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; version 3 of the License.
//
// This community edition 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 org.projectforge.plugins.skillmatrix;

import java.util.List;

import org.hibernate.criterion.Order;
import org.hibernate.criterion.Restrictions;
import org.projectforge.core.BaseDao;
import org.projectforge.core.BaseSearchFilter;
import org.projectforge.core.QueryFilter;
import org.projectforge.user.PFUserDO;
import org.projectforge.user.UserDao;
import org.projectforge.user.UserRightId;

/**
 * This is the base data access object class. Most functionality such as access checking, select, insert, update, save, delete etc. is
 * implemented by the super class.
 * @author Werner Feder (werner.feder@t-online.de)
 */
public class TrainingAttendeeDao extends BaseDao<TrainingAttendeeDO>
{

  public static final String UNIQUE_PLUGIN_ID = "PLUGIN_SKILL_MATRIX_TRAINING_ATTENDEE";

  public static final String I18N_KEY_SKILL_PREFIX = "plugins.skillmatrix.training";

  public static final UserRightId USER_RIGHT_ID = new UserRightId(UNIQUE_PLUGIN_ID, "plugin20", I18N_KEY_SKILL_PREFIX);

  private static final String[] ADDITIONAL_SEARCH_FIELDS = new String[] { "trainingAttendee.firstname", "trainingAttendee.lastname", "trainingAttendee.username", "training.title",
    "training.skill.title", "rating", "certificate" };

  private TrainingDao trainingDao;
  private UserDao userDao;

  public TrainingAttendeeDao()
  {
    super(TrainingAttendeeDO.class);
    userRightId = USER_RIGHT_ID;
  }

  @Override
  protected String[] getAdditionalSearchFields()
  {
    return ADDITIONAL_SEARCH_FIELDS;
  }

  @Override
  public TrainingAttendeeDO newInstance()
  {
    return new TrainingAttendeeDO();
  }

  public TrainingDao getTraingDao()
  {
    return trainingDao;
  }

  public TrainingAttendeeDao setTraingDao(final TrainingDao trainingDao)
  {
    this.trainingDao = trainingDao;
    return this;
  }

  public UserDao getUserDao()
  {
    return userDao;
  }

  public TrainingAttendeeDao setUserDao(final UserDao userDao)
  {
    this.userDao = userDao;
    return this;
  }

  @Override
  public List<TrainingAttendeeDO> getList(final BaseSearchFilter filter)
  {
    final TrainingAttendeeFilter myFilter;
    if (filter instanceof TrainingAttendeeFilter) {
      myFilter = (TrainingAttendeeFilter) filter;
    } else {
      myFilter = new TrainingAttendeeFilter(filter);
    }
    final QueryFilter queryFilter = new QueryFilter(myFilter);
    final String searchString = myFilter.getSearchString();

    if (myFilter.getAttendeeId() != null) {
      final PFUserDO attendee = new PFUserDO();
      attendee.setId(myFilter.getAttendeeId());
      queryFilter.add(Restrictions.eq("trainingAttendee", attendee));
    }
    if (myFilter.getTrainingId() != null) {
      final TrainingDO training = new TrainingDO();
      training.setId(myFilter.getTrainingId());
      queryFilter.add(Restrictions.eq("training", training));
    }
    queryFilter.addOrder(Order.desc("created"));
    final List<TrainingAttendeeDO> list = getList(queryFilter);
    myFilter.setSearchString(searchString); // Restore search string.
    return list;
  }
}