2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2020 AT&T Intellectual Property. All rights
7 * ================================================================================
8 * Licensed under the Apache License, Version 2.0 (the "License");
9 * you may not use this file except in compliance with the License.
10 * You may obtain a copy of the License at
12 * http://www.apache.org/licenses/LICENSE-2.0
14 * Unless required by applicable law or agreed to in writing, software
15 * distributed under the License is distributed on an "AS IS" BASIS,
16 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17 * See the License for the specific language governing permissions and
18 * limitations under the License.
19 * ============LICENSE_END============================================
20 * ===================================================================
24 package org.onap.clamp.loop.template;
26 import com.google.gson.JsonArray;
27 import com.google.gson.JsonObject;
28 import java.util.ArrayList;
29 import java.util.Collections;
30 import java.util.List;
31 import org.onap.clamp.clds.tosca.ToscaSchemaConstants;
32 import org.onap.clamp.clds.tosca.ToscaYamlToJsonConvertor;
33 import org.onap.clamp.policy.pdpgroup.PdpGroup;
34 import org.onap.clamp.util.SemanticVersioning;
35 import org.springframework.beans.factory.annotation.Autowired;
36 import org.springframework.stereotype.Service;
37 import org.springframework.transaction.annotation.Propagation;
38 import org.springframework.transaction.annotation.Transactional;
41 public class PolicyModelsService {
42 private final PolicyModelsRepository policyModelsRepository;
43 private ToscaYamlToJsonConvertor toscaYamlToJsonConvertor;
46 public PolicyModelsService(PolicyModelsRepository policyModelrepo,
47 ToscaYamlToJsonConvertor convertor) {
48 policyModelsRepository = policyModelrepo;
49 toscaYamlToJsonConvertor = convertor;
53 * Save or Update Policy Model.
55 * @param policyModel The policyModel
56 * @return The Policy Model
58 public PolicyModel saveOrUpdatePolicyModel(PolicyModel policyModel) {
59 return policyModelsRepository.saveAndFlush(policyModel);
63 * Verify whether Policy Model exist by ID.
65 * @param policyModelId The policyModel Id
66 * @return The flag indicates whether Policy Model exist
68 public boolean existsById(PolicyModelId policyModelId) {
69 return policyModelsRepository.existsById(policyModelId);
73 * Creates or updates the Tosca Policy Model.
75 * @param policyModelTosca The Policymodel object
76 * @return The Policy Model created
78 public PolicyModel createNewPolicyModelFromTosca(String policyModelTosca) {
79 JsonObject jsonObject = toscaYamlToJsonConvertor.validateAndConvertToJson(policyModelTosca);
80 String policyModelTypeFromTosca = toscaYamlToJsonConvertor.getValueFromMetadata(jsonObject,
81 ToscaSchemaConstants.METADATA_POLICY_MODEL_TYPE);
82 Iterable<PolicyModel> models = getAllPolicyModelsByType(policyModelTypeFromTosca);
83 Collections.sort((List<PolicyModel>) models);
84 PolicyModel newPolicyModel = new PolicyModel(policyModelTypeFromTosca, policyModelTosca,
85 SemanticVersioning.incrementMajorVersion(((ArrayList) models).isEmpty() ? null
86 : ((ArrayList<PolicyModel>) models).get(0).getVersion()),
87 toscaYamlToJsonConvertor.getValueFromMetadata(jsonObject,
88 ToscaSchemaConstants.METADATA_ACRONYM));
89 return saveOrUpdatePolicyModel(newPolicyModel);
93 * Update an existing Tosca Policy Model.
95 * @param policyModelType The policy Model type in Tosca yaml
96 * @param policyModelVersion The policy Version to update
97 * @param policyModelTosca The Policy Model tosca
98 * @return The Policy Model updated
100 public PolicyModel updatePolicyModelTosca(String policyModelType, String policyModelVersion,
101 String policyModelTosca) {
102 JsonObject jsonObject = toscaYamlToJsonConvertor.validateAndConvertToJson(policyModelTosca);
103 PolicyModel thePolicyModel = getPolicyModelByType(policyModelType, policyModelVersion);
104 thePolicyModel.setPolicyAcronym(toscaYamlToJsonConvertor.getValueFromMetadata(jsonObject,
105 ToscaSchemaConstants.METADATA_ACRONYM));
106 thePolicyModel.setPolicyModelTosca(policyModelTosca);
107 return saveOrUpdatePolicyModel(thePolicyModel);
110 public List<String> getAllPolicyModelTypes() {
111 return policyModelsRepository.getAllPolicyModelType();
114 public Iterable<PolicyModel> getAllPolicyModels() {
115 return policyModelsRepository.findAll();
118 public PolicyModel getPolicyModel(String type, String version) {
119 return policyModelsRepository.findById(new PolicyModelId(type, version)).orElse(null);
122 public Iterable<PolicyModel> getAllPolicyModelsByType(String type) {
123 return policyModelsRepository.findByPolicyModelType(type);
126 public PolicyModel getPolicyModelByType(String type, String version) {
127 return policyModelsRepository.findById(new PolicyModelId(type, version)).orElse(null);
131 * Retrieves the Tosca model Yaml string.
133 * @param type The Policy Model Type
134 * @param version The policy model version
135 * @return The Tosca model Yaml string
137 public String getPolicyModelTosca(String type, String version) {
138 return policyModelsRepository.findById(new PolicyModelId(type, version))
139 .orElse(new PolicyModel()).getPolicyModelTosca();
143 * This method creates an PolicyModel in Db if it does not exist.
145 * @param policyModel The policyModel to save
147 @Transactional(propagation = Propagation.REQUIRES_NEW)
148 public PolicyModel savePolicyModelInNewTransaction(PolicyModel policyModel) {
149 return policyModelsRepository.saveAndFlush(policyModel);
153 * Update the Pdp Group info in Policy Model DB.
155 * @param pdpGroupList The list of Pdp Group info received from Policy Engine
157 public void updatePdpGroupInfo(List<PdpGroup> pdpGroupList) {
158 List<PolicyModel> policyModelList = policyModelsRepository.findAll();
159 for (PolicyModel policyModel : policyModelList) {
160 JsonArray supportedPdpGroups = new JsonArray();
161 for (PdpGroup pdpGroup : pdpGroupList) {
162 JsonObject supportedPdpGroup = pdpGroup.getSupportedSubgroups(
163 policyModel.getPolicyModelType(), policyModel.getVersion());
164 if (supportedPdpGroup != null) {
165 supportedPdpGroups.add(supportedPdpGroup);
169 if (supportedPdpGroups.size() > 0) {
170 JsonObject supportedPdpJson = new JsonObject();
171 supportedPdpJson.add("supportedPdpGroups", supportedPdpGroups);
172 policyModel.setPolicyPdpGroup(supportedPdpJson);
173 policyModelsRepository.saveAndFlush(policyModel);