fi.foyt.fni.persistence.dao.users.UserDAO Maven / Gradle / Ivy

Go to download
package fi.foyt.fni.persistence.dao.users;

import java.util.Date;
import java.util.List;
import java.util.Locale;

import javax.persistence.EntityManager;
import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Root;

import fi.foyt.fni.persistence.dao.GenericDAO;
import fi.foyt.fni.persistence.model.users.User;
import fi.foyt.fni.persistence.model.users.UserProfileImageSource;
import fi.foyt.fni.persistence.model.users.UserRole;
import fi.foyt.fni.persistence.model.users.User_;

public class UserDAO extends GenericDAO {

	private static final long serialVersionUID = 1L;

	public User create(String firstName, String lastName, String nickname, Locale locale, Date registrationDate, UserRole role, UserProfileImageSource profileImageSource) {
    EntityManager entityManager = getEntityManager();

    User user = new User();
    user.setArchived(Boolean.FALSE);
    user.setFirstName(firstName);
    user.setLastName(lastName);
    user.setLocale(locale.toString());
    user.setNickname(nickname);
    user.setPremiumExpires(null);
    user.setRole(role);
    user.setRegistrationDate(registrationDate);
    user.setProfileImageSource(profileImageSource);

    entityManager.persist(user);

    return user;
  }
  
  public List listByArchived(Boolean archived) {
    EntityManager entityManager = getEntityManager();

    CriteriaBuilder criteriaBuilder = entityManager.getCriteriaBuilder();
    CriteriaQuery criteria = criteriaBuilder.createQuery(User.class);
    Root root = criteria.from(User.class);
    criteria.select(root);
    criteria.where(criteriaBuilder.equal(root.get(User_.archived), archived));

    return entityManager.createQuery(criteria).getResultList();
  }
  
  public List listByArchivedAndRoleIn(Boolean archived, List roles) {
    EntityManager entityManager = getEntityManager();

    CriteriaBuilder criteriaBuilder = entityManager.getCriteriaBuilder();
    CriteriaQuery criteria = criteriaBuilder.createQuery(User.class);
    Root root = criteria.from(User.class);
    criteria.select(root);
    criteria.where(
      criteriaBuilder.and(
        root.get(User_.role).in(roles),
        criteriaBuilder.equal(root.get(User_.archived), archived)
      )
    );

    return entityManager.createQuery(criteria).getResultList();
  }

  public User updateFirstName(User user, String firstName) {
    EntityManager entityManager = getEntityManager();

    user.setFirstName(firstName);

    entityManager.persist(user);
    
    return user;
  }

  public User updateLastName(User user, String lastName) {
    EntityManager entityManager = getEntityManager();
    user.setLastName(lastName);

    entityManager.persist(user);
    
    return user;
  }

  public User updateNickname(User user, String nickname) {
    EntityManager entityManager = getEntityManager();

    user.setNickname(nickname);

    entityManager.persist(user);
    
    return user;

  }

  public User updateLocale(User user, Locale locale) {
    EntityManager entityManager = getEntityManager();
    user.setLocale(locale.toString());

    entityManager.persist(user);
    
    return user;

  }

  public User updateRegistrationDate(User user, Date registrationDate) {
    EntityManager entityManager = getEntityManager();
    user.setRegistrationDate(registrationDate);

    entityManager.persist(user);
    
    return user;

  }

  public User updateArchived(User user, Boolean archived) {
    EntityManager entityManager = getEntityManager();

    user.setArchived(archived);

    entityManager.persist(user);
    
    return user;

  }

  public User updateRole(User user, UserRole role) {
    EntityManager entityManager = getEntityManager();

    user.setRole(role);

    entityManager.persist(user);
    
    return user;
  }

	public User updateAbout(User user, String about) {
		user.setAbout(about);
		return persist(user);
	}

	public User updateCompany(User user, String company) {
		user.setCompany(company);
		return persist(user);
	}

	public User updateMobile(User user, String mobile) {
		user.setMobile(mobile);
		return persist(user);
	}

	public User updatePhone(User user, String phone) {
		user.setPhone(phone);
		return persist(user);
	}

	public User updateProfileImageSource(User user, UserProfileImageSource profileImageSource) {
		user.setProfileImageSource(profileImageSource);
		return persist(user);
	}

}