1 files changed, 338 insertions, 0 deletions
diff --git a/bpmn/so-bpmn-infrastructure-common/src/main/groovy/org/onap/so/bpmn/infrastructure/scripts/DoAllocateNSIandNSSI.groovy b/bpmn/so-bpmn-infrastructure-common/src/main/groovy/org/onap/so/bpmn/infrastructure/scripts/DoAllocateNSIandNSSI.groovy
new file mode 100644
index 0000000000..d5b554d841
--- /dev/null
+++ b/bpmn/so-bpmn-infrastructure-common/src/main/groovy/org/onap/so/bpmn/infrastructure/scripts/DoAllocateNSIandNSSI.groovy
@@ -0,0 +1,338 @@
+package org.onap.so.bpmn.infrastructure.scripts
+
+import com.google.common.reflect.TypeToken
+import com.google.gson.Gson
+import org.camunda.bpm.engine.delegate.BpmnError
+import org.camunda.bpm.engine.delegate.DelegateExecution
+import org.onap.aai.domain.yang.AllottedResource
+import org.onap.aai.domain.yang.Relationship
+import org.onap.aai.domain.yang.RelationshipList
+import org.onap.aai.domain.yang.ServiceInstance
+import org.onap.so.beans.nsmf.SliceTaskParams
+import org.onap.so.bpmn.common.scripts.ExceptionUtil
+import org.onap.so.bpmn.core.domain.ServiceDecomposition
+import org.onap.so.bpmn.core.domain.ServiceProxy
+import org.onap.so.bpmn.core.json.JsonUtils
+import org.onap.so.client.aai.AAIObjectType
+import org.onap.so.client.aai.AAIResourcesClient
+import org.onap.so.client.aai.entities.AAIEdgeLabel
+import org.onap.so.client.aai.entities.AAIResultWrapper
+import org.onap.so.client.aai.entities.uri.AAIResourceUri
+import org.onap.so.client.aai.entities.uri.AAIUriFactory
+import org.onap.so.db.request.client.RequestsDbClient
+import org.onap.so.db.request.beans.OrchestrationTask
+import org.slf4j.Logger
+import org.slf4j.LoggerFactory
+
+import javax.ws.rs.NotFoundException
+import javax.ws.rs.core.UriBuilder
+
+import static org.apache.commons.lang3.StringUtils.isBlank
+
+class DoAllocateNSIandNSSI extends org.onap.so.bpmn.common.scripts.AbstractServiceTaskProcessor{
+
+    private static final Logger logger = LoggerFactory.getLogger( DoAllocateNSIandNSSI.class);
+
+    ExceptionUtil exceptionUtil = new ExceptionUtil()
+
+    JsonUtils jsonUtil = new JsonUtils()
+    RequestsDbClient requestsDbClient = new RequestsDbClient()
+
+    /**
+     * Pre Process the BPMN Flow Request
+     * Inclouds:
+     * generate the nsOperationKey
+     * generate the nsParameters
+     */
+
+    void preProcessRequest (DelegateExecution execution) {
+        String msg = ""
+        logger.trace("Enter preProcessRequest()")
+        Map<String, Object> nssiMap = new HashMap<>()
+        execution.setVariable("nssiMap", nssiMap)
+        boolean isMoreNSSTtoProcess = true
+        execution.setVariable("isMoreNSSTtoProcess", isMoreNSSTtoProcess)
+        List<String> nsstSequence = new ArrayList<>(Arrays.asList("cn"))
+        execution.setVariable("nsstSequence", nsstSequence)
+        logger.trace("Exit preProcessRequest")
+    }
+
+    void retriveSliceOption(DelegateExecution execution) {
+        logger.trace("Enter retriveSliceOption() of DoAllocateNSIandNSSI")
+        String uuiRequest = execution.getVariable("uuiRequest")
+        boolean isNSIOptionAvailable = false
+        List<String> nssiAssociated = new ArrayList<>()
+        SliceTaskParams sliceParams = execution.getVariable("sliceTaskParams")
+        try
+        {
+            String modelUuid = jsonUtil.getJsonValue(uuiRequest, "service.parameters.requestInputs.nstar0_allottedresource0_providing_service_uuid")
+            String modelInvariantUuid = jsonUtil.getJsonValue(uuiRequest, "service.parameters.requestInputs.nstar0_allottedresource0_providing_service_invariant_uuid")
+            String serviceModelInfo = """{
+            "modelInvariantUuid":"${modelInvariantUuid}",
+            "modelUuid":"${modelUuid}",
+            "modelVersion":""
+             }"""
+            execution.setVariable("serviceModelInfo", serviceModelInfo)
+            //Params sliceParams = new Gson().fromJson(params, new TypeToken<Params>() {}.getType());
+            execution.setVariable("sliceParams", sliceParams)
+        }catch (Exception ex) {
+            logger.debug( "Unable to get the task information from request DB: " + ex)
+            exceptionUtil.buildAndThrowWorkflowException(execution, 401, "Unable to get task information from request DB.")
+        }
+
+        if(isBlank(sliceParams.getSuggestNsiId()))
+        {
+            isNSIOptionAvailable=false
+        }
+        else
+        {
+            isNSIOptionAvailable=true
+            execution.setVariable('nsiServiceInstanceId',sliceParams.getSuggestNsiId())
+            execution.setVariable('nsiServiceInstanceName',sliceParams.getSuggestNsiName())
+        }
+        execution.setVariable("isNSIOptionAvailable",isNSIOptionAvailable)
+        logger.trace("Exit retriveSliceOption() of DoAllocateNSIandNSSI")
+    }
+
+    void updateRelationship(DelegateExecution execution) {
+        logger.trace("Enter update relationship in DoAllocateNSIandNSSI()")
+        String nsiServiceInstanceId = execution.getVariable("nsiServiceInstanceId")
+        String allottedResourceId = execution.getVariable("allottedResourceId")
+        //Need to check whether nsi exist : Begin
+        org.onap.aai.domain.yang.ServiceInstance nsiServiceInstance = new org.onap.aai.domain.yang.ServiceInstance()
+        SliceTaskParams sliceParams = execution.getVariable("sliceParams")
+
+        String nsiServiceInstanceID = sliceParams.getSuggestNsiId()
+
+        AAIResourcesClient resourceClient = new AAIResourcesClient()
+        AAIResourceUri nsiServiceuri = AAIUriFactory.createResourceUri(AAIObjectType.SERVICE_INSTANCE, execution.getVariable("globalSubscriberId"), execution.getVariable("serviceType"), nsiServiceInstanceID)
+        //AAIResourceUri nsiServiceuri = AAIUriFactory.createResourceUri(AAIObjectType.QUERY_ALLOTTED_RESOURCE, execution.getVariable("globalSubscriberId"), execution.getVariable("serviceType"), nsiServiceInstanceID)
+
+        try {
+            AAIResultWrapper wrapper = resourceClient.get(nsiServiceuri, NotFoundException.class)
+            Optional<org.onap.aai.domain.yang.ServiceInstance> si = wrapper.asBean(org.onap.aai.domain.yang.ServiceInstance.class)
+            nsiServiceInstance = si.get()
+            //allottedResourceId=nsiServiceInstance.getAllottedResources().getAllottedResource().get(0).getId()
+
+//            if(resourceClient.exists(nsiServiceuri)){
+//                execution.setVariable("nsi_resourceLink", nsiServiceuri.build().toString())
+//            }else{
+//                exceptionUtil.buildAndThrowWorkflowException(execution, 2500, "Service instance was not found in aai to " +
+//                        "associate for service :"+serviceInstanceId)
+//            }
+        }catch(BpmnError e) {
+            throw e;
+        }catch (Exception ex){
+            String msg = "NSI suggested in the option doesn't exist. " + nsiServiceInstanceID
+            logger.debug(msg)
+            exceptionUtil.buildAndThrowWorkflowException(execution, 7000, msg)
+        }
+        AAIResourceUri allottedResourceUri = AAIUriFactory.createResourceUri(AAIObjectType.ALLOTTED_RESOURCE, execution.getVariable("globalSubscriberId"), execution.getVariable("serviceType"), nsiServiceInstanceId, allottedResourceId)
+        getAAIClient().connect(allottedResourceUri,nsiServiceuri)
+
+        List<String> nssiAssociated = new ArrayList<>()
+        RelationshipList relationshipList = nsiServiceInstance.getRelationshipList()
+        List<Relationship> relationships = relationshipList.getRelationship()
+        for(Relationship relationship in relationships)
+        {
+            if(relationship.getRelatedTo().equalsIgnoreCase("service-instance"))
+            {
+                String NSSIassociated = relationship.getRelatedLink().substring(relationship.getRelatedLink().lastIndexOf("/") + 1);
+                if(!NSSIassociated.equals(nsiServiceInstanceID))
+                    nssiAssociated.add(NSSIassociated)
+            }
+        }
+        execution.setVariable("nssiAssociated",nssiAssociated)
+        execution.setVariable("nsiServiceInstanceName",nsiServiceInstance.getServiceInstanceName())
+        logger.trace("Exit update relationship in DoAllocateNSIandNSSI()")
+    }
+
+    void prepareNssiModelInfo(DelegateExecution execution){
+        logger.trace("Enter prepareNssiModelInfo in DoAllocateNSIandNSSI()")
+        List<String> nssiAssociated = new ArrayList<>()
+        Map<String, Object> nssiMap = new HashMap<>()
+        nssiAssociated=execution.getVariable("nssiAssociated")
+        for(String nssiID in nssiAssociated)
+        {
+            try {
+                AAIResourcesClient resourceClient = new AAIResourcesClient()
+                AAIResourceUri serviceInstanceUri = AAIUriFactory.createResourceUri(AAIObjectType.SERVICE_INSTANCE, execution.getVariable("globalSubscriberId"), execution.getVariable("serviceType"), nssiID)
+                AAIResultWrapper wrapper = resourceClient.get(serviceInstanceUri, NotFoundException.class)
+                Optional<org.onap.aai.domain.yang.ServiceInstance> si = wrapper.asBean(org.onap.aai.domain.yang.ServiceInstance.class)
+                org.onap.aai.domain.yang.ServiceInstance nssi = si.get()
+                nssiMap.put(nssi.getEnvironmentContext(),"""{
+                    "serviceInstanceId":"${nssi.getServiceInstanceId()}",
+                    "modelUuid":"${nssi.getModelVersionId()}"
+                     }""")
+
+            }catch(NotFoundException e)
+            {
+                logger.debug("NSSI Service Instance not found in AAI: " + nssiID)
+            }catch(Exception e)
+            {
+                logger.debug("NSSI Service Instance not found in AAI: " + nssiID)
+            }
+            execution.setVariable("nssiMap",nssiMap)
+
+        }
+        logger.trace("Exit prepareNssiModelInfo in DoAllocateNSIandNSSI()")
+    }
+
+    void createNSIinAAI(DelegateExecution execution) {
+        logger.trace("Enter CreateNSIinAAI in DoAllocateNSIandNSSI()")
+        ServiceDecomposition serviceDecomposition = execution.getVariable("serviceDecomposition")
+        org.onap.aai.domain.yang.ServiceInstance nsi = new ServiceInstance();
+        String sliceInstanceId = UUID.randomUUID().toString()
+        execution.setVariable("sliceInstanceId",sliceInstanceId)
+        nsi.setServiceInstanceId(sliceInstanceId)
+        String sliceInstanceName = "nsi_"+execution.getVariable("serviceInstanceName")
+        nsi.setServiceInstanceName(sliceInstanceName)
+        String serviceType = execution.getVariable("serviceType")
+        nsi.setServiceType(serviceType)
+        String serviceStatus = "deactivated"
+        nsi.setOrchestrationStatus(serviceStatus)
+        String modelInvariantUuid = serviceDecomposition.getModelInfo().getModelInvariantUuid()
+        String modelUuid = serviceDecomposition.getModelInfo().getModelUuid()
+        nsi.setModelInvariantId(modelInvariantUuid)
+        nsi.setModelVersionId(modelUuid)
+        String uuiRequest = execution.getVariable("uuiRequest")
+        String serviceInstanceLocationid = jsonUtil.getJsonValue(uuiRequest, "service.parameters.requestInputs.plmnIdList")
+        nsi.setServiceInstanceLocationId(serviceInstanceLocationid)
+        //String snssai = jsonUtil.getJsonValue(uuiRequest, "service.requestInputs.snssai")
+        //nsi.setEnvironmentContext(snssai)
+        String serviceRole = "nsi"
+        nsi.setServiceRole(serviceRole)
+        try {
+
+            AAIResourcesClient client = new AAIResourcesClient()
+            AAIResourceUri uri = AAIUriFactory.createResourceUri(AAIObjectType.SERVICE_INSTANCE, execution.getVariable("globalSubscriberId"), execution.getVariable("subscriptionServiceType"), sliceInstanceId)
+            client.create(uri, nsi)
+
+        } catch (BpmnError e) {
+            throw e
+        } catch (Exception ex) {
+            msg = "Exception in DoCreateSliceServiceInstance.instantiateSliceService. " + ex.getMessage()
+            logger.info(msg)
+            exceptionUtil.buildAndThrowWorkflowException(execution, 7000, msg)
+        }
+        Map<String, Object> nssiMap = new HashMap<>()
+        List<ServiceProxy> serviceProxyList = serviceDecomposition.getServiceProxy()
+        List<String> nsstModelInfoList = new ArrayList<>()
+        for(ServiceProxy serviceProxy : serviceProxyList)
+        {
+            //String nsstModelUuid = serviceProxy.getModelInfo().getModelUuid()
+            String nsstModelUuid = serviceProxy.getSourceModelUuid()
+            //String nsstModelInvariantUuid = serviceProxy.getModelInfo().getModelInvariantUuid()
+            String nsstServiceModelInfo = """{
+            "modelInvariantUuid":"",
+            "modelUuid":"${nsstModelUuid}",
+            "modelVersion":""
+             }"""
+            nsstModelInfoList.add(nsstServiceModelInfo)
+        }
+        int currentIndex=0
+        int maxIndex=nsstModelInfoList.size()
+        if(maxIndex < 1)
+        {
+            msg = "Exception in DoAllocateNSIandNSSI. There is no NSST associated with NST "
+            logger.info(msg)
+            exceptionUtil.buildAndThrowWorkflowException(execution, 7000, msg)
+        }
+        execution.setVariable("nsstModelInfoList",nsstModelInfoList)
+        execution.setVariable("currentIndex",currentIndex)
+        execution.setVariable("maxIndex",maxIndex)
+        execution.setVariable('nsiServiceInstanceId',sliceInstanceId)
+        execution.setVariable("nsiServiceInstanceName",sliceInstanceName)
+        logger.trace("Exit CreateNSIinAAI in DoAllocateNSIandNSSI()")
+    }
+
+    void getOneNsstInfo(DelegateExecution execution){
+        List<String> nsstModelInfoList = new ArrayList<>()
+        nsstModelInfoList = execution.getVariable("nsstModelInfoList")
+        int currentIndex = execution.getVariable("currentIndex")
+        int maxIndex = execution.getVariable("maxIndex")
+        boolean isMoreNSSTtoProcess = true
+        String nsstServiceModelInfo = nsstModelInfoList.get(currentIndex)
+        execution.setVariable("serviceModelInfo", nsstServiceModelInfo)
+        execution.setVariable("currentIndex", currentIndex)
+        currentIndex = currentIndex+1
+        if(currentIndex <= maxIndex )
+            isMoreNSSTtoProcess = false
+        execution.setVariable("isMoreNSSTtoProcess", isMoreNSSTtoProcess)
+    }
+
+    void createNSSTMap(DelegateExecution execution){
+        ServiceDecomposition serviceDecomposition = execution.getVariable("serviceDecomposition")
+        String modelUuid= serviceDecomposition.getModelInfo().getModelUuid()
+        String content = serviceDecomposition.getServiceInfo().getServiceArtifact().get(0).getContent()
+        //String nsstID = jsonUtil.getJsonValue(content, "metadata.id")
+        //String vendor = jsonUtil.getJsonValue(content, "metadata.vendor")
+        //String type = jsonUtil.getJsonValue(content, "metadata.type")
+        String domain = jsonUtil.getJsonValue(content, "metadata.domainType")
+
+        Map<String, Object> nssiMap = execution.getVariable("nssiMap")
+        String servicename = execution.getVariable("serviceInstanceName")
+        String nsiname = "nsi_"+servicename
+        nssiMap.put(domain,"""{
+                    "serviceInstanceId":"",
+                    "modelUuid":"${modelUuid}"
+                     }""")
+        execution.setVariable("nssiMap",nssiMap)
+    }
+
+    void prepareNSSIList(DelegateExecution execution){
+        logger.trace("Enter prepareNSSIList in DoAllocateNSIandNSSI()")
+        Map<String, Object> nssiMap = new HashMap<>()
+        Boolean isMoreNSSI = false
+        nssiMap = execution.getVariable("nssiMap")
+        List<String> keys=new ArrayList<String>(nssiMap.values())
+        List<String> nsstSequence = execution.getVariable("nsstSequence")
+        Integer currentIndex=0;
+        execution.setVariable("currentNssiIndex",currentIndex)
+        Integer maxIndex=keys.size()
+        execution.setVariable("maxIndex",maxIndex)
+        if(maxIndex>0)
+            isMoreNSSI=true
+        execution.setVariable("isMoreNSSI",isMoreNSSI)
+        logger.trace("Exit prepareNSSIList in DoAllocateNSIandNSSI()")
+    }
+
+
+    void getOneNSSIInfo(DelegateExecution execution){
+        logger.trace("Enter getOneNSSIInfo in DoAllocateNSIandNSSI()")
+
+        //ServiceDecomposition serviceDecompositionObj = execution.getVariable("serviceDecompositionObj")
+        Map<String, Object> nssiMap=execution.getVariable("nssiMap")
+        List<String> nsstSequence = execution.getVariable("nsstSequence")
+        String currentNSST= nsstSequence.get(execution.getVariable("currentNssiIndex"))
+        boolean isNSSIOptionAvailable = false
+        String nsstInput=nssiMap.get(currentNSST)
+        execution.setVariable("nsstInput",nsstInput)
+        String modelUuid = jsonUtil.getJsonValue(nsstInput, "modelUuid")
+        String nssiInstanceId = jsonUtil.getJsonValue(nsstInput, "serviceInstanceId")
+        String nssiserviceModelInfo = """{
+            "modelInvariantUuid":"",
+            "modelUuid":"${modelUuid}",
+            "modelVersion":""
+             }"""
+        Integer currentIndex = execution.getVariable("currentNssiIndex")
+        currentIndex=currentIndex+1;
+        execution.setVariable("currentNssiIndex",currentIndex)
+        execution.setVariable("nssiserviceModelInfo",nssiserviceModelInfo)
+        execution.setVariable("nssiInstanceId",nssiInstanceId)
+        logger.trace("Exit getOneNSSIInfo in DoAllocateNSIandNSSI()")
+    }
+
+    void updateCurrentIndex(DelegateExecution execution){
+
+        logger.trace("Enter updateCurrentIndex in DoAllocateNSIandNSSI()")
+        Integer currentIndex = execution.getVariable("currentNssiIndex")
+        Integer maxIndex = execution.getVariable("maxIndex")
+        if(currentIndex>=maxIndex)
+        {
+            Boolean isMoreNSSI=false
+            execution.setVariable("isMoreNSSI",isMoreNSSI)
+        }
+        logger.trace("Exit updateCurrentIndex in DoAllocateNSIandNSSI()")
+    }
+}