bpmn/so-bpmn-tasks/src/main/java/org/onap/so/bpmn/infrastructure/aai/tasks/AAIQueryTasks.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP - SO
   4  * ================================================================================
   5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.onap.so.bpmn.infrastructure.aai.tasks;
  22
  23 import java.util.ArrayList;
  24 import java.util.List;
  25 import java.util.Optional;
  26
  27 import org.modelmapper.ModelMapper;
  28 import org.modelmapper.PropertyMap;
  29 import org.onap.aai.domain.yang.NetworkPolicy;
  30 import org.onap.aai.domain.yang.RouteTableReference;
  31 import org.onap.aai.domain.yang.RouteTargets;
  32 import org.onap.aai.domain.yang.VpnBinding;
  33 import org.onap.so.adapters.nwrest.CreateNetworkRequest;
  34 import org.onap.so.bpmn.common.BuildingBlockExecution;
  35 import org.onap.so.bpmn.servicedecomposition.bbobjects.L3Network;
  36 import org.onap.so.bpmn.servicedecomposition.entities.ResourceKey;
  37 import org.onap.so.bpmn.servicedecomposition.tasks.ExtractPojosForBB;
  38 import org.onap.so.client.aai.AAIObjectType;
  39 import org.onap.so.client.aai.entities.AAIResultWrapper;
  40 import org.onap.so.client.aai.entities.Relationships;
  41 import org.onap.so.client.aai.entities.uri.AAIResourceUri;
  42 import org.onap.so.client.exception.ExceptionBuilder;
  43 import org.onap.so.client.orchestration.AAINetworkResources;
  44 import org.onap.so.logger.MsoLogger;
  45 import org.springframework.beans.factory.annotation.Autowired;
  46 import org.springframework.stereotype.Component;
  47
  48 @Component
  49 public class AAIQueryTasks {
  50
  51         private static final MsoLogger msoLogger = MsoLogger.getMsoLogger(MsoLogger.Catalog.BPEL, AAIQueryTasks.class);
  52         private static final String ERROR_MSG = "No relationships were returned from AAIResultWrapper.getRelationships()";
  53         @Autowired
  54         private ExtractPojosForBB extractPojosForBB;
  55         @Autowired
  56         private ExceptionBuilder exceptionUtil;
  57         @Autowired
  58         private AAINetworkResources aaiNetworkResources;
  59         private static final ModelMapper modelMapper = new ModelMapper();
  60
  61         /**
  62          * BPMN access method to query data for VPN bindings from the AAI result wrapper.
  63          * The resulting VPN bindings are mapped to the corresponding bbobject and placed in the customer bbobject
  64          *
  65          * @param execution
  66          */
  67
  68         public void queryNetworkVpnBinding(BuildingBlockExecution execution) {
  69                 try {
  70                         L3Network l3network = extractPojosForBB.extractByKey(execution, ResourceKey.NETWORK_ID,
  71                                         execution.getLookupMap().get(ResourceKey.NETWORK_ID));
  72                         AAIResultWrapper aaiResultWrapper = aaiNetworkResources.queryNetworkWrapperById(l3network);
  73                         Optional<Relationships> networkRelationships = aaiResultWrapper.getRelationships();
  74                         if (!networkRelationships.isPresent()) {
  75                                 throw (new Exception(ERROR_MSG));
  76                         }
  77                         List<AAIResourceUri> netBindingsUriList = networkRelationships.get().getRelatedAAIUris(AAIObjectType.VPN_BINDING);
  78
  79                         List<org.onap.so.bpmn.servicedecomposition.bbobjects.VpnBinding> mappedVpnBindings = new ArrayList<>();
  80                         if(netBindingsUriList != null && !netBindingsUriList.isEmpty()) {
  81                                 for(AAIResourceUri netBindingUri : netBindingsUriList) {
  82                                         Optional<VpnBinding> oVpnBinding = aaiNetworkResources.getVpnBinding(netBindingUri);
  83                                         if(oVpnBinding.isPresent()) {
  84                                                 org.onap.so.bpmn.servicedecomposition.bbobjects.VpnBinding mappedVpnBinding =
  85                                                                 modelMapper.map(oVpnBinding.get(), org.onap.so.bpmn.servicedecomposition.bbobjects.VpnBinding.class);
  86                                                 if (oVpnBinding.get().getRouteTargets() != null){
  87                                                         mappedVpnBinding.getRouteTargets().addAll(mapRouteTargets(oVpnBinding.get().getRouteTargets().getRouteTarget()));
  88                                                         mappedVpnBindings.add(mappedVpnBinding);
  89                                                 }
  90                                         }
  91                                 }
  92                         }
  93                         execution.getGeneralBuildingBlock().getCustomer().getVpnBindings().addAll(mappedVpnBindings);
  94                 } catch(Exception ex) {
  95                         exceptionUtil.buildAndThrowWorkflowException(execution, 7000, ex);
  96                 }
  97         }
  98
  99         /**
 100          * BPMN access method to extract VPN Binding data from AAI result and populate proper fields into CreateNetworkRequest
 101          */
 102         public void getNetworkVpnBinding(BuildingBlockExecution execution) {
 103
 104                 try {
 105                         L3Network l3network = extractPojosForBB.extractByKey(execution, ResourceKey.NETWORK_ID,
 106                                         execution.getLookupMap().get(ResourceKey.NETWORK_ID));
 107                         AAIResultWrapper aaiResultWrapper = aaiNetworkResources.queryNetworkWrapperById(l3network);
 108                         CreateNetworkRequest createNetworkRequest = execution.getVariable("createNetworkRequest");
 109
 110                         Optional<Relationships> networkRelationships = aaiResultWrapper.getRelationships();
 111                         if (!networkRelationships.isPresent()) {
 112                                 throw (new Exception(ERROR_MSG));
 113                         }
 114                         List<AAIResourceUri> netBindingsUriList = networkRelationships.get()
 115                                         .getRelatedAAIUris(AAIObjectType.VPN_BINDING);
 116                         List<org.onap.so.openstack.beans.RouteTarget> routeTargets = new ArrayList<>();
 117                         for (AAIResourceUri netBindingUri : netBindingsUriList) {
 118                                 msoLogger.info("Get Route Targests");
 119                                 Optional<VpnBinding> oVpnBinding = aaiNetworkResources.getVpnBinding(netBindingUri);
 120                                 if (oVpnBinding.isPresent()) {
 121                                         VpnBinding vpnBinding = oVpnBinding.get();
 122                                         RouteTargets rts = vpnBinding.getRouteTargets();
 123                                         if (rts != null) {
 124                                                 List<org.onap.aai.domain.yang.RouteTarget> rtList = rts.getRouteTarget();
 125                                                 if (!rtList.isEmpty()) {
 126                                                         PropertyMap<org.onap.aai.domain.yang.RouteTarget, org.onap.so.openstack.beans.RouteTarget> personMap = new PropertyMap<org.onap.aai.domain.yang.RouteTarget, org.onap.so.openstack.beans.RouteTarget>() {
 127                                                                 @Override
 128                                                                 protected void configure() {
 129                                                                         map().setRouteTarget(source.getGlobalRouteTarget());
 130                                                                         map(source.getRouteTargetRole(), destination.getRouteTargetRole());
 131                                                                 }
 132                                                         };
 133                                                         modelMapper.addMappings(personMap);
 134                                                         for (org.onap.aai.domain.yang.RouteTarget rt : rtList) {
 135                                                                 org.onap.so.openstack.beans.RouteTarget openstackRtBean = modelMapper.map(rt,
 136                                                                                 org.onap.so.openstack.beans.RouteTarget.class);
 137                                                                 routeTargets.add(openstackRtBean);
 138                                                         }
 139                                                 }
 140                                         }
 141                                 }
 142                         }
 143                         // store route targets data in execution - to be used as part of
 144                         // Network adapter input
 145                         createNetworkRequest.getContrailNetwork().setRouteTargets(routeTargets);
 146                         execution.setVariable("createNetworkRequest", createNetworkRequest);
 147                 } catch (Exception ex) {
 148                         exceptionUtil.buildAndThrowWorkflowException(execution, 7000, ex);
 149                 }
 150         }
 151
 152         /**
 153          * BPMN access method to query data for network policies from the AAI result wrapper
 154          * From the resulting network policy, the network policy fqdn parameter is added to
 155          * the network bbobject contrail network policy fqdns list
 156          *
 157          * @param execution
 158          */
 159         public void queryNetworkPolicy(BuildingBlockExecution execution) {
 160                 try {
 161                         L3Network l3network = extractPojosForBB.extractByKey(execution, ResourceKey.NETWORK_ID,
 162                                         execution.getLookupMap().get(ResourceKey.NETWORK_ID));
 163                         AAIResultWrapper aaiResultWrapper = aaiNetworkResources.queryNetworkWrapperById(l3network);
 164                         Optional<Relationships> networkRelationships = aaiResultWrapper.getRelationships();
 165                         if (!networkRelationships.isPresent()) {
 166                                 throw (new Exception(ERROR_MSG));
 167                         }
 168                         List<AAIResourceUri> netPoliciesUriList = networkRelationships.get().getRelatedAAIUris(AAIObjectType.NETWORK_POLICY);
 169
 170                         if(!netPoliciesUriList.isEmpty()) {
 171                                 for(AAIResourceUri netPolicyUri : netPoliciesUriList) {
 172                                         Optional<NetworkPolicy> oNetPolicy = aaiNetworkResources.getNetworkPolicy(netPolicyUri);
 173                                         if(oNetPolicy.isPresent()) {
 174                                                 l3network.getNetworkPolicies().add(modelMapper.map(oNetPolicy.get(), org.onap.so.bpmn.servicedecomposition.bbobjects.NetworkPolicy.class));
 175                                         }
 176                                 }
 177                         }
 178                 } catch(Exception ex) {
 179                         exceptionUtil.buildAndThrowWorkflowException(execution, 7000, ex);
 180                 }
 181         }
 182
 183         /**
 184          * BPMN access method to query data for network table ref from the AAI result wrapper
 185          * The resulting route table reference is mapped to the corresponding bbobject and added
 186          * to the network bbobject contrail network route table references list
 187          *
 188          * @param execution
 189          */
 190         public void queryNetworkTableRef(BuildingBlockExecution execution) {
 191                 try {
 192                         L3Network l3network = extractPojosForBB.extractByKey(execution, ResourceKey.NETWORK_ID,
 193                                         execution.getLookupMap().get(ResourceKey.NETWORK_ID));
 194                         AAIResultWrapper aaiResultWrapper = aaiNetworkResources.queryNetworkWrapperById(l3network);
 195                         Optional<Relationships> networkRelationships = aaiResultWrapper.getRelationships();
 196                         if (!networkRelationships.isPresent()) {
 197                                 throw (new Exception(ERROR_MSG));
 198                         }
 199                         List<AAIResourceUri> routeTableUriList = networkRelationships.get().getRelatedAAIUris(AAIObjectType.ROUTE_TABLE_REFERENCE);
 200
 201                         if(!routeTableUriList.isEmpty()) {
 202                                 for(AAIResourceUri routeTableUri : routeTableUriList) {
 203                                         Optional<RouteTableReference> oRouteTableReference = aaiNetworkResources.getRouteTable(routeTableUri);
 204                                         if(oRouteTableReference.isPresent()) {
 205                                                 org.onap.so.bpmn.servicedecomposition.bbobjects.RouteTableReference mappedRouteTableReference = modelMapper.map(oRouteTableReference.get(), org.onap.so.bpmn.servicedecomposition.bbobjects.RouteTableReference.class);
 206                                                 l3network.getContrailNetworkRouteTableReferences().add(mappedRouteTableReference);
 207                                         }
 208                                 }
 209                         }
 210                 } catch(Exception ex) {
 211                         exceptionUtil.buildAndThrowWorkflowException(execution, 7000, ex);
 212                 }
 213         }
 214
 215         private List<org.onap.so.bpmn.servicedecomposition.bbobjects.RouteTarget> mapRouteTargets(List<org.onap.aai.domain.yang.RouteTarget> routeTargets) {
 216                 List<org.onap.so.bpmn.servicedecomposition.bbobjects.RouteTarget> mappedRouteTargets = new ArrayList<>();
 217                 if(!routeTargets.isEmpty()) {
 218                         for(org.onap.aai.domain.yang.RouteTarget routeTarget : routeTargets) {
 219                                 org.onap.so.bpmn.servicedecomposition.bbobjects.RouteTarget mappedRouteTarget = modelMapper.map(routeTarget, org.onap.so.bpmn.servicedecomposition.bbobjects.RouteTarget.class);
 220                                 mappedRouteTargets.add(mappedRouteTarget);
 221                         }
 222                 }
 223                 return mappedRouteTargets;
 224         }
 225 }