X-Git-Url: https://gerrit.akraino.org/r/gitweb?p=validation.git;a=blobdiff_plain;f=ui%2Fsrc%2Fmain%2Fjava%2Forg%2Fakraino%2Fvalidation%2Fui%2Fdaoimpl%2FLabDAOImpl.java;fp=ui%2Fsrc%2Fmain%2Fjava%2Forg%2Fakraino%2Fvalidation%2Fui%2Fdaoimpl%2FLabDAOImpl.java;h=1bfc23153d05967de8943d23497574f170a21cac;hp=36512ab7b5bfcf61e2032f61313f6f1334cf7d84;hb=147ecf7bf79ea9967a121d0038103151a38ebef2;hpb=e6b82a7ccd840c8b089ae4d5e69930fd0dd5ef35

diff --git a/ui/src/main/java/org/akraino/validation/ui/daoimpl/LabDAOImpl.java b/ui/src/main/java/org/akraino/validation/ui/daoimpl/LabDAOImpl.java
index 36512ab..1bfc231 100644
--- a/ui/src/main/java/org/akraino/validation/ui/daoimpl/LabDAOImpl.java
+++ b/ui/src/main/java/org/akraino/validation/ui/daoimpl/LabDAOImpl.java
@@ -20,7 +20,6 @@ import java.util.List;
 import javax.annotation.Nonnull;
 
 import org.akraino.validation.ui.dao.LabDAO;
-import org.akraino.validation.ui.data.Lab;
 import org.akraino.validation.ui.entity.LabInfo;
 import org.hibernate.Criteria;
 import org.hibernate.Session;
@@ -45,21 +44,32 @@ public class LabDAOImpl implements LabDAO {
     @Override
     public List<LabInfo> getLabs() {
         Criteria criteria = getSession().createCriteria(LabInfo.class);
+        criteria.setResultTransformer(Criteria.DISTINCT_ROOT_ENTITY);
         return criteria.list();
     }
 
     @Override
     public LabInfo getLab(@Nonnull Integer labId) {
         Criteria criteria = getSession().createCriteria(LabInfo.class);
-        criteria.add(Restrictions.eq("id", String.valueOf(labId)));
-        return criteria.list() == null ? null : (LabInfo) criteria.list().get(0);
+        criteria.setResultTransformer(Criteria.DISTINCT_ROOT_ENTITY);
+        criteria.add(Restrictions.eq("id", labId));
+        return criteria.list() == null || criteria.list().size() < 1 ? null : (LabInfo) criteria.list().get(0);
     }
 
     @Override
-    public LabInfo getLab(@Nonnull Lab lab) {
+    public LabInfo getLab(@Nonnull String lab) {
         Criteria criteria = getSession().createCriteria(LabInfo.class);
+        criteria.setResultTransformer(Criteria.DISTINCT_ROOT_ENTITY);
         criteria.add(Restrictions.eq("lab", lab));
-        return criteria.list() == null ? null : (LabInfo) criteria.list().get(0);
+        return criteria.list() == null || criteria.list().size() < 1 ? null : (LabInfo) criteria.list().get(0);
+    }
+
+    @Override
+    public LabInfo getLabBasedOnSilo(String silo) {
+        Criteria criteria = getSession().createCriteria(LabInfo.class);
+        criteria.setResultTransformer(Criteria.DISTINCT_ROOT_ENTITY);
+        criteria.add(Restrictions.eq("silo", silo));
+        return criteria.list() == null || criteria.list().size() < 1 ? null : (LabInfo) criteria.list().get(0);
     }
 
     @Override