enabled NFV Affinity and Anti-affinity Rules
[sdc.git] / catalog-be / src / main / java / org / openecomp / sdc / be / components / impl / PolicyTypeImportManager.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * SDC
4  * ================================================================================
5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
6  * ================================================================================
7  * Licensed under the Apache License, Version 2.0 (the "License");
8  * you may not use this file except in compliance with the License.
9  * You may obtain a copy of the License at
10  * 
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  * 
13  * Unless required by applicable law or agreed to in writing, software
14  * distributed under the License is distributed on an "AS IS" BASIS,
15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16  * See the License for the specific language governing permissions and
17  * limitations under the License.
18  * ============LICENSE_END=========================================================
19  */
20
21 package org.openecomp.sdc.be.components.impl;
22
23 import fj.data.Either;
24 import org.apache.commons.lang3.tuple.ImmutablePair;
25 import org.openecomp.sdc.be.components.impl.CommonImportManager.ElementTypeEnum;
26 import org.openecomp.sdc.be.components.impl.model.ToscaTypeImportData;
27 import org.openecomp.sdc.be.components.impl.utils.PolicyTypeImportUtils;
28 import org.openecomp.sdc.be.dao.api.ActionStatus;
29 import org.openecomp.sdc.be.impl.ComponentsUtils;
30 import org.openecomp.sdc.be.model.GroupTypeDefinition;
31 import org.openecomp.sdc.be.model.PolicyTypeDefinition;
32 import org.openecomp.sdc.be.model.jsontitan.operations.ToscaOperationFacade;
33 import org.openecomp.sdc.be.model.operations.api.IGroupOperation;
34 import org.openecomp.sdc.be.model.operations.api.IGroupTypeOperation;
35 import org.openecomp.sdc.be.model.operations.api.IPolicyTypeOperation;
36 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
37 import org.openecomp.sdc.be.utils.TypeUtils;
38 import org.openecomp.sdc.exception.ResponseFormat;
39 import org.springframework.stereotype.Component;
40
41 import java.util.List;
42 import java.util.Map;
43 import java.util.function.Consumer;
44
45 @Component("policyTypeImportManager")
46 public class PolicyTypeImportManager {
47
48     private final IPolicyTypeOperation policyTypeOperation;
49     private final ComponentsUtils componentsUtils;
50     private final IGroupOperation groupOperation;
51     private final ToscaOperationFacade toscaOperationFacade;
52     private final CommonImportManager commonImportManager;
53     private final IGroupTypeOperation groupTypeOperation;
54
55     public PolicyTypeImportManager(IPolicyTypeOperation policyTypeOperation, ComponentsUtils componentsUtils,
56         IGroupOperation groupOperation, ToscaOperationFacade toscaOperationFacade,
57         CommonImportManager commonImportManager, IGroupTypeOperation groupTypeOperation) {
58         this.policyTypeOperation = policyTypeOperation;
59         this.componentsUtils = componentsUtils;
60         this.groupOperation = groupOperation;
61         this.toscaOperationFacade = toscaOperationFacade;
62         this.commonImportManager = commonImportManager;
63         this.groupTypeOperation = groupTypeOperation;
64     }
65
66     public Either<List<ImmutablePair<PolicyTypeDefinition, Boolean>>, ResponseFormat> createPolicyTypes(ToscaTypeImportData toscaTypeImportData) {
67         return commonImportManager.createElementTypes(toscaTypeImportData, this::createPolicyTypesFromYml, this::upsertPolicyTypesByDao);
68     }
69
70     private Either<List<PolicyTypeDefinition>, ActionStatus> createPolicyTypesFromYml(String policyTypesYml) {
71         return commonImportManager.createElementTypesFromYml(policyTypesYml, this::createPolicyType);
72     }
73
74     private Either<List<ImmutablePair<PolicyTypeDefinition, Boolean>>, ResponseFormat> upsertPolicyTypesByDao(List<PolicyTypeDefinition> policyTypesToCreate) {
75         return commonImportManager.createElementTypesByDao(policyTypesToCreate, this::validatePolicyType, policyType -> new ImmutablePair<>(ElementTypeEnum.POLICY_TYPE, policyType.getType()),
76                 policyTypeOperation::getLatestPolicyTypeByType, policyTypeOperation::addPolicyType, this::updatePolicyType);
77     }
78
79     private Either<PolicyTypeDefinition, StorageOperationStatus> updatePolicyType(PolicyTypeDefinition newPolicyType, PolicyTypeDefinition oldPolicyType) {
80         if (PolicyTypeImportUtils.isPolicyTypesEquals(newPolicyType, oldPolicyType)) {
81             return policyTypeAlreadyExists();
82         }
83         return policyTypeOperation.updatePolicyType(newPolicyType, oldPolicyType);
84     }
85
86     private Either<PolicyTypeDefinition, StorageOperationStatus> policyTypeAlreadyExists() {
87         return Either.right(StorageOperationStatus.OK);
88     }
89
90     private Either<ActionStatus, ResponseFormat> validatePolicyType(PolicyTypeDefinition policyType) {
91         Either<ActionStatus, ResponseFormat> result = Either.left(ActionStatus.OK);
92         if (policyType.getTargets() != null) {
93             if (policyType.getTargets().isEmpty()) {
94                 ResponseFormat responseFormat = componentsUtils.getResponseFormat(ActionStatus.TARGETS_EMPTY, policyType.getType());
95                 result = Either.right(responseFormat);
96             }
97             if (result.isLeft()) {
98                 for (String targetId : policyType.getTargets()) {
99                         boolean isValid = toscaOperationFacade.getLatestByToscaResourceName(targetId).isLeft();
100
101                         if (!isValid) { // check if it is a groupType
102                             final Either<GroupTypeDefinition, StorageOperationStatus> groupTypeFound = groupTypeOperation
103                                 .getLatestGroupTypeByType(targetId, false);
104                             isValid = groupTypeFound.isLeft() && !groupTypeFound.left().value().isEmpty();
105                         }
106
107                         if (!isValid) {
108                             isValid = groupOperation.isGroupExist(targetId, false);
109                         }
110
111                         if (!isValid) {
112                             ResponseFormat responseFormat = componentsUtils.getResponseFormat(ActionStatus.TARGETS_NON_VALID, policyType.getType(), targetId);
113                             result = Either.right(responseFormat);
114                             break;
115                         }
116                     }
117                 }
118             }
119
120
121         return result;
122     }
123
124     private PolicyTypeDefinition createPolicyType(String groupTypeName, Map<String, Object> toscaJson) {
125
126         PolicyTypeDefinition policyType = new PolicyTypeDefinition();
127
128         if (toscaJson != null) {
129             // Description
130             final Consumer<String> descriptionSetter = policyType::setDescription;
131             commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.DESCRIPTION.getElementName(), descriptionSetter);
132             // Derived From
133             final Consumer<String> derivedFromSetter = policyType::setDerivedFrom;
134             commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.DERIVED_FROM.getElementName(), derivedFromSetter);
135             // Properties
136             CommonImportManager.setProperties(toscaJson, policyType::setProperties);
137             // Metadata
138             final Consumer<Map<String, String>> metadataSetter = policyType::setMetadata;
139             commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.METADATA.getElementName(), metadataSetter);
140             // Targets
141             final Consumer <List<String>> targetsSetter = policyType::setTargets;
142             commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.TARGETS.getElementName(), targetsSetter);
143
144             policyType.setType(groupTypeName);
145
146             policyType.setHighestVersion(true);
147
148             policyType.setVersion(TypeUtils.FIRST_CERTIFIED_VERSION_VERSION);
149         }
150         return policyType;
151     }
152
153 }