Merge from ECOMP's repository
[vid.git] / vid-app-common / src / main / java / org / onap / vid / controller / ProbeController.java
@@ -1,8 +1,11 @@
-package org.onap.vid.controllers;
+package org.onap.vid.controller;
 
+import org.onap.portalsdk.core.controller.RestrictedBaseController;
 import org.onap.vid.aai.AaiClient;
 import org.onap.vid.model.probes.ExternalComponentStatus;
-import org.onap.portalsdk.core.controller.RestrictedBaseController;
+import org.onap.vid.mso.MsoBusinessLogic;
+import org.onap.vid.scheduler.SchedulerService;
+import org.onap.vid.services.VidService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RequestMethod;
@@ -14,13 +17,25 @@ import java.util.List;
 @RestController
 @RequestMapping("probe")
 public class ProbeController extends RestrictedBaseController {
+
+    private final AaiClient aaiClient;
+    private final VidService vidService;
+    private final MsoBusinessLogic msoBusinessLogic;
+    private final SchedulerService schedulerService;
+
     @Autowired
-    private AaiClient aaiClient;
+    public ProbeController(AaiClient aaiClient, VidService vidService, MsoBusinessLogic msoBusinessLogic, SchedulerService schedulerService) {
+        this.aaiClient = aaiClient;
+        this.vidService = vidService;
+        this.msoBusinessLogic = msoBusinessLogic;
+        this.schedulerService = schedulerService;
+    }
 
     @RequestMapping(method= RequestMethod.GET)
-    public List<ExternalComponentStatus> getProbe(){
+    public List<ExternalComponentStatus> getProbe() {
         List<ExternalComponentStatus> componentStatuses = new ArrayList<>();
         componentStatuses.add(aaiClient.probeAaiGetAllSubscribers());
+        componentStatuses.add(schedulerService.probeGetSchedulerChangeManagements());
         return componentStatuses;
     }