UserApprovalFacade.java 1.31 KB
package fi.codecrew.moya.facade;

import java.util.Date;

import javax.ejb.LocalBean;
import javax.ejb.Stateless;
import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Root;

import fi.codecrew.moya.model.Approval;
import fi.codecrew.moya.model.EventUser;
import fi.codecrew.moya.model.UserApproval;
import fi.codecrew.moya.model.UserApproval_;

@Stateless
@LocalBean
public class UserApprovalFacade extends IntegerPkGenericFacade<UserApproval> {

	public UserApprovalFacade() {
		super(UserApproval.class);
	}

	public UserApproval findOrCreateApproval(EventUser user, Approval approval) {
		UserApproval ret = findApproval(user, approval);
		if (ret == null)
		{
			ret = new UserApproval();
			ret.setUser(user);
			ret.setApproval(approval);
			ret.setApprovalTime(new Date());
			create(ret);
		}
		return ret;
	}

	public UserApproval findApproval(EventUser user, Approval approval) {
		CriteriaBuilder cb = getEm().getCriteriaBuilder();
		CriteriaQuery<UserApproval> cq = cb.createQuery(UserApproval.class);
		Root<UserApproval> root = cq.from(UserApproval.class);
		cq.where(cb.equal(root.get(UserApproval_.approval), approval),
				cb.equal(root.get(UserApproval_.user), user)
				);
		return super.getSingleNullableResult(getEm().createQuery(cq));
	}

}