Merge "add in volume group model info lookup"

1 files changed, 26 insertions, 10 deletions

diff --git a/bpmn/MSOCommonBPMN/src/main/java/org/onap/so/bpmn/servicedecomposition/tasks/BBInputSetup.java b/bpmn/MSOCommonBPMN/src/main/java/org/onap/so/bpmn/servicedecomposition/tasks/BBInputSetup.java
index b1e5d0d208..4be2f030b1 100644
--- a/bpmn/MSOCommonBPMN/src/main/java/org/onap/so/bpmn/servicedecomposition/tasks/BBInputSetup.java
+++ b/bpmn/MSOCommonBPMN/src/main/java/org/onap/so/bpmn/servicedecomposition/tasks/BBInputSetup.java
@@ -390,16 +390,10 @@ public class BBInputSetup implements JavaDelegate {
 				ModelInfo vnfModelInfo = new ModelInfo();
 				vnfModelInfo.setModelCustomizationUuid(vnfModelCustomizationUUID);
 				this.mapCatalogVnf(tempVnf, vnfModelInfo, service);
-				if (lookupKeyMap.get(ResourceKey.VOLUME_GROUP_ID) == null) {
-					for(VolumeGroup volumeGroup : tempVnf.getVolumeGroups()) {
-						String volumeGroupCustId = 
-								this.bbInputSetupUtils.getAAIVolumeGroup(cloudConfiguration.getCloudOwner(), 
-										cloudConfiguration.getLcpCloudRegionId(), volumeGroup.getVolumeGroupId()).getModelCustomizationId();
-						if(modelInfo.getModelCustomizationId().equalsIgnoreCase(volumeGroupCustId)) {
-							lookupKeyMap.put(ResourceKey.VOLUME_GROUP_ID, volumeGroup.getVolumeGroupId());
-							break;
-						}
-					}
+				Optional<String> volumeGroupIdOp = getVolumeGroupIdRelatedToVfModule(tempVnf, modelInfo, cloudConfiguration.getCloudOwner(), 
+						cloudConfiguration.getLcpCloudRegionId(), lookupKeyMap);
+				if(volumeGroupIdOp.isPresent()) {
+					lookupKeyMap.put(ResourceKey.VOLUME_GROUP_ID, volumeGroupIdOp.get());
 				}
 				break;
 			}
@@ -428,6 +422,21 @@ public class BBInputSetup implements JavaDelegate {
 			throw new Exception("Could not find relevant information for related VNF");
 		}
 	}
+	
+	protected Optional<String> getVolumeGroupIdRelatedToVfModule(GenericVnf vnf, ModelInfo modelInfo, 
+			String cloudOwner, String cloudRegionId, Map<ResourceKey, String> lookupKeyMap) {
+		if (lookupKeyMap.get(ResourceKey.VOLUME_GROUP_ID) == null) {
+			for(VolumeGroup volumeGroup : vnf.getVolumeGroups()) {
+				String volumeGroupCustId = 
+						bbInputSetupUtils.getAAIVolumeGroup(cloudOwner, 
+								cloudRegionId, volumeGroup.getVolumeGroupId()).getModelCustomizationId();
+				if(modelInfo.getModelCustomizationId().equalsIgnoreCase(volumeGroupCustId)) {
+					return Optional.of(volumeGroup.getVolumeGroupId());
+				}
+			}
+		}
+		return Optional.empty();
+	}
 
 	protected void mapCatalogVfModule(VfModule vfModule, ModelInfo modelInfo, Service service,
 			String vnfModelCustomizationUUID) {
@@ -1029,6 +1038,13 @@ public class BBInputSetup implements JavaDelegate {
 						ModelInfo vfModuleModelInfo = new ModelInfo();
 						vfModuleModelInfo.setModelCustomizationId(vfModuleCustomizationUUID);
 						this.mapCatalogVfModule(vfModule, vfModuleModelInfo, service, vnfModelCustomizationUUID);
+						if(cloudRegion != null) {
+							Optional<String> volumeGroupIdOp = getVolumeGroupIdRelatedToVfModule(vnf, vfModuleModelInfo, cloudRegion.getCloudOwner(), 
+									cloudRegion.getLcpCloudRegionId(), lookupKeyMap);
+							if(volumeGroupIdOp.isPresent()) {
+								lookupKeyMap.put(ResourceKey.VOLUME_GROUP_ID, volumeGroupIdOp.get());
+							}
+						}
 						break;
 					}
 				}