src/main/java/org/openecomp/sdc/toscaparser/api/CapabilityAssignments.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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

package org.openecomp.sdc.toscaparser.api;

import org.openecomp.sdc.toscaparser.api.CapabilityAssignment;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class CapabilityAssignments {

    private Map<String,CapabilityAssignment> capabilityAssignments;

    public CapabilityAssignments(Map<String,CapabilityAssignment> capabilityAssignments) {
        this.capabilityAssignments = capabilityAssignments != null ? new HashMap<>(capabilityAssignments) : new HashMap<>();
    }

    /**
     * Get all capability assignments for node template.<br>
     * This object can be either the original one, holding all capability assignments for this node template,or a filtered one, holding a filtered subset.<br>
     * @return list of capability assignments for the node template. <br>
     * If there are no capability assignments, empty list is returned.
     */
    public List<CapabilityAssignment> getAll() {
        return new ArrayList<>(capabilityAssignments.values());
    }

    /**
     * Filter capability assignments by capability tosca type.
     * @param type - The tosca type of capability assignments.
     * @return CapabilityAssignments object, containing capability assignments of this type.<br>
     * If no such found, filtering will result in an empty collection.
     */
    public CapabilityAssignments getCapabilitiesByType(String type) {
        Map<String,CapabilityAssignment> capabilityAssignmentsMap = capabilityAssignments.entrySet().stream()
                .filter(cap -> cap.getValue().getDefinition().getType().equals(type)).collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));

        return new CapabilityAssignments(capabilityAssignmentsMap);
    }

    /**
     * Get capability assignment by capability name.
     * @param name - The name of capability assignment
     * @return capability assignment with this name, or null if no such capability assignment was found.
     */
    public CapabilityAssignment getCapabilityByName(String name) {
        return capabilityAssignments.get(name);
    }

}