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.function.Consumer;
  26 import org.apache.commons.lang3.StringUtils;
  27 import org.apache.commons.lang3.tuple.ImmutablePair;
  28 import org.openecomp.sdc.be.components.impl.CommonImportManager.ElementTypeEnum;
  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.PolicyTypeDefinition;
  36 import org.openecomp.sdc.be.model.jsonjanusgraph.operations.ToscaOperationFacade;
  37 import org.openecomp.sdc.be.model.operations.api.IPolicyTypeOperation;
  38 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
  39 import org.openecomp.sdc.be.model.operations.impl.GroupOperation;
  40 import org.openecomp.sdc.be.model.operations.impl.GroupTypeOperation;
  41 import org.openecomp.sdc.be.model.operations.impl.UniqueIdBuilder;
  42 import org.openecomp.sdc.be.utils.TypeUtils;
  43 import org.openecomp.sdc.exception.ResponseFormat;
  44 import org.springframework.stereotype.Component;
  45
  46 @Component("policyTypeImportManager")
  47 public class PolicyTypeImportManager {
  48
  49     private final IPolicyTypeOperation policyTypeOperation;
  50     private final ComponentsUtils componentsUtils;
  51     private final GroupOperation groupOperation;
  52     private final ToscaOperationFacade toscaOperationFacade;
  53     private final CommonImportManager commonImportManager;
  54     private final GroupTypeOperation groupTypeOperation;
  55
  56     public PolicyTypeImportManager(IPolicyTypeOperation policyTypeOperation, ComponentsUtils componentsUtils, GroupOperation groupOperation,
  57                                    ToscaOperationFacade toscaOperationFacade, CommonImportManager commonImportManager,
  58                                    GroupTypeOperation groupTypeOperation) {
  59         this.policyTypeOperation = policyTypeOperation;
  60         this.componentsUtils = componentsUtils;
  61         this.groupOperation = groupOperation;
  62         this.toscaOperationFacade = toscaOperationFacade;
  63         this.commonImportManager = commonImportManager;
  64         this.groupTypeOperation = groupTypeOperation;
  65     }
  66
  67     public Either<List<ImmutablePair<PolicyTypeDefinition, Boolean>>, ResponseFormat> createPolicyTypes(ToscaTypeImportData toscaTypeImportData, String modelName) {
  68         return commonImportManager.createElementTypes(toscaTypeImportData, this::createPolicyTypesFromYml, this::upsertPolicyTypesByDao, modelName);
  69     }
  70
  71     private Either<List<PolicyTypeDefinition>, ActionStatus> createPolicyTypesFromYml(String policyTypesYml, String modelName) {
  72         Either<List<PolicyTypeDefinition>, ActionStatus> policyTypes = commonImportManager.createElementTypesFromYml(policyTypesYml, this::createPolicyType);
  73         if (policyTypes.isLeft() && StringUtils.isNotEmpty(modelName)){
  74             policyTypes.left().value().forEach(policyType -> policyType.setModel(modelName));
  75         }
  76         return policyTypes;
  77     }
  78
  79     private Either<List<ImmutablePair<PolicyTypeDefinition, Boolean>>, ResponseFormat> upsertPolicyTypesByDao(
  80         List<PolicyTypeDefinition> policyTypesToCreate) {
  81         return commonImportManager.createElementTypesByDao(policyTypesToCreate, this::validatePolicyType,
  82             policyType -> new ImmutablePair<>(ElementTypeEnum.POLICY_TYPE, UniqueIdBuilder.buildPolicyTypeUid(policyType.getModel(),
  83                 policyType.getType(), policyType.getVersion(), NodeTypeEnum.PolicyType.getName()).toLowerCase()), policyTypeOperation::getLatestPolicyTypeByType,
  84             policyTypeOperation::addPolicyType, this::updatePolicyType);
  85     }
  86
  87     private Either<PolicyTypeDefinition, StorageOperationStatus> updatePolicyType(PolicyTypeDefinition newPolicyType,
  88                                                                                   PolicyTypeDefinition oldPolicyType) {
  89         if (PolicyTypeImportUtils.isPolicyTypesEquals(newPolicyType, oldPolicyType)) {
  90             return policyTypeAlreadyExists();
  91         }
  92         return policyTypeOperation.updatePolicyType(newPolicyType, oldPolicyType);
  93     }
  94
  95     private Either<PolicyTypeDefinition, StorageOperationStatus> policyTypeAlreadyExists() {
  96         return Either.right(StorageOperationStatus.OK);
  97     }
  98
  99     private Either<ActionStatus, ResponseFormat> validatePolicyType(PolicyTypeDefinition policyType) {
 100         Either<ActionStatus, ResponseFormat> result = Either.left(ActionStatus.OK);
 101         if (policyType.getTargets() != null) {
 102             if (policyType.getTargets().isEmpty()) {
 103                 ResponseFormat responseFormat = componentsUtils.getResponseFormat(ActionStatus.TARGETS_EMPTY, policyType.getType());
 104                 result = Either.right(responseFormat);
 105             }
 106             if (result.isLeft()) {
 107                 for (String targetId : policyType.getTargets()) {
 108                     boolean isValid = toscaOperationFacade.getLatestByToscaResourceName(targetId).isLeft();
 109                     if (!isValid) { // check if it is a groupType
 110                         final Either<GroupTypeDefinition, StorageOperationStatus> groupTypeFound = groupTypeOperation
 111                             .getLatestGroupTypeByType(targetId, false);
 112                         isValid = groupTypeFound.isLeft() && !groupTypeFound.left().value().isEmpty();
 113                     }
 114                     if (!isValid) {
 115                         isValid = groupOperation.isGroupExist(targetId, false);
 116                     }
 117                     if (!isValid) {
 118                         ResponseFormat responseFormat = componentsUtils
 119                             .getResponseFormat(ActionStatus.TARGETS_NON_VALID, policyType.getType(), targetId);
 120                         result = Either.right(responseFormat);
 121                         break;
 122                     }
 123                 }
 124             }
 125         }
 126         return result;
 127     }
 128
 129     private PolicyTypeDefinition createPolicyType(String groupTypeName, Map<String, Object> toscaJson) {
 130         PolicyTypeDefinition policyType = new PolicyTypeDefinition();
 131         if (toscaJson != null) {
 132             // Description
 133             final Consumer<String> descriptionSetter = policyType::setDescription;
 134             commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.DESCRIPTION.getElementName(), descriptionSetter);
 135             // Derived From
 136             final Consumer<String> derivedFromSetter = policyType::setDerivedFrom;
 137             commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.DERIVED_FROM.getElementName(), derivedFromSetter);
 138             // Properties
 139             CommonImportManager.setProperties(toscaJson, policyType::setProperties);
 140             // Metadata
 141             final Consumer<Map<String, String>> metadataSetter = policyType::setMetadata;
 142             commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.METADATA.getElementName(), metadataSetter);
 143             // Targets
 144             final Consumer<List<String>> targetsSetter = policyType::setTargets;
 145             commonImportManager.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.TARGETS.getElementName(), targetsSetter);
 146             policyType.setType(groupTypeName);
 147             policyType.setHighestVersion(true);
 148             policyType.setVersion(TypeUtils.getFirstCertifiedVersionVersion());
 149         }
 150         return policyType;
 151     }
 152 }