2 * ============LICENSE_START=======================================================
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
11 * http://www.apache.org/licenses/LICENSE-2.0
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=========================================================
20 package org.openecomp.sdc.be.components.impl;
22 import fj.data.Either;
23 import java.util.List;
25 import java.util.Optional;
26 import java.util.function.Consumer;
27 import org.apache.commons.lang3.StringUtils;
28 import org.apache.commons.lang3.tuple.ImmutablePair;
29 import org.openecomp.sdc.be.components.impl.model.ToscaTypeImportData;
30 import org.openecomp.sdc.be.components.impl.utils.PolicyTypeImportUtils;
31 import org.openecomp.sdc.be.dao.api.ActionStatus;
32 import org.openecomp.sdc.be.datatypes.enums.NodeTypeEnum;
33 import org.openecomp.sdc.be.impl.ComponentsUtils;
34 import org.openecomp.sdc.be.model.GroupTypeDefinition;
35 import org.openecomp.sdc.be.model.Model;
36 import org.openecomp.sdc.be.model.PolicyTypeDefinition;
37 import org.openecomp.sdc.be.model.jsonjanusgraph.operations.ToscaOperationFacade;
38 import org.openecomp.sdc.be.model.normatives.ElementTypeEnum;
39 import org.openecomp.sdc.be.model.operations.api.IPolicyTypeOperation;
40 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
41 import org.openecomp.sdc.be.model.operations.impl.GroupOperation;
42 import org.openecomp.sdc.be.model.operations.impl.GroupTypeOperation;
43 import org.openecomp.sdc.be.model.operations.impl.ModelOperation;
44 import org.openecomp.sdc.be.model.operations.impl.UniqueIdBuilder;
45 import org.openecomp.sdc.be.utils.TypeUtils;
46 import org.openecomp.sdc.exception.ResponseFormat;
47 import org.springframework.stereotype.Component;
49 @Component("policyTypeImportManager")
50 public class PolicyTypeImportManager {
52 private final IPolicyTypeOperation policyTypeOperation;
53 private final ComponentsUtils componentsUtils;
54 private final GroupOperation groupOperation;
55 private final ToscaOperationFacade toscaOperationFacade;
56 private final CommonImportManager commonImportManager;
57 private final GroupTypeOperation groupTypeOperation;
58 private final ModelOperation modelOperation;
60 public PolicyTypeImportManager(IPolicyTypeOperation policyTypeOperation, ComponentsUtils componentsUtils, GroupOperation groupOperation,
61 ToscaOperationFacade toscaOperationFacade, CommonImportManager commonImportManager,
62 GroupTypeOperation groupTypeOperation, ModelOperation modelOperation) {
63 this.policyTypeOperation = policyTypeOperation;
64 this.componentsUtils = componentsUtils;
65 this.groupOperation = groupOperation;
66 this.toscaOperationFacade = toscaOperationFacade;
67 this.commonImportManager = commonImportManager;
68 this.groupTypeOperation = groupTypeOperation;
69 this.modelOperation = modelOperation;
72 public Either<List<ImmutablePair<PolicyTypeDefinition, Boolean>>, ResponseFormat> createPolicyTypes(final ToscaTypeImportData toscaTypeImportData,
73 final String modelName,
74 final boolean includeToModelDefaultImports) {
75 final Either<List<ImmutablePair<PolicyTypeDefinition, Boolean>>, ResponseFormat> elementTypes = commonImportManager.createElementTypes(
76 toscaTypeImportData, this::createPolicyTypesFromYml, this::upsertPolicyTypesByDao, modelName);
77 if (includeToModelDefaultImports && StringUtils.isNotEmpty(modelName)) {
78 commonImportManager.addTypesToDefaultImports(ElementTypeEnum.POLICY_TYPE, toscaTypeImportData.getToscaTypesYml(), modelName);
83 private Either<List<PolicyTypeDefinition>, ActionStatus> createPolicyTypesFromYml(String policyTypesYml, String modelName) {
84 Either<List<PolicyTypeDefinition>, ActionStatus> policyTypes = commonImportManager.createElementTypesFromYml(policyTypesYml,
85 this::createPolicyType);
86 if (policyTypes.isLeft() && StringUtils.isNotEmpty(modelName)) {
87 final Optional<Model> modelOptional = modelOperation.findModelByName(modelName);
88 if (modelOptional.isPresent()) {
89 policyTypes.left().value().forEach(policyType -> policyType.setModel(modelName));
92 return Either.right(ActionStatus.INVALID_MODEL);
97 private Either<List<ImmutablePair<PolicyTypeDefinition, Boolean>>, ResponseFormat> upsertPolicyTypesByDao(
98 List<PolicyTypeDefinition> policyTypesToCreate, String modelName) {
99 return commonImportManager.createElementTypesWithVersionByDao(policyTypesToCreate, this::validatePolicyType,
100 policyType -> new ImmutablePair<>(ElementTypeEnum.POLICY_TYPE, UniqueIdBuilder.buildPolicyTypeUid(policyType.getModel(),
101 policyType.getType(), policyType.getVersion(), NodeTypeEnum.PolicyType.getName()).toLowerCase()),
102 policyTypeOperation::getLatestPolicyTypeByType,
103 policyTypeOperation::addPolicyType, this::updatePolicyType, modelName);
106 private Either<PolicyTypeDefinition, StorageOperationStatus> updatePolicyType(PolicyTypeDefinition newPolicyType,
107 PolicyTypeDefinition oldPolicyType) {
108 if (PolicyTypeImportUtils.isPolicyTypesEquals(newPolicyType, oldPolicyType)) {
109 return policyTypeAlreadyExists();
111 return policyTypeOperation.updatePolicyType(newPolicyType, oldPolicyType);
114 private Either<PolicyTypeDefinition, StorageOperationStatus> policyTypeAlreadyExists() {
115 return Either.right(StorageOperationStatus.OK);
118 private Either<ActionStatus, ResponseFormat> validatePolicyType(PolicyTypeDefinition policyType) {
119 Either<ActionStatus, ResponseFormat> result = Either.left(ActionStatus.OK);
120 if (policyType.getTargets() != null) {
121 if (policyType.getTargets().isEmpty()) {
122 ResponseFormat responseFormat = componentsUtils.getResponseFormat(ActionStatus.TARGETS_EMPTY, policyType.getType());
123 result = Either.right(responseFormat);
125 if (result.isLeft()) {
126 for (String targetId : policyType.getTargets()) {
127 boolean isValid = toscaOperationFacade.getLatestByToscaResourceName(targetId, policyType.getModel()).isLeft();
128 if (!isValid) { // check if it is a groupType
129 final Either<GroupTypeDefinition, StorageOperationStatus> groupTypeFound = groupTypeOperation
130 .getLatestGroupTypeByType(targetId, policyType.getModel(), false);
131 isValid = groupTypeFound.isLeft() && !groupTypeFound.left().value().isEmpty();
134 isValid = groupOperation.isGroupExist(targetId, false);
137 ResponseFormat responseFormat = componentsUtils
138 .getResponseFormat(ActionStatus.TARGETS_NON_VALID, policyType.getType(), targetId);
139 result = Either.right(responseFormat);
148 private PolicyTypeDefinition createPolicyType(String groupTypeName, Map<String, Object> toscaJson) {
149 PolicyTypeDefinition policyType = new PolicyTypeDefinition();
150 if (toscaJson != null) {
152 final Consumer<String> descriptionSetter = policyType::setDescription;
153 commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.DESCRIPTION.getElementName(), descriptionSetter);
155 final Consumer<String> derivedFromSetter = policyType::setDerivedFrom;
156 commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.DERIVED_FROM.getElementName(), derivedFromSetter);
158 CommonImportManager.setProperties(toscaJson, policyType::setProperties);
160 final Consumer<Map<String, String>> metadataSetter = policyType::setMetadata;
161 commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.METADATA.getElementName(), metadataSetter);
163 final Consumer<List<String>> targetsSetter = policyType::setTargets;
164 commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.TARGETS.getElementName(), targetsSetter);
165 policyType.setType(groupTypeName);
166 policyType.setHighestVersion(true);
167 policyType.setVersion(TypeUtils.getFirstCertifiedVersionVersion());