src/main/java/org/onap/nbi/apis/serviceorder/workflow/SOTaskManager.java

   1 /**
   2  *     Copyright (c) 2018 Orange
   3  *
   4  *     Licensed under the Apache License, Version 2.0 (the "License");
   5  *     you may not use this file except in compliance with the License.
   6  *     You may obtain a copy of the License at
   7  *
   8  *         http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  *     Unless required by applicable law or agreed to in writing, software
  11  *     distributed under the License is distributed on an "AS IS" BASIS,
  12  *     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  *     See the License for the specific language governing permissions and
  14  *     limitations under the License.
  15  */
  16 package org.onap.nbi.apis.serviceorder.workflow;
  17
  18 import org.onap.nbi.apis.serviceorder.model.OrderItemRelationship;
  19 import org.onap.nbi.apis.serviceorder.model.ServiceOrder;
  20 import org.onap.nbi.apis.serviceorder.model.ServiceOrderItem;
  21 import org.onap.nbi.apis.serviceorder.model.orchestrator.ExecutionTask;
  22 import org.onap.nbi.apis.serviceorder.model.orchestrator.ServiceOrderInfo;
  23 import org.onap.nbi.apis.serviceorder.model.orchestrator.ServiceOrderInfoJson;
  24 import org.onap.nbi.apis.serviceorder.repositories.ExecutionTaskRepository;
  25 import org.onap.nbi.apis.serviceorder.repositories.ServiceOrderInfoRepository;
  26 import org.onap.nbi.apis.serviceorder.utils.JsonEntityConverter;
  27 import org.slf4j.Logger;
  28 import org.slf4j.LoggerFactory;
  29 import org.springframework.beans.factory.annotation.Autowired;
  30 import org.springframework.scheduling.annotation.EnableScheduling;
  31 import org.springframework.scheduling.annotation.Scheduled;
  32 import org.springframework.stereotype.Service;
  33 import java.util.ArrayList;
  34 import java.util.HashMap;
  35 import java.util.List;
  36 import java.util.Map;
  37
  38 @Service
  39 @EnableScheduling
  40 public class SOTaskManager {
  41
  42     @Autowired
  43     private ExecutionTaskRepository executionTaskRepository;
  44
  45     @Autowired
  46     private ServiceOrderInfoRepository serviceOrderInfoRepository;
  47
  48     @Autowired
  49     private SOTaskProcessor soTaskProcessor;
  50
  51     private static final Logger LOGGER = LoggerFactory.getLogger(SOTaskManager.class);
  52
  53     /**
  54      * @param orderItems
  55      * @param serviceOrderInfoJson
  56      */
  57     private void registerOrderItemExecutionPlan(List<ServiceOrderItem> orderItems,
  58             ServiceOrderInfoJson serviceOrderInfoJson) {
  59         List<ExecutionTask> executionTasksSaved = new ArrayList<>();
  60         Map<String, Long> internalIdOrderItemsMap = new HashMap<>();
  61         if (orderItems != null) {
  62             // first we save create all the execution tasks with order item id in relied tasks
  63             for (ServiceOrderItem orderItem : orderItems) {
  64                 ExecutionTask task = new ExecutionTask();
  65                 task.setOrderItemId(orderItem.getId());
  66                 task.setNbRetries(3);
  67                 StringBuilder sb = new StringBuilder();
  68                 for (OrderItemRelationship orderItemRelationship : orderItem.getOrderItemRelationship()) {
  69                     sb.append(orderItemRelationship.getId()).append(" ");
  70                 }
  71                 task.setReliedTasks(sb.toString());
  72                 task.setServiceOrderInfoJson(serviceOrderInfoJson);
  73                 ExecutionTask savedTask = executionTaskRepository.save(task);
  74                 executionTasksSaved.add(savedTask);
  75                 internalIdOrderItemsMap.put(savedTask.getOrderItemId(), savedTask.getInternalId());
  76             }
  77             // then we replace all orderitem ids in reliedtasks field with internalid of the tasks
  78             for (ExecutionTask executionTask : executionTasksSaved) {
  79                 for (String key : internalIdOrderItemsMap.keySet()) {
  80                     String replace = executionTask.getReliedTasks().replace(key,
  81                             String.valueOf(internalIdOrderItemsMap.get(key)));
  82                     executionTask.setReliedTasks(replace);
  83                 }
  84                 executionTaskRepository.save(executionTask);
  85             }
  86         }
  87     }
  88
  89     /**
  90      *
  91      * @param serviceOrder
  92      * @param serviceOrderInfo
  93      */
  94     public void registerServiceOrder(ServiceOrder serviceOrder, ServiceOrderInfo serviceOrderInfo) {
  95         String json = JsonEntityConverter.convertServiceOrderInfoToJson(serviceOrderInfo);
  96         ServiceOrderInfoJson serviceOrderInfoJson = new ServiceOrderInfoJson(serviceOrder.getId(), json);
  97         serviceOrderInfoRepository.save(serviceOrderInfoJson);
  98         registerOrderItemExecutionPlan(serviceOrder.getOrderItem(), serviceOrderInfoJson);
  99     }
 100
 101
 102     @Scheduled(fixedRate = 2000)
 103     private void processExecutionPlan() {
 104         List<ExecutionTask> taskToExecute = executionTaskRepository.findByReliedTasksIsEmpty();
 105         for (ExecutionTask executionTask : taskToExecute) {
 106             soTaskProcessor.processOrderItem(executionTask);
 107         }
 108     }
 109 }