PlaceSlotFacade.java

/*
 * Copyright Codecrew Ry
 * 
 * All rights reserved.
 * 
 * This license applies to any software containing a notice placed by the 
 * copyright holder. Such software is herein referred to as the Software. 
 * This license covers modification, distribution and use of the Software. 
 * 
 * Any distribution and use in source and binary forms, with or without 
 * modification is not permitted without explicit written permission from the 
 * copyright owner. 
 * 
 * A non-exclusive royalty-free right is granted to the copyright owner of the 
 * Software to use, modify and distribute all modifications to the Software in 
 * future versions of the Software. 
 * 
 */
package fi.codecrew.moya.facade;

import java.util.ArrayList;
import java.util.Calendar;
import java.util.List;

import javax.ejb.EJB;
import javax.ejb.LocalBean;
import javax.ejb.Stateless;
import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Path;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;
import javax.persistence.criteria.Subquery;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import fi.codecrew.moya.beans.EventBeanLocal;
import fi.codecrew.moya.model.Bill;
import fi.codecrew.moya.model.Bill_;
import fi.codecrew.moya.model.EventMap;
import fi.codecrew.moya.model.EventMap_;
import fi.codecrew.moya.model.EventUser;
import fi.codecrew.moya.model.Place;
import fi.codecrew.moya.model.PlaceSlot;
import fi.codecrew.moya.model.PlaceSlot_;
import fi.codecrew.moya.model.Place_;
import fi.codecrew.moya.model.Product;
import fi.codecrew.moya.model.Product_;

@Stateless
@LocalBean
public class PlaceSlotFacade extends IntegerPkGenericFacade<PlaceSlot> {

	private static final Logger logger = LoggerFactory.getLogger(PlaceSlotFacade.class);

	@EJB
	EventBeanLocal eventBean;

	public PlaceSlotFacade() {

		super(PlaceSlot.class);
	}

	/**
	 * Returns placeslots this user can use.
	 * 
	 * @param user
	 * @param product
	 * @return
	 */
	public List<PlaceSlot> findFreePlaceSlotsForProduct(EventUser user, Product product) {
		CriteriaBuilder cb = getEm().getCriteriaBuilder();
		CriteriaQuery<PlaceSlot> q = cb.createQuery(PlaceSlot.class);
		Root<PlaceSlot> root = q.from(PlaceSlot.class);
		Path<Bill> bill = root.get(PlaceSlot_.bill);
		final List<Predicate> preds = new ArrayList<>();
		preds.add(cb.equal(bill.get(Bill_.user), user));
		preds.add(cb.isNotNull(bill.get(Bill_.paidDate)));
		preds.add(cb.isNull(root.get(PlaceSlot_.used)));
		if (product != null) {
			preds.add(cb.equal(root.get(PlaceSlot_.product), product));
		}
		q.where(preds.toArray(new Predicate[preds.size()]));

		return getEm().createQuery(q).getResultList();
	}

	//	private List<PlaceSlot> getSlotsForUser(EventUser u) {
	//
	//		CriteriaBuilder cb = getEm().getCriteriaBuilder();
	//		CriteriaQuery<PlaceSlot> cq = cb.createQuery(PlaceSlot.class);
	//
	//	}

	public PlaceSlot findSlotForPlace(Place place) {
		CriteriaBuilder cb = getEm().getCriteriaBuilder();
		CriteriaQuery<PlaceSlot> q = cb.createQuery(PlaceSlot.class);
		Root<PlaceSlot> root = q.from(PlaceSlot.class);
		q.where(cb.equal(root.get(PlaceSlot_.place), place));
		PlaceSlot slot = super.getSingleNullableResult(getEm().createQuery(q));

		return slot;
	}

	public List<PlaceSlot> finForUser(EventUser user) {
		CriteriaBuilder cb = getEm().getCriteriaBuilder();
		CriteriaQuery<PlaceSlot> q = cb.createQuery(PlaceSlot.class);
		Root<PlaceSlot> root = q.from(PlaceSlot.class);
		Path<Bill> bill = root.get(PlaceSlot_.bill);
		q.where(cb.equal(bill.get(Bill_.user), user),
				cb.isNotNull(bill.get(Bill_.paidDate)));

		q.orderBy(cb.asc(root.get(PlaceSlot_.place)));

		return getEm().createQuery(q).getResultList();
	}

