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


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

package tools;

import json.response.ElementsResponse.Element;
import json.response.ItemsResponse.Item;
import utilities.IDcaeRestClient;

import java.util.HashMap;
import java.util.List;
import java.util.Map;


public class EntitiesRetriever {
    private final IDcaeRestClient dcaeRestClient;

    EntitiesRetriever(IDcaeRestClient dcaeRestClient) {

        this.dcaeRestClient = dcaeRestClient;
    }

    public Map<String, List<Item>> getElementsByFolder() {
        List<Element> elementList = dcaeRestClient.getElements();
        Map<String, List<Item>> elementsByFolderNames = new HashMap<>();

        for (Element element : elementList) {
            List<Item> items = dcaeRestClient.getItem(element.getName());
            if (items == null) {
                continue;
            }
            elementsByFolderNames.put(element.getName(), items);
        }
        return elementsByFolderNames;
    }
}