2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2019-2021 AT&T Intellectual Property. All rights reserved.
6 * Modifications Copyright (C) 2019-2021 Nordix Foundation.
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.
20 * SPDX-License-Identifier: Apache-2.0
21 * ============LICENSE_END=========================================================
24 package org.onap.policy.api.main.startstop;
26 import java.util.LinkedHashMap;
27 import java.util.LinkedList;
28 import java.util.List;
29 import org.onap.policy.api.main.exception.PolicyApiException;
30 import org.onap.policy.api.main.parameters.ApiParameterGroup;
31 import org.onap.policy.common.utils.coder.CoderException;
32 import org.onap.policy.common.utils.coder.StandardYamlCoder;
33 import org.onap.policy.common.utils.resources.ResourceUtils;
34 import org.onap.policy.models.base.PfModelException;
35 import org.onap.policy.models.base.PfModelRuntimeException;
36 import org.onap.policy.models.provider.PolicyModelsProvider;
37 import org.onap.policy.models.provider.PolicyModelsProviderFactory;
38 import org.onap.policy.models.tosca.authorative.concepts.ToscaEntityFilter;
39 import org.onap.policy.models.tosca.authorative.concepts.ToscaPolicyType;
40 import org.onap.policy.models.tosca.authorative.concepts.ToscaServiceTemplate;
41 import org.onap.policy.models.tosca.authorative.concepts.ToscaTopologyTemplate;
42 import org.slf4j.Logger;
43 import org.slf4j.LoggerFactory;
46 * This class creates initial policy types in the database.
48 * @author Chenfei Gao (cgao@research.att.com)
50 public class ApiDatabaseInitializer {
52 private static final Logger LOGGER = LoggerFactory.getLogger(ApiDatabaseInitializer.class);
54 private static final StandardYamlCoder coder = new StandardYamlCoder();
55 private PolicyModelsProviderFactory factory;
58 * Constructs the object.
60 public ApiDatabaseInitializer() {
61 factory = new PolicyModelsProviderFactory();
65 * Initializes database by preloading policy types and policies.
67 * @param apiParameterGroup the apiParameterGroup parameters
68 * @throws PolicyApiException in case of errors.
70 public void initializeApiDatabase(final ApiParameterGroup apiParameterGroup) throws PolicyApiException {
72 try (var databaseProvider =
73 factory.createPolicyModelsProvider(apiParameterGroup.getDatabaseProviderParameters())) {
75 if (alreadyExists(databaseProvider)) {
76 LOGGER.warn("DB already contains policy data - skipping preload");
80 var serviceTemplate = new ToscaServiceTemplate();
81 serviceTemplate.setDataTypes(new LinkedHashMap<>());
82 serviceTemplate.setPolicyTypes(new LinkedHashMap<>());
83 serviceTemplate.setToscaDefinitionsVersion("tosca_simple_yaml_1_1_0");
85 ToscaServiceTemplate createdPolicyTypes = preloadServiceTemplate(serviceTemplate,
86 apiParameterGroup.getPreloadPolicyTypes(), databaseProvider::createPolicyTypes);
87 preloadServiceTemplate(createdPolicyTypes, apiParameterGroup.getPreloadPolicies(),
88 databaseProvider::createPolicies);
89 } catch (final PolicyApiException | PfModelException | CoderException exp) {
90 throw new PolicyApiException(exp);
94 private boolean alreadyExists(PolicyModelsProvider databaseProvider) throws PfModelException {
96 ToscaServiceTemplate serviceTemplate =
97 databaseProvider.getFilteredPolicyTypes(ToscaEntityFilter.<ToscaPolicyType>builder().build());
98 if (!serviceTemplate.getPolicyTypes().isEmpty()) {
102 } catch (PfModelRuntimeException e) {
103 LOGGER.trace("DB does not yet contain policy types", e);
109 private ToscaServiceTemplate preloadServiceTemplate(ToscaServiceTemplate serviceTemplate, List<String> entities,
110 FunctionWithEx<ToscaServiceTemplate, ToscaServiceTemplate> getter)
111 throws PolicyApiException, CoderException, PfModelException {
113 for (String entity : entities) {
114 var entityAsStringYaml = ResourceUtils.getResourceAsString(entity);
115 if (entityAsStringYaml == null) {
116 LOGGER.warn("Preloading entity cannot be found: {}", entity);
120 ToscaServiceTemplate singleEntity = coder.decode(entityAsStringYaml, ToscaServiceTemplate.class);
121 if (singleEntity == null) {
122 throw new PolicyApiException("Error deserializaing entity from file: " + entity);
125 // Consolidate data types and policy types
126 if (singleEntity.getDataTypes() != null) {
127 serviceTemplate.getDataTypes().putAll(singleEntity.getDataTypes());
129 if (singleEntity.getPolicyTypes() != null) {
130 serviceTemplate.getPolicyTypes().putAll(singleEntity.getPolicyTypes());
133 // Consolidate policies
134 var topologyTemplate = singleEntity.getToscaTopologyTemplate();
135 if (topologyTemplate != null && topologyTemplate.getPolicies() != null) {
136 serviceTemplate.setToscaTopologyTemplate(new ToscaTopologyTemplate());
137 serviceTemplate.getToscaTopologyTemplate().setPolicies(new LinkedList<>());
138 serviceTemplate.getToscaTopologyTemplate().getPolicies()
139 .addAll(singleEntity.getToscaTopologyTemplate().getPolicies());
142 // Preload the specified entities
143 ToscaServiceTemplate createdServiceTemplate = getter.apply(serviceTemplate);
144 LOGGER.debug("Created initial tosca service template in DB - {}", createdServiceTemplate);
145 return createdServiceTemplate;
149 protected interface FunctionWithEx<T, R> {
150 public R apply(T value) throws PfModelException;