[UI] Common class for results

[validation.git] / ui / src / main / java / org / akraino / validation / ui / service / LabService.java

diff --git a/ui/src/main/java/org/akraino/validation/ui/service/LabService.java b/ui/src/main/java/org/akraino/validation/ui/service/LabService.java
index 4280f37..81fe7b1 100644 (file)
--- a/ui/src/main/java/org/akraino/validation/ui/service/LabService.java
+++ b/ui/src/main/java/org/akraino/validation/ui/service/LabService.java
@@ -18,7 +18,6 @@ package org.akraino.validation.ui.service;
 import java.util.List;
 
 import org.akraino.validation.ui.dao.LabDAO;
-import org.akraino.validation.ui.data.Lab;
 import org.akraino.validation.ui.entity.LabInfo;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
@@ -35,10 +34,14 @@ public class LabService {
         labDAO.saveOrUpdate(lab);
     }
 
-    public LabInfo getLab(Lab lab) {
+    public LabInfo getLab(String lab) {
         return labDAO.getLab(lab);
     }
 
+    public LabInfo getLabBasedOnSilo(String silo) {
+        return labDAO.getLabBasedOnSilo(silo);
+    }
+
     public List<LabInfo> getLabs() {
         return labDAO.getLabs();
     }