dcaedt_tools/src/main/java/tools/EntitiesRetriever.java

   1 package tools;
   2
   3 import json.response.ElementsResponse.Element;
   4 import json.response.ItemsResponse.Item;
   5 import utilities.IDcaeRestClient;
   6
   7 import java.util.HashMap;
   8 import java.util.List;
   9 import java.util.Map;
  10
  11
  12 public class EntitiesRetriever {
  13     private final IDcaeRestClient dcaeRestClient;
  14
  15     EntitiesRetriever(IDcaeRestClient dcaeRestClient) {
  16
  17         this.dcaeRestClient = dcaeRestClient;
  18     }
  19
  20     public Map<String, List<Item>> getElementsByFolder() {
  21         List<Element> elementList = dcaeRestClient.getElements();
  22         Map<String, List<Item>> elementsByFolderNames = new HashMap<>();
  23
  24         for (Element element : elementList) {
  25             List<Item> items = dcaeRestClient.getItem(element.getName());
  26             if (items == null) {
  27                 continue;
  28             }
  29             elementsByFolderNames.put(element.getName(), items);
  30         }
  31         return elementsByFolderNames;
  32     }
  33 }