aboutsummaryrefslogtreecommitdiffstats
path: root/vid-app-common/src/main/java/org/onap/vid/aai/util/AAITreeConverter.java
blob: 8fa6f6c6d3c4ddbbffe0ad13446f0810bbfff74b (plain)
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
package org.onap.vid.aai.util;

import org.apache.commons.lang3.StringUtils;
import org.onap.vid.model.aaiTree.*;
import org.onap.vid.mso.model.ModelInfo;
import org.onap.vid.services.AAITreeNodeBuilder;
import org.springframework.stereotype.Component;

import java.util.Objects;

import static java.util.function.Function.identity;
import static java.util.stream.Collectors.counting;
import static java.util.stream.Collectors.groupingBy;
import static org.onap.vid.asdc.parser.ToscaParserImpl2.Constants.A_LA_CARTE;

@Component
public class AAITreeConverter {

    public static final String VNF_TYPE = "vnf-type";
    public static final String NETWORK_TYPE = "network-type";

    public static final String IS_BASE_VF_MODULE = "is-base-vf-module";

    public enum ModelType {
        service,
        vnf,
        network,
        instanceGroup,
        vfModule
    }

    public ServiceInstance convertTreeToUIModel(AAITreeNode rootNode, String globalCustomerId, String serviceType, String instantiationType) {
        ServiceInstance serviceInstance = new ServiceInstance();
        serviceInstance.setInstanceId(rootNode.getId());
        serviceInstance.setInstanceName(rootNode.getName());
        serviceInstance.setOrchStatus(rootNode.getOrchestrationStatus());
        serviceInstance.setGlobalSubscriberId(globalCustomerId);
        serviceInstance.setSubscriptionServiceType(serviceType);
        serviceInstance.setIsALaCarte(StringUtils.equals(instantiationType, A_LA_CARTE));

        serviceInstance.setModelInfo(createModelInfo(rootNode, ModelType.service));

        //set children: vnf, network,group
        rootNode.getChildren().forEach(child -> {
            if (child.getType().equals(AAITreeNodeBuilder.GENERIC_VNF)) {
                serviceInstance.getVnfs().put(child.getUniqueNodeKey(), Vnf.from(child));
            } else if (child.getType().equals(AAITreeNodeBuilder.NETWORK)) {
                serviceInstance.getNetworks().put(child.getUniqueNodeKey(), Network.from(child));
            } else if (child.getType().equals(AAITreeNodeBuilder.INSTANCE_GROUP)) {
                serviceInstance.getVnfGroups().put(child.getUniqueNodeKey(), VnfGroup.from(child));
            }
        });

        serviceInstance.setExistingVNFCounterMap(
                serviceInstance.getVnfs().entrySet().stream()
                        .map(k -> k.getValue().getModelInfo().getModelVersionId())
                        .collect(groupingBy(identity(), counting()))
        );

        serviceInstance.setExistingNetworksCounterMap(
                serviceInstance.getNetworks().entrySet().stream()
                .map(k -> k.getValue().getModelInfo().getModelVersionId())
                .filter(Objects::nonNull)
                        .collect(groupingBy(identity(), counting()))
        );


        serviceInstance.setExistingVnfGroupCounterMap(
                serviceInstance.getVnfGroups().entrySet().stream()
                .map(k -> k.getValue().getModelInfo().getModelVersionId())
                .filter(Objects::nonNull)
                        .collect(groupingBy(identity(), counting()))
        );

        return serviceInstance;
    }

    private static ModelInfo createModelInfo(AAITreeNode aaiNode, ModelType modelType) {
        ModelInfo modelInfo = new ModelInfo();
        modelInfo.setModelType(modelType.name());
        modelInfo.setModelName(aaiNode.getModelName());
        modelInfo.setModelVersion(aaiNode.getModelVersion());
        modelInfo.setModelVersionId(aaiNode.getModelVersionId());
        modelInfo.setModelInvariantId(aaiNode.getModelInvariantId());
        modelInfo.setModelCustomizationId(aaiNode.getModelCustomizationId());

        return modelInfo;
    }
}