catalog-be/src/main/java/org/openecomp/sdc/be/components/impl/PolicyTypeBusinessLogic.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * SDC
   4  * ================================================================================
   5  * Copyright (C) 2019 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
  21 package org.openecomp.sdc.be.components.impl;
  22
  23 import static java.util.Collections.emptySet;
  24
  25 import java.util.Collections;
  26 import java.util.List;
  27 import java.util.Map;
  28 import java.util.Set;
  29 import org.apache.commons.lang3.StringUtils;
  30 import org.openecomp.sdc.be.components.impl.exceptions.ByActionStatusComponentException;
  31 import org.openecomp.sdc.be.components.impl.model.ToscaTypeImportData;
  32 import org.openecomp.sdc.be.components.validation.UserValidations;
  33 import org.openecomp.sdc.be.config.ConfigurationManager;
  34 import org.openecomp.sdc.be.dao.api.ActionStatus;
  35 import org.openecomp.sdc.be.dao.janusgraph.JanusGraphDao;
  36 import org.openecomp.sdc.be.model.PolicyTypeDefinition;
  37 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
  38 import org.openecomp.sdc.be.model.operations.impl.PolicyTypeOperation;
  39 import org.springframework.beans.factory.annotation.Autowired;
  40 import org.springframework.stereotype.Component;
  41
  42 @Component("policyTypeBusinessLogic")
  43 public class PolicyTypeBusinessLogic {
  44
  45     private final PolicyTypeOperation policyTypeOperation;
  46     private final JanusGraphDao janusGraphDao;
  47     private final UserValidations userValidations;
  48     private final PolicyTypeImportManager policyTypeImportManager;
  49
  50     @Autowired
  51     public PolicyTypeBusinessLogic(final PolicyTypeOperation policyTypeOperation, final JanusGraphDao janusGraphDao,
  52             final UserValidations userValidations, final PolicyTypeImportManager policyTypeImportManager) {
  53         this.policyTypeOperation = policyTypeOperation;
  54         this.janusGraphDao = janusGraphDao;
  55         this.userValidations = userValidations;
  56         this.policyTypeImportManager = policyTypeImportManager;
  57     }
  58
  59     public List<PolicyTypeDefinition> getAllPolicyTypes(String userId, String internalComponentType, String modelName) {
  60         Set<String> excludedPolicyTypes = getExcludedPolicyTypes(internalComponentType);
  61         userValidations.validateUserExists(userId);
  62         return getPolicyTypes(excludedPolicyTypes, modelName);
  63     }
  64
  65     public PolicyTypeDefinition getLatestPolicyTypeByType(String policyTypeName, String modelName) {
  66         return policyTypeOperation.getLatestPolicyTypeByType(policyTypeName, modelName).left()
  67                 .on(e -> failOnPolicyType(e, policyTypeName));
  68     }
  69
  70     public Set<String> getExcludedPolicyTypes(String internalComponentType) {
  71         if (StringUtils.isEmpty(internalComponentType)) {
  72             return emptySet();
  73         }
  74         Map<String, Set<String>> excludedPolicyTypesMapping =
  75                 ConfigurationManager.getConfigurationManager().getConfiguration().getExcludedPolicyTypesMapping();
  76         Set<String> excludedTypes = excludedPolicyTypesMapping.get(internalComponentType);
  77         return excludedTypes == null ? emptySet() : excludedTypes;
  78     }
  79
  80     /**
  81      * Create policy types based on yaml string.
  82      * @param policyTypesYaml the policy types to create in yaml format. It can contain multiple entries.
  83      * @param model Model name to associate with policy type
  84      * @param includeToModelDefaultImports Add policy type entry to default imports for model
  85      */
  86     public void createPolicyTypeFromYaml(final String policyTypesYaml, final String model, final boolean includeToModelDefaultImports) {
  87         policyTypeImportManager
  88             .createPolicyTypes(new ToscaTypeImportData(policyTypesYaml, Collections.emptyMap()), model, includeToModelDefaultImports);
  89     }
  90
  91     private List<PolicyTypeDefinition> getPolicyTypes(Set<String> excludedTypes, String modelName) {
  92         return policyTypeOperation.getAllPolicyTypes(excludedTypes, modelName);
  93     }
  94
  95     private PolicyTypeDefinition failOnPolicyType(StorageOperationStatus status, String policyType) {
  96         janusGraphDao.rollback();
  97         if (status == StorageOperationStatus.INVALID_ID) {
  98             throw new ByActionStatusComponentException(ActionStatus.POLICY_TYPE_IS_INVALID, policyType);
  99         } else {
 100             throw new ByActionStatusComponentException(ActionStatus.GENERAL_ERROR);
 101         }
 102     }
 103 }