import java.util.List;
+import javax.annotation.Nonnull;
+
import org.akraino.validation.ui.dao.LabDAO;
import org.akraino.validation.ui.entity.LabInfo;
import org.hibernate.Criteria;
@Override
public List<LabInfo> getLabs() {
Criteria criteria = getSession().createCriteria(LabInfo.class);
+ criteria.setResultTransformer(Criteria.DISTINCT_ROOT_ENTITY);
return criteria.list();
}
@Override
- public LabInfo getLab(Integer labId) {
+ public LabInfo getLab(@Nonnull Integer labId) {
+ Criteria criteria = getSession().createCriteria(LabInfo.class);
+ 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 String lab) {
+ Criteria criteria = getSession().createCriteria(LabInfo.class);
+ criteria.setResultTransformer(Criteria.DISTINCT_ROOT_ENTITY);
+ criteria.add(Restrictions.eq("lab", lab));
+ 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.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("silo", silo));
+ return criteria.list() == null || criteria.list().size() < 1 ? null : (LabInfo) criteria.list().get(0);
}
@Override
- public void saveOrUpdate(LabInfo lab) {
+ public void saveOrUpdate(@Nonnull LabInfo lab) {
getSession().saveOrUpdate(lab);
+ getSession().flush();
}
@Override
- public void merge(LabInfo lab) {
+ public void merge(@Nonnull LabInfo lab) {
getSession().merge(lab);
+ getSession().flush();
}
@Override
- public void deleteLab(LabInfo lab) {
+ public void deleteLab(@Nonnull LabInfo lab) {
getSession().delete(lab);
+ getSession().flush();
}
@Override
public void deleteAll() {
if (getSession().createQuery("delete from Lab").executeUpdate() > 0) {
LOGGER.info(EELFLoggerDelegate.applicationLogger, "All lab entries are cleaned up");
+ getSession().flush();
}
}