Update cm-data-subscriptions model
[cps.git] / cps-ncmp-service / src / main / java / org / onap / cps / ncmp / init / CmDataSubscriptionModelLoader.java
1 /*
2  *  ============LICENSE_START=======================================================
3  *  Copyright (C) 2024 Nordix Foundation
4  *  ================================================================================
5  *  Licensed under the Apache License, Version 2.0 (the "License");
6  *  you may not use this file except in compliance with the License.
7  *  You may obtain a copy of the License at
8  *
9  *        http://www.apache.org/licenses/LICENSE-2.0
10  *
11  *  Unless required by applicable law or agreed to in writing, software
12  *  distributed under the License is distributed on an "AS IS" BASIS,
13  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  *  See the License for the specific language governing permissions and
15  *  limitations under the License.
16  *
17  *  SPDX-License-Identifier: Apache-2.0
18  *  ============LICENSE_END=========================================================
19  */
20
21 package org.onap.cps.ncmp.init;
22
23 import static org.onap.cps.ncmp.api.impl.ncmppersistence.NcmpPersistence.NCMP_DATASPACE_NAME;
24 import static org.onap.cps.utils.ContentType.JSON;
25
26 import java.time.OffsetDateTime;
27 import lombok.extern.slf4j.Slf4j;
28 import org.onap.cps.api.CpsAnchorService;
29 import org.onap.cps.api.CpsDataService;
30 import org.onap.cps.api.CpsDataspaceService;
31 import org.onap.cps.api.CpsModuleService;
32 import org.onap.cps.ncmp.api.impl.exception.NcmpStartUpException;
33 import org.onap.cps.ncmp.api.impl.operations.DatastoreType;
34 import org.onap.cps.spi.exceptions.AlreadyDefinedException;
35 import org.springframework.beans.factory.annotation.Value;
36 import org.springframework.stereotype.Service;
37
38 @Slf4j
39 @Service
40 public class CmDataSubscriptionModelLoader extends AbstractModelLoader {
41
42     private static final String MODEL_FILENAME = "cm-data-subscriptions@2024-02-12.yang";
43     private static final String SCHEMASET_NAME = "cm-data-subscriptions";
44     private static final String ANCHOR_NAME = "cm-data-subscriptions";
45     private static final String REGISTRY_DATANODE_NAME = "datastores";
46
47     public CmDataSubscriptionModelLoader(final CpsDataspaceService cpsDataspaceService,
48                                          final CpsModuleService cpsModuleService,
49                                          final CpsAnchorService cpsAnchorService,
50                                          final CpsDataService cpsDataService) {
51         super(cpsDataspaceService, cpsModuleService, cpsAnchorService, cpsDataService);
52     }
53
54     @Value("${ncmp.model-loader.subscription:true}")
55     private boolean subscriptionModelLoaderEnabled;
56
57     @Override
58     public void onboardOrUpgradeModel() {
59         if (subscriptionModelLoaderEnabled) {
60             waitUntilDataspaceIsAvailable(NCMP_DATASPACE_NAME);
61             onboardSubscriptionModels();
62             log.info("Subscription Models onboarded successfully");
63         } else {
64             log.info("Subscription Model Loader is disabled");
65         }
66     }
67
68     private void onboardSubscriptionModels() {
69         createSchemaSet(NCMP_DATASPACE_NAME, SCHEMASET_NAME, MODEL_FILENAME);
70         createAnchor(NCMP_DATASPACE_NAME, SCHEMASET_NAME, ANCHOR_NAME);
71         createTopLevelDataNode(NCMP_DATASPACE_NAME, ANCHOR_NAME, REGISTRY_DATANODE_NAME);
72         createDatastore(DatastoreType.PASSTHROUGH_OPERATIONAL.getDatastoreName(),
73                 DatastoreType.PASSTHROUGH_RUNNING.getDatastoreName());
74     }
75
76     private void createDatastore(final String... datastoreNames) {
77         for (final String datastoreName : datastoreNames) {
78             final String nodeData = "{\"datastore\":[{\"name\":\"" + datastoreName + "\",\"cm-handles\":{}}]}";
79             try {
80                 cpsDataService.saveData(NCMP_DATASPACE_NAME, ANCHOR_NAME, "/" + REGISTRY_DATANODE_NAME, nodeData,
81                         OffsetDateTime.now(), JSON);
82             } catch (final AlreadyDefinedException exception) {
83                 log.info("Creating new child data node '{}' for data node '{}' failed as data node already exists",
84                         datastoreName, REGISTRY_DATANODE_NAME);
85             } catch (final Exception exception) {
86                 log.error("Creating data node failed: {}", exception.getMessage());
87                 throw new NcmpStartUpException("Creating data node failed", exception.getMessage());
88             }
89         }
90     }
91
92 }