[UI] Common class for results

[validation.git] / ui / src / main / java / org / akraino / validation / ui / daoimpl / BlueprintDAOImpl.java

diff --git a/ui/src/main/java/org/akraino/validation/ui/daoimpl/BlueprintDAOImpl.java b/ui/src/main/java/org/akraino/validation/ui/daoimpl/BlueprintDAOImpl.java
index a349bcf..6c751ad 100644 (file)
--- a/ui/src/main/java/org/akraino/validation/ui/daoimpl/BlueprintDAOImpl.java
+++ b/ui/src/main/java/org/akraino/validation/ui/daoimpl/BlueprintDAOImpl.java
@@ -44,14 +44,24 @@ public class BlueprintDAOImpl implements BlueprintDAO {
     @Override
     public List<Blueprint> getBlueprints() {
         Criteria criteria = getSession().createCriteria(Blueprint.class);
+        criteria.setResultTransformer(Criteria.DISTINCT_ROOT_ENTITY);
         return criteria.list();
     }
 
     @Override
     public Blueprint getBlueprint(@Nonnull Integer blueprintId) {
         Criteria criteria = getSession().createCriteria(Blueprint.class);
-        criteria.add(Restrictions.eq("id", String.valueOf(blueprintId)));
-        return criteria.list() == null ? null : (Blueprint) criteria.list().get(0);
+        criteria.setResultTransformer(Criteria.DISTINCT_ROOT_ENTITY);
+        criteria.add(Restrictions.eq("id", blueprintId));
+        return criteria.list() == null || criteria.list().size() < 1 ? null : (Blueprint) criteria.list().get(0);
+    }
+
+    @Override
+    public Blueprint getBlueprint(String name) {
+        Criteria criteria = getSession().createCriteria(Blueprint.class);
+        criteria.setResultTransformer(Criteria.DISTINCT_ROOT_ENTITY);
+        criteria.add(Restrictions.eq("blueprintName", name));
+        return criteria.list() == null || criteria.list().size() < 1 ? null : (Blueprint) criteria.list().get(0);
     }
 
     @Override