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.CommonImportManager.ElementTypeEnum;
30 import org.openecomp.sdc.be.components.impl.model.ToscaTypeImportData;
31 import org.openecomp.sdc.be.components.impl.utils.PolicyTypeImportUtils;
32 import org.openecomp.sdc.be.dao.api.ActionStatus;
33 import org.openecomp.sdc.be.datatypes.enums.NodeTypeEnum;
34 import org.openecomp.sdc.be.impl.ComponentsUtils;
35 import org.openecomp.sdc.be.model.GroupTypeDefinition;
36 import org.openecomp.sdc.be.model.Model;
37 import org.openecomp.sdc.be.model.PolicyTypeDefinition;
38 import org.openecomp.sdc.be.model.jsonjanusgraph.operations.ToscaOperationFacade;
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(ToscaTypeImportData toscaTypeImportData, String modelName) {
73 return commonImportManager.createElementTypes(toscaTypeImportData, this::createPolicyTypesFromYml, this::upsertPolicyTypesByDao, modelName);
76 private Either<List<PolicyTypeDefinition>, ActionStatus> createPolicyTypesFromYml(String policyTypesYml, String modelName) {
77 Either<List<PolicyTypeDefinition>, ActionStatus> policyTypes = commonImportManager.createElementTypesFromYml(policyTypesYml, this::createPolicyType);
78 if (policyTypes.isLeft() && StringUtils.isNotEmpty(modelName)){
79 final Optional<Model> modelOptional = modelOperation.findModelByName(modelName);
80 if (modelOptional.isPresent()) {
81 policyTypes.left().value().forEach(policyType -> policyType.setModel(modelName));
84 return Either.right(ActionStatus.INVALID_MODEL);
89 private Either<List<ImmutablePair<PolicyTypeDefinition, Boolean>>, ResponseFormat> upsertPolicyTypesByDao(
90 List<PolicyTypeDefinition> policyTypesToCreate, String modelName) {
91 return commonImportManager.createElementTypesWithVersionByDao(policyTypesToCreate, this::validatePolicyType,
92 policyType -> new ImmutablePair<>(ElementTypeEnum.POLICY_TYPE, UniqueIdBuilder.buildPolicyTypeUid(policyType.getModel(),
93 policyType.getType(), policyType.getVersion(), NodeTypeEnum.PolicyType.getName()).toLowerCase()), policyTypeOperation::getLatestPolicyTypeByType,
94 policyTypeOperation::addPolicyType, this::updatePolicyType, modelName);
97 private Either<PolicyTypeDefinition, StorageOperationStatus> updatePolicyType(PolicyTypeDefinition newPolicyType,
98 PolicyTypeDefinition oldPolicyType) {
99 if (PolicyTypeImportUtils.isPolicyTypesEquals(newPolicyType, oldPolicyType)) {
100 return policyTypeAlreadyExists();
102 return policyTypeOperation.updatePolicyType(newPolicyType, oldPolicyType);
105 private Either<PolicyTypeDefinition, StorageOperationStatus> policyTypeAlreadyExists() {
106 return Either.right(StorageOperationStatus.OK);
109 private Either<ActionStatus, ResponseFormat> validatePolicyType(PolicyTypeDefinition policyType) {
110 Either<ActionStatus, ResponseFormat> result = Either.left(ActionStatus.OK);
111 if (policyType.getTargets() != null) {
112 if (policyType.getTargets().isEmpty()) {
113 ResponseFormat responseFormat = componentsUtils.getResponseFormat(ActionStatus.TARGETS_EMPTY, policyType.getType());
114 result = Either.right(responseFormat);
116 if (result.isLeft()) {
117 for (String targetId : policyType.getTargets()) {
118 boolean isValid = toscaOperationFacade.getLatestByToscaResourceName(targetId, policyType.getModel()).isLeft();
119 if (!isValid) { // check if it is a groupType
120 final Either<GroupTypeDefinition, StorageOperationStatus> groupTypeFound = groupTypeOperation
121 .getLatestGroupTypeByType(targetId, policyType.getModel(), false);
122 isValid = groupTypeFound.isLeft() && !groupTypeFound.left().value().isEmpty();
125 isValid = groupOperation.isGroupExist(targetId, false);
128 ResponseFormat responseFormat = componentsUtils
129 .getResponseFormat(ActionStatus.TARGETS_NON_VALID, policyType.getType(), targetId);
130 result = Either.right(responseFormat);
139 private PolicyTypeDefinition createPolicyType(String groupTypeName, Map<String, Object> toscaJson) {
140 PolicyTypeDefinition policyType = new PolicyTypeDefinition();
141 if (toscaJson != null) {
143 final Consumer<String> descriptionSetter = policyType::setDescription;
144 commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.DESCRIPTION.getElementName(), descriptionSetter);
146 final Consumer<String> derivedFromSetter = policyType::setDerivedFrom;
147 commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.DERIVED_FROM.getElementName(), derivedFromSetter);
149 CommonImportManager.setProperties(toscaJson, policyType::setProperties);
151 final Consumer<Map<String, String>> metadataSetter = policyType::setMetadata;
152 commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.METADATA.getElementName(), metadataSetter);
154 final Consumer<List<String>> targetsSetter = policyType::setTargets;
155 commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.TARGETS.getElementName(), targetsSetter);
156 policyType.setType(groupTypeName);
157 policyType.setHighestVersion(true);
158 policyType.setVersion(TypeUtils.getFirstCertifiedVersionVersion());