2 * ============LICENSE_START=======================================================
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
11 * http://www.apache.org/licenses/LICENSE-2.0
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=========================================================
21 package org.openecomp.sdc.be.model.jsonjanusgraph.operations;
23 import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
24 import org.openecomp.sdc.be.dao.jsongraph.JanusGraphDao;
25 import org.openecomp.sdc.be.dao.jsongraph.types.JsonParseFlagEnum;
26 import org.openecomp.sdc.be.dao.utils.MapUtil;
27 import org.openecomp.sdc.be.datatypes.elements.PropertyDataDefinition;
28 import org.openecomp.sdc.be.model.Component;
29 import org.openecomp.sdc.be.model.PolicyDefinition;
30 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
31 import org.openecomp.sdc.be.model.operations.impl.DaoStatusConverter;
32 import org.openecomp.sdc.common.log.wrappers.Logger;
34 import java.util.ArrayList;
35 import java.util.List;
38 @org.springframework.stereotype.Component
39 public class PolicyOperation {
41 private static final Logger log = Logger.getLogger(PolicyOperation.class.getName());
42 private TopologyTemplateOperation topologyTemplateOperation;
43 private JanusGraphDao janusGraphDao;
45 public PolicyOperation(TopologyTemplateOperation topologyTemplateOperation, JanusGraphDao janusGraphDao) {
46 this.topologyTemplateOperation = topologyTemplateOperation;
47 this.janusGraphDao = janusGraphDao;
51 * updates a list of policy properties by overriding the existing ones with the same name
52 * @param containerComponent the container of the policy of which its properties are to be updated
53 * @param policyId the id of the policy of which its properties are to be updated
54 * @param propertiesToUpdate the policy properties to update
55 * @return the update operation status
57 public StorageOperationStatus updatePolicyProperties(Component containerComponent, String policyId, 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);
65 private StorageOperationStatus updatePolicyProperties(GraphVertex container, PolicyDefinition policy, List<PropertyDataDefinition> propertiesToUpdate) {
66 List<PropertyDataDefinition> policyProperties = policy.getProperties();
67 List<PropertyDataDefinition> updatedPolicyProperties = updatePolicyProperties(policyProperties, propertiesToUpdate);
68 policy.setProperties(updatedPolicyProperties);
69 return topologyTemplateOperation.updatePolicyOfToscaElement(container, policy);
72 private List<PropertyDataDefinition> updatePolicyProperties(List<PropertyDataDefinition> currentPolicyProperties, List<PropertyDataDefinition> toBeUpdatedProperties) {
73 Map<String, PropertyDataDefinition> currPropsByName = MapUtil.toMap(currentPolicyProperties, PropertyDataDefinition::getName);
74 overrideCurrentPropertiesWithUpdatedProperties(currPropsByName, toBeUpdatedProperties);
75 return new ArrayList<>(currPropsByName.values());
78 private void overrideCurrentPropertiesWithUpdatedProperties(Map<String, PropertyDataDefinition> currPropsByName, List<PropertyDataDefinition> toBeUpdatedProperties) {
79 toBeUpdatedProperties.forEach(prop -> currPropsByName.put(prop.getName(), prop));