gerrit.onap Code Review - so.git/blob

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP - SO
   4  * ================================================================================
   5  * Copyright (C) 2017 - 2019 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.workflow.tasks.listeners;
  22
  23 import java.util.Collections;
  24 import java.util.List;
  25 import org.onap.so.bpmn.common.BBConstants;
  26 import org.onap.so.bpmn.common.BuildingBlockExecution;
  27 import org.onap.so.bpmn.common.listener.flowmanipulator.FlowManipulator;
  28 import org.onap.so.bpmn.servicedecomposition.entities.ExecuteBuildingBlock;
  29 import org.onap.so.bpmn.servicedecomposition.tasks.BBInputSetupUtils;
  30 import org.onap.so.db.catalog.beans.VnfResourceCustomization;
  31 import org.onap.so.db.catalog.client.CatalogDbClient;
  32 import org.springframework.beans.factory.annotation.Autowired;
  33 import org.springframework.stereotype.Component;
  34
  35 @Component
  36 public class MultiStageSkipListener implements FlowManipulator {
  37
  38     @Autowired
  39     protected BBInputSetupUtils bbInputSetupUtils;
  40
  41     @Autowired
  42     private CatalogDbClient catalogDbClient;
  43
  44     @Override
  45     public boolean shouldRunFor(String currentBBName, boolean isFirst, BuildingBlockExecution execution) {
  46         return ((boolean) execution.getVariable(BBConstants.G_ALACARTE)) && "AssignVfModuleBB".equals(currentBBName)
  47                 && isFirst;
  48     }
  49
  50     @Override
  51     public void run(List<ExecuteBuildingBlock> flowsToExecute, ExecuteBuildingBlock currentBB,
  52             BuildingBlockExecution execution) {
  53         String vfModuleId = currentBB.getResourceId();
  54         String vnfId = currentBB.getWorkflowResourceIds().getVnfId();
  55         org.onap.aai.domain.yang.VfModule vfModule = bbInputSetupUtils.getAAIVfModule(vnfId, vfModuleId);
  56         if (vfModule == null) {
  57             org.onap.aai.domain.yang.GenericVnf vnf = bbInputSetupUtils.getAAIGenericVnf(vnfId);
  58             if (vnf != null) {
  59                 VnfResourceCustomization vnfCust = catalogDbClient
  60                         .getVnfResourceCustomizationByModelCustomizationUUID(vnf.getModelCustomizationId());
  61                 if (vnfCust != null && vnfCust.getMultiStageDesign() != null
  62                         && vnfCust.getMultiStageDesign().equalsIgnoreCase("true")) {
  63                     flowsToExecute.retainAll(Collections.singletonList(currentBB));
  64                 }
  65             }
  66         }
  67
  68     }
  69
  70 }