ebef4425eb3fca2e473dd76184aed9da2cdf596e
[so.git] /
1 package org.onap.so.adapters.tasks.audit;
2
3 import java.util.HashMap;
4 import java.util.Map;
5 import java.util.Optional;
6 import org.camunda.bpm.client.task.ExternalTask;
7 import org.camunda.bpm.client.task.ExternalTaskService;
8 import org.onap.logging.ref.slf4j.ONAPLogConstants;
9 import org.onap.so.audit.beans.AuditInventory;
10 import org.onap.so.logging.tasks.AuditMDCSetup;
11 import org.onap.so.objects.audit.AAIObjectAuditList;
12 import org.slf4j.Logger;
13 import org.slf4j.LoggerFactory;
14 import org.springframework.beans.factory.annotation.Autowired;
15 import org.springframework.stereotype.Component;
16
17 @Component
18 public class AuditQueryStackService extends AbstractAuditService {
19
20     private static final Logger logger = LoggerFactory.getLogger(AuditQueryStackService.class);
21
22     @Autowired
23     protected HeatStackAudit heatStackAudit;
24
25     @Autowired
26     protected AuditDataService auditDataService;
27
28     @Autowired
29     public AuditMDCSetup mdcSetup;
30
31     public AuditQueryStackService() {
32         super();
33     }
34
35     public void executeExternalTask(ExternalTask externalTask, ExternalTaskService externalTaskService) {
36         mdcSetup.setupMDC(externalTask);
37         AuditInventory auditInventory = externalTask.getVariable("auditInventory");
38         boolean success = false;
39         Map<String, Object> variables = new HashMap<>();
40         try {
41             Integer retryCount = externalTask.getRetries();
42             logger.info("Executing External Task Query Audit Inventory. Audit Inventory: {} \n Retry Number: {}",
43                     auditInventory.toString(), retryCount);
44
45             Optional<AAIObjectAuditList> auditList = heatStackAudit.queryHeatStack(auditInventory.getCloudOwner(),
46                     auditInventory.getCloudRegion(), auditInventory.getTenantId(), auditInventory.getHeatStackName());
47
48             if (auditList.isPresent()) {
49                 mdcSetup.setResponseCode(ONAPLogConstants.ResponseStatus.COMPLETE.toString());
50                 success = true;
51                 auditDataService.writeStackDataToRequestDb(auditInventory, auditList.get());
52             }
53             mdcSetup.setElapsedTime();
54             String externalTaskId = externalTask.getId();
55             if (success) {
56                 externalTaskService.complete(externalTask, variables);
57                 mdcSetup.setResponseCode(ONAPLogConstants.ResponseStatus.COMPLETE.toString());
58                 logger.debug("The External Task {}  was Successful", externalTaskId);
59                 logger.info(ONAPLogConstants.Markers.EXIT, "Exiting");
60                 mdcSetup.clearClientMDCs();
61             } else {
62                 if (retryCount == null) {
63                     logger.debug("The External Task {} Failed. Setting Retries to Default Start Value: {}",
64                             externalTaskId, getRetrySequence().length);
65                     externalTaskService.handleFailure(externalTask, UNABLE_TO_FIND_V_SERVERS_IN_OPENSTACK,
66                             UNABLE_TO_FIND_V_SERVERS_IN_OPENSTACK, getRetrySequence().length, 10000);
67                 } else if (retryCount == 1) {
68                     externalTaskService.complete(externalTask, variables);
69                     mdcSetup.setResponseCode(ONAPLogConstants.ResponseStatus.ERROR.toString());
70                     logger.debug("The External Task {} Failed. All Retries Exhausted", externalTaskId);
71                     logger.info(ONAPLogConstants.Markers.EXIT, "Exiting");
72                     mdcSetup.clearClientMDCs();
73                 } else {
74                     logger.debug("The External Task {} Failed. Decrementing Retries to {} , Retry Delay: ",
75                             externalTaskId, retryCount - 1, calculateRetryDelay(retryCount));
76                     externalTaskService.handleFailure(externalTask, UNABLE_TO_FIND_V_SERVERS_IN_OPENSTACK,
77                             UNABLE_TO_FIND_V_SERVERS_IN_OPENSTACK, retryCount - 1, calculateRetryDelay(retryCount));
78                 }
79                 logger.debug("The External Task {} Failed", externalTaskId);
80             }
81         } catch (Exception e) {
82             logger.error("Error during audit query of stack", e);
83         }
84     }
85 }