Revize 16c80fab

https://kivprogrammers.atlassian.net/browse/TSP2-62 přidání filteru

src/main/java/cz/zcu/fav/kiv/antipatterndetectionapp/v2/controller/management/WorkUnitController.java

@@ -16 +16 @@
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
 
+import javax.annotation.Nullable;
 import javax.persistence.criteria.CriteriaBuilder;
-import java.util.Arrays;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 
 @RequestMapping("v2/management")
 @Controller

src/main/java/cz/zcu/fav/kiv/antipatterndetectionapp/v2/service/workUnit/WorkUnitServiceV2Impl.java

@@ -6 +6 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
+import java.util.HashSet;
 import java.util.List;
+import java.util.Set;
 
 @Service
 public class WorkUnitServiceV2Impl implements WorkUnitServiceV2 {
@@ -30 +32 @@
         return updatedRows == wuCount;
     }
 
+    @Override
+    public Set<String> parseWorkUnitCategories(List<WorkUnit> workUnits) {
+        Set<String> workUnitCategories = new HashSet<>();
+        for(WorkUnit unit : workUnits) {
+            for(Category categ : unit.getCategories())
+                workUnitCategories.add(categ.getName());
+        }
+        return workUnitCategories;
+    }
+
+    @Override
+    public Set<String> parseWorkUnitTypes(List<WorkUnit> workUnits) {
+        Set<String> workUnitTypes = new HashSet<>();
+        for(WorkUnit unit : workUnits) {
+            workUnitTypes.add(unit.getType().getName());
+        }
+        return workUnitTypes;
+    }
+
 }