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.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;
  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(ToscaTypeImportData toscaTypeImportData, String modelName) {
  73         return commonImportManager.createElementTypes(toscaTypeImportData, this::createPolicyTypesFromYml, this::upsertPolicyTypesByDao, modelName);
  74     }
  75
  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));
  82                 return policyTypes;
  83             }
  84             return Either.right(ActionStatus.INVALID_MODEL);
  85         }
  86         return policyTypes;
  87     }
  88
  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);
  95     }
  96
  97     private Either<PolicyTypeDefinition, StorageOperationStatus> updatePolicyType(PolicyTypeDefinition newPolicyType,
  98                                                                                   PolicyTypeDefinition oldPolicyType) {
  99         if (PolicyTypeImportUtils.isPolicyTypesEquals(newPolicyType, oldPolicyType)) {
 100             return policyTypeAlreadyExists();
 101         }
 102         return policyTypeOperation.updatePolicyType(newPolicyType, oldPolicyType);
 103     }
 104
 105     private Either<PolicyTypeDefinition, StorageOperationStatus> policyTypeAlreadyExists() {
 106         return Either.right(StorageOperationStatus.OK);
 107     }
 108
 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);
 115             }
 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();
 123                     }
 124                     if (!isValid) {
 125                         isValid = groupOperation.isGroupExist(targetId, false);
 126                     }
 127                     if (!isValid) {
 128                         ResponseFormat responseFormat = componentsUtils
 129                             .getResponseFormat(ActionStatus.TARGETS_NON_VALID, policyType.getType(), targetId);
 130                         result = Either.right(responseFormat);
 131                         break;
 132                     }
 133                 }
 134             }
 135         }
 136         return result;
 137     }
 138
 139     private PolicyTypeDefinition createPolicyType(String groupTypeName, Map<String, Object> toscaJson) {
 140         PolicyTypeDefinition policyType = new PolicyTypeDefinition();
 141         if (toscaJson != null) {
 142             // Description
 143             final Consumer<String> descriptionSetter = policyType::setDescription;
 144             commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.DESCRIPTION.getElementName(), descriptionSetter);
 145             // Derived From
 146             final Consumer<String> derivedFromSetter = policyType::setDerivedFrom;
 147             commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.DERIVED_FROM.getElementName(), derivedFromSetter);
 148             // Properties
 149             CommonImportManager.setProperties(toscaJson, policyType::setProperties);
 150             // Metadata
 151             final Consumer<Map<String, String>> metadataSetter = policyType::setMetadata;
 152             commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.METADATA.getElementName(), metadataSetter);
 153             // Targets
 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());
 159         }
 160         return policyType;
 161     }
 162 }