aboutsummaryrefslogtreecommitdiffstats
path: root/vid-app-common/src/main/java/org/onap/vid/aai/util/AAITreeConverter.java
blob: 5be26a0ecd7cf9925b5fbc1bc13b212c8f782276 (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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
/*-
 * ============LICENSE_START=======================================================
 * VID
 * ================================================================================
 * Copyright (C) 2017 - 2019 AT&T Intellectual Property. All rights reserved.
 * ================================================================================
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * ============LICENSE_END=========================================================
 */

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;
    }
}