bpmn/MSOCommonBPMN/src/main/java/org/onap/so/bpmn/common/listener/flowmanipulator/FlowManipulatorListenerRunner.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP - SO
   4  * ================================================================================
   5  * Copyright (C) 2017 - 2019 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Copyright (C) 2020 Nokia.
   8  * ================================================================================
   9  * Licensed under the Apache License, Version 2.0 (the "License");
  10  * you may not use this file except in compliance with the License.
  11  * You may obtain a copy of the License at
  12  *
  13  *      http://www.apache.org/licenses/LICENSE-2.0
  14  *
  15  * Unless required by applicable law or agreed to in writing, software
  16  * distributed under the License is distributed on an "AS IS" BASIS,
  17  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  18  * See the License for the specific language governing permissions and
  19  * limitations under the License.
  20  * ============LICENSE_END=========================================================
  21  */
  22
  23 package org.onap.so.bpmn.common.listener.flowmanipulator;
  24
  25 import java.util.ArrayList;
  26 import java.util.HashMap;
  27 import java.util.List;
  28 import java.util.Optional;
  29 import java.util.stream.Collectors;
  30 import javax.annotation.PostConstruct;
  31 import org.onap.so.bpmn.common.BBConstants;
  32 import org.onap.so.bpmn.common.BuildingBlockExecution;
  33 import org.onap.so.bpmn.servicedecomposition.entities.ExecuteBuildingBlock;
  34 import org.onap.so.listener.ListenerRunner;
  35 import org.slf4j.Logger;
  36 import org.slf4j.LoggerFactory;
  37 import org.springframework.stereotype.Component;
  38
  39 @Component
  40 public class FlowManipulatorListenerRunner extends ListenerRunner {
  41
  42     private static Logger logger = LoggerFactory.getLogger(FlowManipulatorListenerRunner.class);
  43
  44     protected List<PreFlowManipulator> flowManipulators;
  45
  46     protected List<PostFlowManipulator> postflowManipulators;
  47
  48     @PostConstruct
  49     protected void init() {
  50
  51         flowManipulators = new ArrayList<>(
  52                 Optional.ofNullable(context.getBeansOfType(PreFlowManipulator.class)).orElse(new HashMap<>()).values());
  53
  54         postflowManipulators = new ArrayList<>(Optional.ofNullable(context.getBeansOfType(PostFlowManipulator.class))
  55                 .orElse(new HashMap<>()).values());
  56
  57     }
  58
  59     public void modifyFlows(List<ExecuteBuildingBlock> flowsToExecute, BuildingBlockExecution execution) {
  60         int sequenceBeforeFlowManipulator;
  61         do {
  62             sequenceBeforeFlowManipulator = execution.getVariable(BBConstants.G_CURRENT_SEQUENCE);
  63             if (sequenceBeforeFlowManipulator >= flowsToExecute.size()) {
  64                 break;
  65             }
  66             ExecuteBuildingBlock currentBB = flowsToExecute.get(execution.getCurrentSequence());
  67             List<PreFlowManipulator> filtered = filterListeners(flowManipulators,
  68                     (item -> item.shouldRunFor(currentBB.getBuildingBlock().getBpmnFlowName(),
  69                             execution.getCurrentSequence() == 0, execution)));
  70
  71             logger.info("Running pre flow manipulators:\n{}",
  72                     filtered.stream().map(item -> item.getClass().getName()).collect(Collectors.joining("\n")));
  73             filtered.forEach(item -> item.run(flowsToExecute, currentBB, execution));
  74         } while (isBuildingBlockSkipped(sequenceBeforeFlowManipulator, execution));
  75     }
  76
  77     public void postModifyFlows(List<ExecuteBuildingBlock> flowsToExecute, BuildingBlockExecution execution) {
  78         int sequenceBeforeFlowManipulator;
  79         do {
  80             sequenceBeforeFlowManipulator = execution.getVariable(BBConstants.G_CURRENT_SEQUENCE);
  81             ExecuteBuildingBlock currentBB = flowsToExecute.get(execution.getCurrentSequence() - 1);
  82             List<PostFlowManipulator> filtered = filterListeners(postflowManipulators,
  83                     (item -> item.shouldRunFor(currentBB.getBuildingBlock().getBpmnFlowName(),
  84                             execution.getCurrentSequence() == 0, execution)));
  85
  86             logger.info("Running post flow manipulators:\n{}",
  87                     filtered.stream().map(item -> item.getClass().getName()).collect(Collectors.joining("\n")));
  88             filtered.forEach(item -> item.run(flowsToExecute, currentBB, execution));
  89         } while (isBuildingBlockSkipped(sequenceBeforeFlowManipulator, execution));
  90     }
  91
  92     private boolean isBuildingBlockSkipped(int sequenceBeforeFlowManipulator, BuildingBlockExecution execution) {
  93         return sequenceBeforeFlowManipulator != (int) execution.getVariable(BBConstants.G_CURRENT_SEQUENCE);
  94     }
  95 }