package fi.otavanopisto.pyramus.dao.base;
import javax.ejb.Stateless;
import javax.persistence.EntityManager;
import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Root;
import fi.otavanopisto.pyramus.dao.PyramusEntityDAO;
import fi.otavanopisto.pyramus.domainmodel.base.ContactInfo;
import fi.otavanopisto.pyramus.domainmodel.base.ContactType;
import fi.otavanopisto.pyramus.domainmodel.base.Email;
import fi.otavanopisto.pyramus.domainmodel.base.Email_;
@Stateless
public class EmailDAO extends PyramusEntityDAO<Email> {
public Email create(ContactInfo contactInfo, ContactType contactType, Boolean defaultAddress, String address) {
EntityManager entityManager = getEntityManager();
Email email = new Email();
email.setContactInfo(contactInfo);
email.setContactType(contactType);
email.setDefaultAddress(defaultAddress);
email.setAddress(address);
entityManager.persist(email);
contactInfo.addEmail(email);
entityManager.persist(contactInfo);
return email;
}
public Email findByContactInfoAndDefaultAddress(ContactInfo contactInfo, Boolean defaultAddress) {
EntityManager entityManager = getEntityManager();
CriteriaBuilder criteriaBuilder = entityManager.getCriteriaBuilder();
CriteriaQuery<Email> criteria = criteriaBuilder.createQuery(Email.class);
Root<Email> root = criteria.from(Email.class);
criteria.select(root);
criteria.where(
criteriaBuilder.and(
criteriaBuilder.equal(root.get(Email_.contactInfo), contactInfo),
criteriaBuilder.equal(root.get(Email_.defaultAddress), defaultAddress)
)
);
return getSingleResult(entityManager.createQuery(criteria));
}
public Email findByAddress(String emailAddress) {
EntityManager entityManager = getEntityManager();
CriteriaBuilder criteriaBuilder = entityManager.getCriteriaBuilder();
CriteriaQuery<Email> criteria = criteriaBuilder.createQuery(Email.class);
Root<Email> root = criteria.from(Email.class);
criteria.select(root);
criteria.where(
criteriaBuilder.equal(root.get(Email_.address), emailAddress)
);
return getSingleResult(entityManager.createQuery(criteria));
}
public Email update(Email email, ContactType contactType, Boolean defaultAddress, String address) {
EntityManager entityManager = getEntityManager();
email.setContactType(contactType);
email.setDefaultAddress(defaultAddress);
email.setAddress(address);
entityManager.persist(email);
return email;
}
@Override
public void delete(Email email) {
if (email.getContactInfo() != null) {
email.getContactInfo().removeEmail(email);
}
super.delete(email);
}
}