	public List<PlaceSlot> findUnusedSlots(Product prod, boolean paidOnly) {
		CriteriaBuilder cb = getEm().getCriteriaBuilder();
		CriteriaQuery<PlaceSlot> q = cb.createQuery(PlaceSlot.class);

		Root<PlaceSlot> root = q.from(PlaceSlot.class);
		q.select(root);

		List<Predicate> preds = getUnusedSlotPredicates(cb, root, prod, paidOnly);
		q.where(preds.toArray(new Predicate[preds.size()]));

		return getEm().createQuery(q).getResultList();

	}

	private static List<Predicate> getUnusedSlotPredicates(CriteriaBuilder cb, Root<PlaceSlot> root, Product prod, boolean paidOnly)
	{
		Path<Bill> bill = root.get(PlaceSlot_.bill);

		List<Predicate> preds = new ArrayList<>();
		preds.add(cb.equal(root.get(PlaceSlot_.product), prod));

		if (paidOnly) {
			// Only if bill is paid
			preds.add(cb.isNotNull(bill.get(Bill_.paidDate)));
		} else {
			// If expire is null or has not passed, count it
			Path<Calendar> billexp = bill.get(Bill_.expires);
			preds.add(cb.or(
					cb.isNull(billexp),
					cb.greaterThan(billexp, Calendar.getInstance())
					));
		}

		// Check that slot is not used
		preds.add(cb.isNull(root.get(PlaceSlot_.place)));
		preds.add(cb.isNull(root.get(PlaceSlot_.used)));
		return preds;

	}

	public Long findUnusedSlotsCount(Product prod, boolean paidOnly) {
		CriteriaBuilder cb = getEm().getCriteriaBuilder();
		CriteriaQuery<Long> q = cb.createQuery(Long.class);
		Root<PlaceSlot> root = q.from(PlaceSlot.class);
		q.select(cb.count(root));

		List<Predicate> preds = getUnusedSlotPredicates(cb, root, prod, paidOnly);
		q.where(preds.toArray(new Predicate[preds.size()]));

		Long count = super.getSingleNullableResult(getEm().createQuery(q));
		return count;
	}

	public Long totalSlotcount(Product prod, boolean paidOnly) {
		CriteriaBuilder cb = getEm().getCriteriaBuilder();
		CriteriaQuery<Long> q = cb.createQuery(Long.class);
		Root<PlaceSlot> root = q.from(PlaceSlot.class);
		q.select(cb.count(root));
		Path<Bill> bill = root.get(PlaceSlot_.bill);

		List<Predicate> preds = new ArrayList<>();
		preds.add(cb.equal(root.get(PlaceSlot_.product), prod));
		if (paidOnly) {
			preds.add(cb.isNotNull(bill.get(Bill_.paidDate)));
		} else {
			Path<Calendar> billexp = bill.get(Bill_.expires);
			preds.add(cb.or(
					cb.isNull(billexp),
					cb.greaterThan(billexp, Calendar.getInstance())
					));
		}

		q.where(preds.toArray(new Predicate[preds.size()]));

		Long count = super.getSingleNullableResult(getEm().createQuery(q));
		return count;
	}

	public List<PlaceSlot> findFreePlaceSlots(EventUser user, EventMap map) {
		CriteriaBuilder cb = getEm().getCriteriaBuilder();
		CriteriaQuery<PlaceSlot> q = cb.createQuery(PlaceSlot.class);
		Root<PlaceSlot> root = q.from(PlaceSlot.class);
		Path<Bill> bill = root.get(PlaceSlot_.bill);
		final List<Predicate> preds = new ArrayList<>();
		preds.add(cb.equal(bill.get(Bill_.user), user));
		preds.add(cb.isNotNull(bill.get(Bill_.paidDate)));
		preds.add(cb.isNull(root.get(PlaceSlot_.used)));

		Subquery<Integer> mapProdq = q.subquery(Integer.class);
		Root<EventMap> mapProdqRoot = mapProdq.from(EventMap.class);
		mapProdq.select(mapProdqRoot.join(EventMap_.places).get(Place_.product).get(Product_.id));
		mapProdq.where(cb.equal(mapProdqRoot, map));
		mapProdq.distinct(true);

		preds.add(root.get(PlaceSlot_.product).get(Product_.id).in(mapProdq));

		q.where(preds.toArray(new Predicate[preds.size()]));

		return getEm().createQuery(q).getResultList();
	}

}