Add Apache license header per file
[externalapi/nbi.git] / 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 }