catalog-model/src/main/java/org/openecomp/sdc/be/model/jsonjanusgraph/operations/PolicyOperation.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 package org.openecomp.sdc.be.model.jsonjanusgraph.operations;
  21
  22 import java.util.ArrayList;
  23 import java.util.List;
  24 import java.util.Map;
  25 import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
  26 import org.openecomp.sdc.be.dao.janusgraph.JanusGraphDao;
  27 import org.openecomp.sdc.be.dao.jsongraph.types.JsonParseFlagEnum;
  28 import org.openecomp.sdc.be.dao.utils.MapUtil;
  29 import org.openecomp.sdc.be.datatypes.elements.PropertyDataDefinition;
  30 import org.openecomp.sdc.be.model.Component;
  31 import org.openecomp.sdc.be.model.PolicyDefinition;
  32 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
  33 import org.openecomp.sdc.be.model.operations.impl.DaoStatusConverter;
  34 import org.openecomp.sdc.common.log.wrappers.Logger;
  35
  36 @org.springframework.stereotype.Component
  37 public class PolicyOperation {
  38
  39     private static final Logger log = Logger.getLogger(PolicyOperation.class.getName());
  40     private TopologyTemplateOperation topologyTemplateOperation;
  41     private JanusGraphDao janusGraphDao;
  42
  43     public PolicyOperation(TopologyTemplateOperation topologyTemplateOperation, JanusGraphDao janusGraphDao) {
  44         this.topologyTemplateOperation = topologyTemplateOperation;
  45         this.janusGraphDao = janusGraphDao;
  46     }
  47
  48     /**
  49      * updates a list of policy properties by overriding the existing ones with the same name
  50      *
  51      * @param containerComponent the container of the policy of which its properties are to be updated
  52      * @param policyId           the id of the policy of which its properties are to be updated
  53      * @param propertiesToUpdate the policy properties to update
  54      * @return the update operation status
  55      */
  56     public StorageOperationStatus updatePolicyProperties(Component containerComponent, String policyId,
  57                                                          List<PropertyDataDefinition> propertiesToUpdate) {
  58         log.debug("#updatePolicyProperties - updating the properties of policy {} in component {}", policyId, containerComponent.getUniqueId());
  59         PolicyDefinition policy = containerComponent.getPolicyById(policyId);
  60         return janusGraphDao.getVertexById(containerComponent.getUniqueId(), JsonParseFlagEnum.NoParse)
  61             .either(containerVertex -> updatePolicyProperties(containerVertex, policy, propertiesToUpdate),
  62                 DaoStatusConverter::convertJanusGraphStatusToStorageStatus);
  63     }
  64
  65     private StorageOperationStatus updatePolicyProperties(GraphVertex container, PolicyDefinition policy,
  66                                                           List<PropertyDataDefinition> propertiesToUpdate) {
  67         List<PropertyDataDefinition> policyProperties = policy.getProperties();
  68         List<PropertyDataDefinition> updatedPolicyProperties = updatePolicyProperties(policyProperties, propertiesToUpdate);
  69         policy.setProperties(updatedPolicyProperties);
  70         return topologyTemplateOperation.updatePolicyOfToscaElement(container, policy);
  71     }
  72
  73     private List<PropertyDataDefinition> updatePolicyProperties(List<PropertyDataDefinition> currentPolicyProperties,
  74                                                                 List<PropertyDataDefinition> toBeUpdatedProperties) {
  75         Map<String, PropertyDataDefinition> currPropsByName = MapUtil.toMap(currentPolicyProperties, PropertyDataDefinition::getName);
  76         overrideCurrentPropertiesWithUpdatedProperties(currPropsByName, toBeUpdatedProperties);
  77         return new ArrayList<>(currPropsByName.values());
  78     }
  79
  80     private void overrideCurrentPropertiesWithUpdatedProperties(Map<String, PropertyDataDefinition> currPropsByName,
  81                                                                 List<PropertyDataDefinition> toBeUpdatedProperties) {
  82         toBeUpdatedProperties.forEach(prop -> currPropsByName.put(prop.getName(), prop));
  83     }
  84 }