Process Service-level Instance Groups
[aai/babel.git] / src / main / java / org / onap / aai / babel / xml / generator / api / AaiArtifactGenerator.java
index 67c0b2f..531a044 100644 (file)
@@ -23,7 +23,7 @@ package org.onap.aai.babel.xml.generator.api;
 import java.io.IOException;
 import java.nio.file.Files;
 import java.nio.file.Path;
-import java.util.HashMap;
+import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
 import org.apache.commons.io.FileUtils;
@@ -37,13 +37,16 @@ import org.onap.aai.babel.xml.generator.data.ArtifactType;
 import org.onap.aai.babel.xml.generator.data.GenerationData;
 import org.onap.aai.babel.xml.generator.data.GeneratorUtil;
 import org.onap.aai.babel.xml.generator.data.GroupType;
+import org.onap.aai.babel.xml.generator.data.WidgetConfigurationUtil;
 import org.onap.aai.babel.xml.generator.model.Model;
 import org.onap.aai.babel.xml.generator.model.ProvidingService;
 import org.onap.aai.babel.xml.generator.model.Resource;
 import org.onap.aai.babel.xml.generator.model.Service;
+import org.onap.aai.babel.xml.generator.model.TunnelXconnectWidget;
 import org.onap.aai.cl.api.Logger;
 import org.onap.sdc.tosca.parser.api.ISdcCsarHelper;
 import org.onap.sdc.tosca.parser.impl.SdcToscaParserFactory;
+import org.onap.sdc.toscaparser.api.Group;
 import org.onap.sdc.toscaparser.api.NodeTemplate;
 import org.slf4j.MDC;
 
