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 package org.openecomp.sdc.be.components.impl;
  21
  22 import fj.data.Either;
  23 import java.util.List;
  24 import java.util.Map;
  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;
  48
  49 @Component("policyTypeImportManager")
  50 public class PolicyTypeImportManager {
  51
  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;
  59
  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;
  70     }
  71
  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 (elementTypes.isLeft() && includeToModelDefaultImports && StringUtils.isNotEmpty(modelName)) {
  78             commonImportManager.addTypesToDefaultImports(ElementTypeEnum.POLICY_TYPE, toscaTypeImportData.getToscaTypesYml(), modelName);
  79         }
  80         return elementTypes;
  81     }
  82
  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));
  90                 return policyTypes;
  91             }
  92             return Either.right(ActionStatus.INVALID_MODEL);
  93         }
  94         return policyTypes;
  95     }
  96
  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);
 104     }
 105
 106     private Either<PolicyTypeDefinition, StorageOperationStatus> updatePolicyType(PolicyTypeDefinition newPolicyType,
 107                                                                                   PolicyTypeDefinition oldPolicyType) {
 108         if (PolicyTypeImportUtils.isPolicyTypesEquals(newPolicyType, oldPolicyType)) {
 109             return policyTypeAlreadyExists();
 110         }
 111         return policyTypeOperation.updatePolicyType(newPolicyType, oldPolicyType);
 112     }
 113
 114     private Either<PolicyTypeDefinition, StorageOperationStatus> policyTypeAlreadyExists() {
 115         return Either.right(StorageOperationStatus.OK);
 116     }
 117
 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);
 124             }
 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();
 132                     }
 133                     if (!isValid) {
 134                         isValid = groupOperation.isGroupExist(targetId, false);
 135                     }
 136                     if (!isValid) {
 137                         ResponseFormat responseFormat = componentsUtils
 138                             .getResponseFormat(ActionStatus.TARGETS_NON_VALID, policyType.getType(), targetId);
 139                         result = Either.right(responseFormat);
 140                         break;
 141                     }
 142                 }
 143             }
 144         }
 145         return result;
 146     }
 147
 148     private PolicyTypeDefinition createPolicyType(String groupTypeName, Map<String, Object> toscaJson) {
 149         PolicyTypeDefinition policyType = new PolicyTypeDefinition();
 150         if (toscaJson != null) {
 151             // Description
 152             final Consumer<String> descriptionSetter = policyType::setDescription;
 153             commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.DESCRIPTION.getElementName(), descriptionSetter);
 154             // Derived From
 155             final Consumer<String> derivedFromSetter = policyType::setDerivedFrom;
 156             commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.DERIVED_FROM.getElementName(), derivedFromSetter);
 157             // Properties
 158             CommonImportManager.setProperties(toscaJson, policyType::setProperties);
 159             // Metadata
 160             final Consumer<Map<String, String>> metadataSetter = policyType::setMetadata;
 161             commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.METADATA.getElementName(), metadataSetter);
 162             // Targets
 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());
 168         }
 169         return policyType;
 170     }
 171 }