1 files changed, 158 insertions, 0 deletions
diff --git a/src/main/java/org/onap/nbi/apis/serviceorder/ServiceOrderResource.java b/src/main/java/org/onap/nbi/apis/serviceorder/ServiceOrderResource.java
new file mode 100644
index 0000000..1842cfe
--- /dev/null
+++ b/src/main/java/org/onap/nbi/apis/serviceorder/ServiceOrderResource.java
@@ -0,0 +1,158 @@
+package org.onap.nbi.apis.serviceorder;
+
+import java.util.Date;
+import java.util.List;
+import javax.validation.Valid;
+import org.onap.nbi.apis.serviceorder.model.ServiceOrder;
+import org.onap.nbi.apis.serviceorder.model.ServiceOrderItem;
+import org.onap.nbi.apis.serviceorder.model.StateType;
+import org.onap.nbi.apis.serviceorder.model.orchestrator.ServiceOrderInfo;
+import org.onap.nbi.apis.serviceorder.repositories.ServiceOrderInfoRepository;
+import org.onap.nbi.apis.serviceorder.repositories.ServiceOrderRepository;
+import org.onap.nbi.apis.serviceorder.workflow.CheckOrderConsistenceManager;
+import org.onap.nbi.apis.serviceorder.workflow.CreateAAICustomerManager;
+import org.onap.nbi.apis.serviceorder.workflow.CreateAAIServiceTypeManager;
+import org.onap.nbi.apis.serviceorder.workflow.SOTaskManager;
+import org.onap.nbi.commons.JsonRepresentation;
+import org.onap.nbi.commons.ResourceManagement;
+import org.onap.nbi.exceptions.ValidationException;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.data.mongodb.core.MongoTemplate;
+import org.springframework.data.mongodb.core.query.Query;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.MediaType;
+import org.springframework.http.ResponseEntity;
+import org.springframework.scheduling.annotation.EnableScheduling;
+import org.springframework.scheduling.annotation.Scheduled;
+import org.springframework.util.MultiValueMap;
+import org.springframework.validation.Errors;
+import org.springframework.web.bind.annotation.DeleteMapping;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PathVariable;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.RestController;
+
+@RestController
+@RequestMapping("/serviceOrder")
+@EnableScheduling
+public class ServiceOrderResource extends ResourceManagement<ServiceOrder> {
+
+    @Autowired
+    ServiceOrderRepository serviceOrderRepository;
+
+    @Autowired
+    CheckOrderConsistenceManager checkOrderConsistenceManager;
+
+    @Autowired
+    CreateAAICustomerManager createAAICustomer;
+
+    @Autowired
+    CreateAAIServiceTypeManager createAAIServiceType;
+
+    @Autowired
+    MongoTemplate mongoTemplate;
+
+    @Autowired
+    SOTaskManager serviceOrchestratorManager;
+
+    @Autowired
+    ServiceOrderInfoRepository serviceOrderInfoRepository;
+
+    @Autowired
+    MultiCriteriaRequestBuilder multiCriteriaRequestBuilder;
+
+
+    @GetMapping(value = "/{serviceOrderId}", produces = MediaType.APPLICATION_JSON_VALUE)
+    public ResponseEntity<Object> getServiceOrder(@PathVariable String serviceOrderId,
+            @RequestParam MultiValueMap<String, String> params) {
+
+        ServiceOrder serviceOrder = serviceOrderRepository.findOne(serviceOrderId);
+        if (serviceOrder == null) {
+            return ResponseEntity.notFound().build();
+        }
+
+        JsonRepresentation filter = new JsonRepresentation(params);
+        return this.getResponse(serviceOrder, filter);
+    }
+
+    @GetMapping(value = "", produces = MediaType.APPLICATION_JSON_VALUE)
+    public ResponseEntity<Object> findServiceOrder(@RequestParam MultiValueMap<String, String> params) {
+
+        Query query = multiCriteriaRequestBuilder.buildRequest(params);
+        List<ServiceOrder> serviceOrders = mongoTemplate.find(query, ServiceOrder.class);
+        JsonRepresentation filter = new JsonRepresentation(params);
+        long totalCount = serviceOrderRepository.count();
+        HttpHeaders headers = new HttpHeaders();
+        headers.add("X-Total-Count", String.valueOf(totalCount));
+        headers.add("X-Result-Count", String.valueOf(serviceOrders.size()));
+
+        ResponseEntity<Object> response = this.findResponse(serviceOrders, filter, headers);
+        return response;
+
+    }
+
+    @DeleteMapping(value = "/{serviceOrderId}")
+    public ResponseEntity<Object> deleteServiceOrder(@PathVariable String serviceOrderId) {
+
+        serviceOrderRepository.delete(serviceOrderId);
+
+        return this.deleteResponse();
+
+    }
+
+    @PostMapping(consumes = MediaType.APPLICATION_JSON_VALUE)
+    public ResponseEntity<Object> createServiceOrder(@Valid @RequestBody ServiceOrder serviceOrder, Errors errors,
+            @RequestParam MultiValueMap<String, String> params) {
+
+
+        if (errors != null && errors.hasErrors()) {
+            throw new ValidationException(errors.getAllErrors());
+        }
+
+        serviceOrder.setState(StateType.ACKNOWLEDGED);
+        serviceOrder.setOrderDate(new Date());
+        for (ServiceOrderItem serviceOrderItem : serviceOrder.getOrderItem()) {
+            serviceOrderItem.setState(StateType.ACKNOWLEDGED);
+        }
+
+        ServiceOrder serviceOrderSaved = serviceOrderRepository.save(serviceOrder);
+        serviceOrderSaved.setHref("serviceOrder/" + serviceOrderSaved.getId());
+        serviceOrderRepository.save(serviceOrderSaved);
+        JsonRepresentation filter = new JsonRepresentation(params);
+        return this.createResponse(serviceOrderSaved, filter);
+
+    }
+
+    @Scheduled(fixedDelay = 5000)
+    public void scheduleCheckServiceOrders() {
+        List<ServiceOrder> acknowledgedOrders = serviceOrderRepository.findByState(StateType.ACKNOWLEDGED);
+        for (ServiceOrder serviceOrder : acknowledgedOrders) {
+            ServiceOrderInfo serviceOrderInfo = checkOrderConsistenceManager.checkServiceOrder(serviceOrder);
+            if (serviceOrderInfo.isServiceOrderRejected()) {
+                changeServiceOrderState(serviceOrder, StateType.REJECTED);
+            } else if (serviceOrderInfo.isAllItemsCompleted()) {
+                changeServiceOrderState(serviceOrder, StateType.COMPLETED);
+            } else {
+                serviceOrderRepository.save(serviceOrder);
+                createAAICustomer.createAAICustomer(serviceOrderInfo);
+                createAAIServiceType.createAAIServiceType(serviceOrder, serviceOrderInfo);
+                serviceOrchestratorManager.registerServiceOrder(serviceOrder, serviceOrderInfo);
+            }
+        }
+    }
+
+    /**
+     *
+     * @param serviceOrder
+     * @param stateType
+     */
+    private void changeServiceOrderState(ServiceOrder serviceOrder, StateType stateType) {
+        serviceOrder.setState(stateType);
+        serviceOrder.setCompletionDateTime(new Date());
+        serviceOrderRepository.save(serviceOrder);
+    }
+
+}