Merge "RTD change to document migration to Spring Boot 3.0"

[cps.git] / cps-ncmp-service / src / main / java / org / onap / cps / ncmp / init / SubscriptionModelLoader.java

diff --git a/cps-ncmp-service/src/main/java/org/onap/cps/ncmp/init/SubscriptionModelLoader.java b/cps-ncmp-service/src/main/java/org/onap/cps/ncmp/init/SubscriptionModelLoader.java
index 6713491..4d1a91c 100644 (file)
--- a/cps-ncmp-service/src/main/java/org/onap/cps/ncmp/init/SubscriptionModelLoader.java
+++ b/cps-ncmp-service/src/main/java/org/onap/cps/ncmp/init/SubscriptionModelLoader.java
@@ -20,105 +20,48 @@
 
 package org.onap.cps.ncmp.init;
 
-import java.io.InputStream;
-import java.nio.charset.StandardCharsets;
-import java.util.Map;
-import lombok.NonNull;
-import lombok.RequiredArgsConstructor;
+import static org.onap.cps.ncmp.api.impl.ncmppersistence.NcmpPersistence.NCMP_DATASPACE_NAME;
+
 import lombok.extern.slf4j.Slf4j;
 import org.onap.cps.api.CpsAdminService;
+import org.onap.cps.api.CpsDataService;
 import org.onap.cps.api.CpsModuleService;
-import org.onap.cps.ncmp.api.impl.exception.NcmpStartUpException;
-import org.onap.cps.spi.exceptions.AlreadyDefinedException;
-import org.springframework.boot.SpringApplication;
-import org.springframework.boot.context.event.ApplicationReadyEvent;
-import org.springframework.stereotype.Component;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Service;
 
 @Slf4j
-@Component
-@RequiredArgsConstructor
-public class SubscriptionModelLoader implements ModelLoader {
-
-    private final CpsAdminService cpsAdminService;
-    private final CpsModuleService cpsModuleService;
-    private static final String SUBSCRIPTION_DATASPACE_NAME = "NCMP-Admin";
-    private static final String SUBSCRIPTION_ANCHOR_NAME = "AVC-Subscriptions";
-    private static final String SUBSCRIPTION_SCHEMASET_NAME = "subscriptions";
+@Service
+public class SubscriptionModelLoader extends AbstractModelLoader {
 
-    /**
-     * Method calls boarding subscription model when Application is ready.
-     *
-     * @param applicationReadyEvent the event to respond to
-     */
-    @Override
-    public void onApplicationEvent(@NonNull final ApplicationReadyEvent applicationReadyEvent) {
-        try {
-            onboardSubscriptionModel();
-        } catch (final NcmpStartUpException ncmpStartUpException) {
-            log.debug("Onboarding model for NCMP failed: {} ", ncmpStartUpException.getMessage());
-            SpringApplication.exit(applicationReadyEvent.getApplicationContext(), () -> 1);
-        }
-    }
+    private static final String MODEL_FILENAME = "subscription.yang";
+    private static final String ANCHOR_NAME = "AVC-Subscriptions";
+    private static final String SCHEMASET_NAME = "subscriptions";
+    private static final String REGISTRY_DATANODE_NAME = "subscription-registry";
 
-    /**
-     * Method to onboard subscription model for NCMP.
-     */
-    private void onboardSubscriptionModel() {
-        final Map<String, String> yangResourceContentMap = createYangResourceToContentMap();
-        if (!yangResourceContentMap.get("subscription.yang").isEmpty()) {
-            createSchemaSet(SUBSCRIPTION_DATASPACE_NAME, SUBSCRIPTION_SCHEMASET_NAME, yangResourceContentMap);
-            createAnchor(SUBSCRIPTION_DATASPACE_NAME, SUBSCRIPTION_SCHEMASET_NAME, SUBSCRIPTION_ANCHOR_NAME);
-        }
+    public SubscriptionModelLoader(final CpsAdminService cpsAdminService,
+                                   final CpsModuleService cpsModuleService,
+                                   final CpsDataService cpsDataService) {
+        super(cpsAdminService, cpsModuleService, cpsDataService);
     }
 
+    @Value("${ncmp.model-loader.subscription:true}")
+    private boolean subscriptionModelLoaderEnabled;
 
     @Override
-    public boolean createSchemaSet(final String dataspaceName,
-                                final String schemaSetName,
-                                final Map<String, String> yangResourceContentMap) {
-        try {
-            cpsModuleService.createSchemaSet(dataspaceName, schemaSetName, yangResourceContentMap);
-        } catch (final AlreadyDefinedException exception) {
-            log.info("Creating new schema set failed as schema set already exists");
-        } catch (final Exception exception) {
-            log.debug("Creating schema set for subscription model failed: {} ", exception.getMessage());
-            throw new NcmpStartUpException("Creating schema set failed", exception.getMessage());
-        }
-        return true;
-    }
-
-    /**
-     * Create Anchor.
-     *
-     * @param dataspaceName dataspace name
-     * @param schemaSetName schema set name
-     * @param anchorName anchor name
-     */
-    @Override
-    public boolean createAnchor(final String dataspaceName, final String schemaSetName,
-                             final String anchorName) {
-        try {
-            cpsAdminService.createAnchor(dataspaceName, schemaSetName, anchorName);
-        } catch (final AlreadyDefinedException exception) {
-            log.info("Creating new anchor failed as anchor already exists");
-        } catch (final Exception exception) {
-            log.debug("Creating anchor for subscription model failed: {} ", exception.getMessage());
-            throw new NcmpStartUpException("Creating anchor failed", exception.getMessage());
+    public void onboardOrUpgradeModel() {
+        if (subscriptionModelLoaderEnabled) {
+            waitUntilDataspaceIsAvailable(NCMP_DATASPACE_NAME);
+            onboardSubscriptionModel();
+            log.info("Subscription Model onboarded successfully");
+        } else {
+            log.info("Subscription Model Loader is disabled");
         }
-        return true;
     }
 
-    private String getFileContentAsString() {
-        try (InputStream inputStream = ClassLoader.getSystemClassLoader()
-                .getResourceAsStream("model/subscription.yang")) {
-            return new String(inputStream.readAllBytes(), StandardCharsets.UTF_8);
-        } catch (final Exception exception) {
-            log.debug("Onboarding failed as unable to read file: {}", exception.getCause().toString());
-            throw new NcmpStartUpException("Onboarding failed as unable to read file: {}", exception.getMessage());
-        }
+    private void onboardSubscriptionModel() {
+        createSchemaSet(NCMP_DATASPACE_NAME, SCHEMASET_NAME, MODEL_FILENAME);
+        createAnchor(NCMP_DATASPACE_NAME, SCHEMASET_NAME, ANCHOR_NAME);
+        createTopLevelDataNode(NCMP_DATASPACE_NAME, ANCHOR_NAME, REGISTRY_DATANODE_NAME);
     }
 
-    private Map<String, String> createYangResourceToContentMap() {
-        return Map.of("subscription.yang", getFileContentAsString());
-    }
 }