StudentFileDAO.java

package fi.otavanopisto.pyramus.dao.file;

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

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.file.FileType;
import fi.otavanopisto.pyramus.domainmodel.file.StudentFile;
import fi.otavanopisto.pyramus.domainmodel.students.Student;
import fi.otavanopisto.pyramus.domainmodel.users.User;
import fi.otavanopisto.pyramus.domainmodel.file.StudentFile_;

@Stateless
public class StudentFileDAO extends PyramusEntityDAO<StudentFile> {

  public StudentFile create(Student student, String name, String fileName, FileType fileType, String contentType, byte[] data, User creator) {
    EntityManager em = getEntityManager();
    Date created = new Date();
    
    StudentFile studentFile = new StudentFile();

    studentFile.setStudent(student);
    studentFile.setName(name);
    studentFile.setFileName(fileName);
    studentFile.setFileType(fileType);

    studentFile.setContentType(contentType);
    studentFile.setData(data);

    studentFile.setCreated(created);
    studentFile.setCreator(creator);
    studentFile.setLastModified(created);
    studentFile.setLastModifier(creator);
    
    em.persist(studentFile);

    return studentFile;
  }
  
  public StudentFile updateBasicInfo(StudentFile studentFile, String name, String fileName, FileType fileType, User modifier) {
    EntityManager em = getEntityManager();
    
    studentFile.setName(name);
    studentFile.setFileName(fileName);
    studentFile.setFileType(fileType);

    studentFile.setLastModified(new Date());
    studentFile.setLastModifier(modifier);
    
    em.persist(studentFile);

    return studentFile;
  }

  public StudentFile updateData(StudentFile studentFile, String contentType, byte[] data, User modifier) {
    EntityManager em = getEntityManager();
    
    studentFile.setContentType(contentType);
    studentFile.setData(data);
    studentFile.setLastModified(new Date());
    studentFile.setLastModifier(modifier);
    
    em.persist(studentFile);

    return studentFile;
  }
  
  public List<StudentFile> listByStudent(Student student) {
    EntityManager entityManager = getEntityManager(); 
    
    CriteriaBuilder criteriaBuilder = entityManager.getCriteriaBuilder();
    CriteriaQuery<StudentFile> criteria = criteriaBuilder.createQuery(StudentFile.class);
    Root<StudentFile> root = criteria.from(StudentFile.class);
    criteria.select(root);
    criteria.where(
        criteriaBuilder.and(
            criteriaBuilder.equal(root.get(StudentFile_.archived), Boolean.FALSE),
            criteriaBuilder.equal(root.get(StudentFile_.student), student)));
    
    return entityManager.createQuery(criteria).getResultList();
  }
}