/////////////////////////////////////////////////////////////////////////////
//
// 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.fibu;
import java.util.List;
import org.apache.commons.collections.CollectionUtils;
import org.hibernate.Hibernate;
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.task.TaskDO;
import org.projectforge.task.TaskDao;
import org.projectforge.user.GroupDO;
import org.projectforge.user.GroupDao;
import org.projectforge.user.UserRightId;
import org.springframework.transaction.annotation.Propagation;
import org.springframework.transaction.annotation.Transactional;
public class ProjektDao extends BaseDao<ProjektDO>
{
public static final UserRightId USER_RIGHT_ID = UserRightId.PM_PROJECT;
private static final String[] ADDITIONAL_SEARCH_FIELDS = new String[] { "kunde.name", "kunde.division", "kost2",
"projektManagerGroup.name"};
private KundeDao kundeDao;
private GroupDao groupDao;
private TaskDao taskDao;
public void setKundeDao(final KundeDao kundeDao)
{
this.kundeDao = kundeDao;
}
public void setGroupDao(final GroupDao groupDao)
{
this.groupDao = groupDao;
}
public void setTaskDao(final TaskDao taskDao)
{
this.taskDao = taskDao;
}
public ProjektDao()
{
super(ProjektDO.class);
this.supportAfterUpdate = true;
userRightId = USER_RIGHT_ID;
}
@Override
protected String[] getAdditionalSearchFields()
{
return ADDITIONAL_SEARCH_FIELDS;
}
/**
* @param projekt
* @param kundeId If null, then kunde will be set to null;
* @see BaseDao#getOrLoad(Integer)
*/
public void setKunde(final ProjektDO projekt, final Integer kundeId)
{
if (kundeId == null) {
projekt.setKunde(null);
} else {
final KundeDO kunde = kundeDao.getOrLoad(kundeId);
projekt.setKunde(kunde);
}
}
/**
* @param projekt
* @param tadkId If null, then task will be set to null;
* @see BaseDao#getOrLoad(Integer)
*/
public void setTask(final ProjektDO projekt, final Integer taskId)
{
if (taskId == null) {
projekt.setTask(null);
} else {
final TaskDO task = taskDao.getOrLoad(taskId);
projekt.setTask(task);
}
}
public void setProjektManagerGroup(final ProjektDO projekt, final Integer groupId)
{
if (groupId == null) {
projekt.setProjektManagerGroup(null);
} else {
final GroupDO group = groupDao.getOrLoad(groupId);
projekt.setProjektManagerGroup(group);
}
}
/**
* Initializes the projekt (projektManagerGroup), so any LazyInitializationException are avoided.
* @param projekt Null safe.
*/
public void initializeProjektManagerGroup(final ProjektDO projekt)
{
if (projekt == null) {
return;
}
// Needed because Hibernate Search rolls back because the project manager group is not loadable.
Hibernate.initialize(projekt);
final GroupDO projectManagerGroup = projekt.getProjektManagerGroup();
if (projectManagerGroup != null) {
final GroupDO group = groupDao.internalGetById(projectManagerGroup.getId());
projekt.setProjektManagerGroup(group);
//Hibernate.initialize(projectManagerGroup); // Does not work.
}
}
@SuppressWarnings("unchecked")
@Transactional(readOnly = true, propagation = Propagation.SUPPORTS)
public ProjektDO getProjekt(final KundeDO kunde, final int nummer)
{
final List<ProjektDO> list = getHibernateTemplate().find("from ProjektDO p where p.kunde.id=? and p.nummer=?",
new Object[] { kunde.getId(), nummer});
if (CollectionUtils.isEmpty(list) == true) {
return null;
}
return list.get(0);
}
@SuppressWarnings("unchecked")
@Transactional(readOnly = true, propagation = Propagation.SUPPORTS)
public ProjektDO getProjekt(final int intern_kost2_4, final int nummer)
{
final List<ProjektDO> list = getHibernateTemplate().find("from ProjektDO p where p.internKost2_4=? and p.nummer=?",
new Object[] { intern_kost2_4, nummer});
if (CollectionUtils.isEmpty(list) == true) {
return null;
}
return list.get(0);
}
@Override
@Transactional(readOnly = true, propagation = Propagation.SUPPORTS)
public List<ProjektDO> getList(final BaseSearchFilter filter)
{
final ProjektFilter myFilter;
if (filter instanceof ProjektFilter) {
myFilter = (ProjektFilter) filter;
} else {
myFilter = new ProjektFilter(filter);
}
final QueryFilter queryFilter = new QueryFilter(myFilter);
if (myFilter.isEnded() == true) {
queryFilter.add(Restrictions.eq("status", ProjektStatus.ENDED));
} else if (myFilter.isNotEnded() == true) {
queryFilter.add(Restrictions.or(Restrictions.ne("status", ProjektStatus.ENDED), Restrictions.isNull("status")));
}
queryFilter.addOrder(Order.asc("internKost2_4")).addOrder(Order.asc("kunde.id")).addOrder(Order.asc("nummer"));
return getList(queryFilter);
}
@Transactional(readOnly = true, propagation = Propagation.SUPPORTS)
public List<ProjektDO> getKundenProjekte(final Integer kundeId)
{
if (kundeId == null) {
return null;
}
final QueryFilter queryFilter = new QueryFilter();
queryFilter.add(Restrictions.eq("kunde.id", kundeId));
queryFilter.addOrder(Order.asc("nummer"));
return getList(queryFilter);
}
@Override
protected void onSaveOrModify(final ProjektDO obj)
{
if (obj.getKunde() != null) {
// Ein Kundenprojekt kann keine interne Kundennummer haben:
obj.setInternKost2_4(null);
}
if (obj.getStatus() == ProjektStatus.NONE) {
obj.setStatus(null);
}
super.onSaveOrModify(obj);
}
@Override
protected void afterSaveOrModify(final ProjektDO projekt)
{
if (projekt.getTaskId() != null) {
taskDao.getTaskTree().internalSetProject(projekt.getTaskId(), projekt);
}
super.afterSaveOrModify(projekt);
}
@Override
protected void afterUpdate(final ProjektDO obj, final ProjektDO dbObj)
{
if (dbObj.getTaskId() != null && obj.getTaskId() == null) {
// Project task was removed:
taskDao.getTaskTree().internalSetProject(dbObj.getTaskId(), null);
}
super.afterUpdate(obj, dbObj);
}
@Override
public ProjektDO newInstance()
{
return new ProjektDO();
}
}