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  * 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             ExecuteBuildingBlock currentBB = flowsToExecute.get(execution.getCurrentSequence());
  64             List<PreFlowManipulator> filtered = filterListeners(flowManipulators,
  65                     (item -> item.shouldRunFor(currentBB.getBuildingBlock().getBpmnFlowName(),
  66                             execution.getCurrentSequence() == 0, execution)));
  67
  68             logger.info("Running pre flow manipulators:\n{}",
  69                     filtered.stream().map(item -> item.getClass().getName()).collect(Collectors.joining("\n")));
  70             filtered.forEach(item -> item.run(flowsToExecute, currentBB, execution));
  71         } while (isBuildingBlockSkipped(sequenceBeforeFlowManipulator, execution));
  72     }
  73
  74     public void postModifyFlows(List<ExecuteBuildingBlock> flowsToExecute, BuildingBlockExecution execution) {
  75         int sequenceBeforeFlowManipulator;
  76         do {
  77             sequenceBeforeFlowManipulator = execution.getVariable(BBConstants.G_CURRENT_SEQUENCE);
  78             ExecuteBuildingBlock currentBB = flowsToExecute.get(execution.getCurrentSequence() - 1);
  79             List<PostFlowManipulator> filtered = filterListeners(postflowManipulators,
  80                     (item -> item.shouldRunFor(currentBB.getBuildingBlock().getBpmnFlowName(),
  81                             execution.getCurrentSequence() == 0, execution)));
  82
  83             logger.info("Running post flow manipulators:\n{}",
  84                     filtered.stream().map(item -> item.getClass().getName()).collect(Collectors.joining("\n")));
  85             filtered.forEach(item -> item.run(flowsToExecute, currentBB, execution));
  86         } while (isBuildingBlockSkipped(sequenceBeforeFlowManipulator, execution));
  87     }
  88
  89     private boolean isBuildingBlockSkipped(int sequenceBeforeFlowManipulator, BuildingBlockExecution execution) {
  90         return sequenceBeforeFlowManipulator != (int) execution.getVariable(BBConstants.G_CURRENT_SEQUENCE);
  91     }
  92 }