@@ -79,7 +82,7 @@ public class AaiArtifactGenerator implements ArtifactGenerator {
             ArtifactGeneratorToscaParser.initGroupFilterConfiguration();
             ISdcCsarHelper csarHelper =
                     SdcToscaParserFactory.getInstance().getSdcCsarHelper(csarPath.toAbsolutePath().toString());
-            return generateService(validateServiceVersion(additionalParams), csarHelper);
+            return generateAllArtifacts(validateServiceVersion(additionalParams), csarHelper);
         } catch (Exception e) {
             log.error(ApplicationMsgs.INVALID_CSAR_FILE, e);
             return createErrorData(e);
@@ -98,37 +101,30 @@ public class AaiArtifactGenerator implements ArtifactGenerator {
      * Generate model artifacts for the Service and its associated Resources.
      *
      * @param serviceVersion
-     * @param csarHelper TOSCA parser
-     * @return the generated Artifacts
+     * @param csarHelper
+     *        interface to the TOSCA parser
+     * @return the generated Artifacts (containing XML models)
      */
-    private GenerationData generateService(final String serviceVersion, ISdcCsarHelper csarHelper) {
+    private GenerationData generateAllArtifacts(final String serviceVersion, ISdcCsarHelper csarHelper) {
         List<NodeTemplate> serviceNodeTemplates = csarHelper.getServiceNodeTemplates();
         if (serviceNodeTemplates == null) {
             throw new IllegalArgumentException(GENERATOR_AAI_ERROR_MISSING_SERVICE_TOSCA);
         }
 
-        // Populate basic service model metadata
-        Service serviceModel = new Service();
-        serviceModel.setModelVersion(serviceVersion);
-        serviceModel.populateModelIdentificationInformation(csarHelper.getServiceMetadataAllProperties());
-
-        Map<String, String> idTypeStore = new HashMap<>();
+        Service serviceModel = createServiceModel(serviceVersion, csarHelper.getServiceMetadataAllProperties());
 
-        ArtifactGeneratorToscaParser parser = new ArtifactGeneratorToscaParser(csarHelper);
-        if (!serviceNodeTemplates.isEmpty()) {
-            parser.processServiceTosca(serviceModel, idTypeStore, serviceNodeTemplates);
-        }
+        MDC.put(MDC_PARAM_MODEL_INFO, serviceModel.getModelName() + "," + getArtifactLabel(serviceModel));
 
-        // Process the resource TOSCA files
-        List<Resource> resources = parser.processResourceToscas(serviceNodeTemplates, idTypeStore);
+        List<Resource> resources = generateResourceModels(csarHelper, serviceNodeTemplates, serviceModel);
 
-        MDC.put(MDC_PARAM_MODEL_INFO, serviceModel.getModelName() + "," + getArtifactLabel(serviceModel));
-        String aaiServiceModel = modelGenerator.generateModelFor(serviceModel);
+        // Generate the A&AI XML model for the Service.
+        final String serviceArtifact = modelGenerator.generateModelFor(serviceModel);
 
+        // Build a Babel Artifact to be returned to the caller.
         GenerationData generationData = new GenerationData();
-        generationData.add(getServiceArtifact(serviceModel, aaiServiceModel));
+        generationData.add(getServiceArtifact(serviceModel, serviceArtifact));
 
-        // Generate AAI XML resource model
+        // For each Resource, generate the A&AI XML model and then create an additional Artifact for that model.
         for (Resource resource : resources) {
             generateResourceArtifact(generationData, resource);
             for (Resource childResource : resource.getResources()) {
@@ -141,6 +137,97 @@ public class AaiArtifactGenerator implements ArtifactGenerator {
         return generationData;
     }
 
+    /**
+     * Create a Service from the provided metadata
+     *
+     * @param serviceVersion
+     * @param properties
+     * @return
+     */
+    private Service createServiceModel(final String serviceVersion, Map<String, String> properties) {
+        log.debug("Processing (TOSCA) Service object");
+        Service serviceModel = new Service();
+        serviceModel.setModelVersion(serviceVersion);
+        serviceModel.populateModelIdentificationInformation(properties);
+        return serviceModel;
+    }
+
+    /**
+     * @param csarHelper
+     * @param serviceNodeTemplates
+     * @param serviceModel
+     * @return the generated Models
+     */
+    private List<Resource> generateResourceModels(ISdcCsarHelper csarHelper, List<NodeTemplate> serviceNodeTemplates,
+            Service serviceModel) {
+        final List<Group> serviceGroups = csarHelper.getGroupsOfTopologyTemplate();
+        final ArtifactGeneratorToscaParser parser = new ArtifactGeneratorToscaParser(csarHelper);
+
+        List<Resource> resources = new ArrayList<>();
+
+        for (NodeTemplate nodeTemplate : serviceNodeTemplates) {
+            if (nodeTemplate.getMetaData() != null) {
+                generateModelFromNodeTemplate(csarHelper, serviceModel, resources, serviceGroups, parser, nodeTemplate);
+            } else {
+                log.warn(ApplicationMsgs.MISSING_SERVICE_METADATA, nodeTemplate.getName());
+            }
+        }
+
+        return resources;
+    }
+
+    private void generateModelFromNodeTemplate(ISdcCsarHelper csarHelper, Service serviceModel,
+            List<Resource> resources, final List<Group> serviceGroups, ArtifactGeneratorToscaParser parser,
+            NodeTemplate nodeTemplate) {
+        String nodeTypeName = parser.normaliseNodeTypeName(nodeTemplate);
+        Model model = Model.getModelFor(nodeTypeName, nodeTemplate.getMetaData().getValue("type"));
+        if (model != null) {
+            if (nodeTemplate.getMetaData() != null) {
+                model.populateModelIdentificationInformation(nodeTemplate.getMetaData().getAllProperties());
+            }
+
+            parser.addRelatedModel(serviceModel, model);
+            if (model instanceof Resource) {
+                generateResourceModel(csarHelper, resources, parser, nodeTemplate, nodeTypeName);
+            }
+        } else {
+            for (Group group : serviceGroups) {
+                if (group.getMembers().contains(nodeTemplate.getName())
+                        && WidgetConfigurationUtil.isSupportedInstanceGroup(group.getType())) {
+                    log.debug(String.format("Adding group %s (type %s) with members %s", group.getName(),
+                            group.getType(), group.getMembers()));
+
+                    Resource groupModel = parser.createInstanceGroupModel(
+                            parser.mergeProperties(group.getMetadata().getAllProperties(), group.getProperties()));
+                    serviceModel.addResource(groupModel);
+                    resources.add(groupModel);
+                }
+            }
+        }
+    }
+
+    private void generateResourceModel(ISdcCsarHelper csarHelper, List<Resource> resources,
+            ArtifactGeneratorToscaParser parser, NodeTemplate nodeTemplate, String nodeTypeName) {
+        log.debug("Processing resource " + nodeTypeName + ": " + nodeTemplate.getMetaData().getValue("UUID"));
+        Model resourceModel = Model.getModelFor(nodeTypeName, nodeTemplate.getMetaData().getValue("type"));
+
+        Map<String, String> serviceMetadata = nodeTemplate.getMetaData().getAllProperties();
+        resourceModel.populateModelIdentificationInformation(serviceMetadata);
+
+        parser.processResourceModels(resourceModel, csarHelper.getNodeTemplateChildren(nodeTemplate));
+
+        if (csarHelper.getServiceVfList() != null) {
+            parser.processVfModules(resources, resourceModel, nodeTemplate);
+        }
+
+        if (parser.hasSubCategoryTunnelXConnect(serviceMetadata) && parser.hasAllottedResource(serviceMetadata)) {
+            resourceModel.addWidget(new TunnelXconnectWidget());
+        }
+
+        resources.addAll(parser.processInstanceGroups(resourceModel, nodeTemplate));
+        resources.add((Resource) resourceModel);
+    }
+
     /**
      * @param generationData
      * @param resource
@@ -148,8 +235,7 @@ public class AaiArtifactGenerator implements ArtifactGenerator {
     private void generateResourceArtifact(GenerationData generationData, Resource resource) {
         if (!isContained(generationData, getArtifactName(resource))) {
             log.info(ApplicationMsgs.DISTRIBUTION_EVENT, "Generating resource model");
-            Artifact resourceArtifact = getResourceArtifact(resource, modelGenerator.generateModelFor(resource));
-            generationData.add(resourceArtifact);
+            generationData.add(getResourceArtifact(resource, modelGenerator.generateModelFor(resource)));
         }
     }
 
@@ -178,7 +264,8 @@ public class AaiArtifactGenerator implements ArtifactGenerator {
     /**
      * Method to generate the artifact name for an AAI model.
      *
-     * @param model AAI artifact model
+     * @param model
+     *        AAI artifact model
      * @return Model artifact name
      */
     private String getArtifactName(Model model) {
@@ -201,8 +288,10 @@ public class AaiArtifactGenerator implements ArtifactGenerator {
     /**
      * Create Resource artifact model from the AAI xml model string.
      *
-     * @param resourceModel Model of the resource artifact
-     * @param aaiResourceModel AAI model as string
+     * @param resourceModel
+     *        Model of the resource artifact
+     * @param aaiResourceModel
+     *        AAI model as string
      * @return Generated {@link Artifact} model for the resource
      */
     private Artifact getResourceArtifact(Model resourceModel, String aaiResourceModel) {
@@ -229,10 +318,12 @@ public class AaiArtifactGenerator implements ArtifactGenerator {
     }
 
     /**
-     * Create Service artifact model from the AAI xml model string.
+     * Create Service artifact model from the AAI XML model.
      *
-     * @param serviceModel Model of the service artifact
-     * @param aaiServiceModel AAI model as string
+     * @param serviceModel
+     *        Model of the service artifact
+     * @param aaiServiceModel
+     *        AAI model as string
      * @return Generated {@link Artifact} model for the service
      */
     private Artifact getServiceArtifact(Service serviceModel, String aaiServiceModel) {