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

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