Merge pull request #96 from barspi/master

DBManager.byId() convenience method, and fix unchecked assignment
jpos · Mar 14, 2019 · d3185f6 · d3185f6
2 parents 8816904 + 023a9cb
commit d3185f6
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/modules/dbsupport/src/main/java/org/jpos/ee/DBManager.java b/modules/dbsupport/src/main/java/org/jpos/ee/DBManager.java
@@ -21,8 +21,8 @@
 import org.hibernate.query.criteria.internal.OrderImpl;
 
 import javax.persistence.NoResultException;
-import javax.persistence.Query;
 import javax.persistence.criteria.*;
+import org.hibernate.query.Query;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
@@ -37,6 +37,11 @@ public DBManager(DB db, Class<T> clazz) {
         this.clazz = clazz;
     }
 
+    /** Convenience method */
+    public T byId(Long id) {
+        return db.session().get(clazz, id);
+    }
+
     public int getItemCount()  {
         CriteriaBuilder criteriaBuilder = db.session().getCriteriaBuilder();
         CriteriaQuery<Long> query = criteriaBuilder.createQuery(Long.class);
@@ -65,7 +70,7 @@ public List<T> getAll(int offset, int limit, Map<String,Boolean> orders) {
             query.where(predicates);
         query.select(root);
         query.orderBy(orderList);
-        Query queryImp = db.session().createQuery(query);
+        Query<T> queryImp = db.session().createQuery(query);
         if (limit != -1) {
             queryImp.setMaxResults(limit);
         }