summaryrefslogtreecommitdiffstats
path: root/src/main/java/nu/xss/jpa/dao/GenericJpaDao.java
blob: 1eb1f59e81f93a1c8f4fad178bcd2eda8b7dec9d (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
package nu.xss.jpa.dao;

import java.io.Serializable;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.util.List;

import javax.persistence.EntityManager;
import javax.persistence.NoResultException;
import javax.persistence.NonUniqueResultException;
import javax.persistence.PersistenceContext;
import javax.persistence.TypedQuery;
import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Root;

import nu.xss.jpa.entity.TypedEntity;

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

public abstract class GenericJpaDao<E extends TypedEntity<K>, K> implements
		Dao<E, K>, Serializable {

	private static final long serialVersionUID = 4998055731089977476L;
	protected final Logger logger = LoggerFactory.getLogger(this.getClass());

	@PersistenceContext
	protected EntityManager entityManager;

	protected Class<E> entity;

	@SuppressWarnings("unchecked")
	public GenericJpaDao() {
		ParameterizedType genericSuperclass = (ParameterizedType) getClass()
				.getGenericSuperclass();
	    Type type = genericSuperclass.getActualTypeArguments()[0];
	    if (type instanceof Class) {
	      this.entity = (Class<E>) type;
	    } else if (type instanceof ParameterizedType) {
	      this.entity = (Class<E>) ((ParameterizedType)type).getRawType();
	    }
	}

	public void save(E entity) {
		logger.info("Persist Entity: {}.", entity.toString());
		entityManager.persist(entity);
		logger.info("Saved entity: {}.", entity.toString());
	}

	public void merge(E entity) {
		logger.info("Merge Entity: {}.", entity.toString());
		entityManager.merge(entity);
		logger.info("Saved entity: {}.", entity.toString());
	}

	public void delete(E entity) {
		E persistentEntity = findById(entity.getId());
		entityManager.remove(persistentEntity);
		logger.info("Deleted entity: {}.", entity.toString());
	}

	public E findById(K id) {
		return entityManager.find(entity, id);
	}

	public List<E> findAll() {
		return find(createSimpleQuery());
	}

	public List<E> findAll(int count) {
		return find(createSimpleQuery(), count);
	}

	public List<E> findAll(int count, int  offset) {
		return find(createSimpleQuery(), count, offset);
	}
	
	protected CriteriaBuilder getCriteriaBuilder() {
		return entityManager.getCriteriaBuilder();
	}

	protected CriteriaQuery<E> createQuery() {
		CriteriaQuery<E> c = getCriteriaBuilder().createQuery(this.entity);
		return c;
	}

	protected CriteriaQuery<E> createSimpleQuery() {
		CriteriaQuery<E> c = getCriteriaBuilder().createQuery(this.entity);
		Root<E> root = c.from(this.entity);
		c.select(root);
		return c;
	}

	protected List<E> find(CriteriaQuery<E> query) {
		return entityManager.createQuery(query).getResultList();
	}

	protected List<E> find(CriteriaQuery<E> query, int count) {
		TypedQuery<E> q = entityManager.createQuery(query);
		q.setMaxResults(count);
		return q.getResultList();
	}

	protected List<E> find(CriteriaQuery<E> query, int count, int offset) {
		TypedQuery<E> q = entityManager.createQuery(query);
		q.setFirstResult(offset);
		q.setMaxResults(count);
		return q.getResultList();
	}

	protected E findSingle(CriteriaQuery<E> query) {
		return entityManager.createQuery(query).getSingleResult();
	}

	protected E findSingleOrNull(CriteriaQuery<E> query) {
		try {
			return findSingle(query);
		} catch (final NoResultException e) {
			return null;
		} catch (final NonUniqueResultException e) {
			logger.error("Found more than one result.. return first one");
			return null;
		}
	}

	protected E findSingleFirstOrNull(CriteriaQuery<E> query) {
		try {
			return findSingle(query);
		} catch (final NoResultException e) {
			return null;
		} catch (final NonUniqueResultException e) {
			logger.warn("Found more than one result.. return first one");
			return find(query).get(0);
		}
	}
}