WIP debugging overall status alert

Author: rladdusaw

src/main/java/edu/tamu/app/controller/ServiceController.java

@@ -6,19 +6,26 @@
 import static edu.tamu.framework.enums.BusinessValidationType.EXISTS;
 import static edu.tamu.framework.enums.BusinessValidationType.NONEXISTS;
 import static edu.tamu.framework.enums.BusinessValidationType.UPDATE;
+import static edu.tamu.app.enums.AppRole.ROLE_ANONYMOUS;
+import static edu.tamu.app.enums.AppRole.ROLE_USER;
 
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.messaging.simp.SimpMessagingTemplate;
 import org.springframework.web.bind.annotation.RestController;
 
+import edu.tamu.app.model.AppUser;
 import edu.tamu.app.model.Service;
+import edu.tamu.app.model.repo.AppUserRepo;
 import edu.tamu.app.model.repo.ServiceRepo;
+import edu.tamu.app.service.SystemMonitorService;
+import edu.tamu.framework.aspect.annotation.ApiCredentials;
 import edu.tamu.framework.aspect.annotation.ApiMapping;
 import edu.tamu.framework.aspect.annotation.ApiValidatedModel;
 import edu.tamu.framework.aspect.annotation.ApiValidation;
 import edu.tamu.framework.aspect.annotation.ApiVariable;
 import edu.tamu.framework.aspect.annotation.Auth;
 import edu.tamu.framework.model.ApiResponse;
+import edu.tamu.framework.model.Credentials;
 
 @RestController
 @ApiMapping("/service")
@@ -29,6 +36,12 @@ public class ServiceController {
 
     @Autowired
     private ServiceRepo serviceRepo;
+    
+    @Autowired
+    private SystemMonitorService systemMonitorService;
+    
+    @Autowired
+    private AppUserRepo userRepo;
 
     @ApiMapping("/all")
     @Auth(role="ROLE_ANONYMOUS")
@@ -52,21 +65,33 @@ public ApiResponse getService(@ApiVariable Long id) {
     @ApiMapping("/create")
     @Auth(role = "ROLE_SERVICE_MANAGER")
     @ApiValidation(business = { @ApiValidation.Business(value = CREATE), @ApiValidation.Business(value = EXISTS) })
-    public ApiResponse createService(@ApiValidatedModel Service service) {
+    public ApiResponse createService(@ApiValidatedModel Service service, @ApiCredentials Credentials credentials) {
         service = serviceRepo.create(service.getName(), service.getStatus(), service.getIsAuto(), service.getIsPublic(), service.getOnShortList(), service.getServiceUrl());
         simpMessagingTemplate.convertAndSend("/channel/service", new ApiResponse(SUCCESS, serviceRepo.findAll()));
+        sendStatusUpdate(service, credentials);
         return new ApiResponse(SUCCESS, service);
     }
 
     @ApiMapping("/update")
     @Auth(role = "ROLE_SERVICE_MANAGER")
     @ApiValidation(business = { @ApiValidation.Business(value = UPDATE), @ApiValidation.Business(value = NONEXISTS) })
-    public ApiResponse updateService(@ApiValidatedModel Service service) {
+    public ApiResponse updateService(@ApiValidatedModel Service service, @ApiCredentials Credentials credentials) {
         service = serviceRepo.save(service);
         simpMessagingTemplate.convertAndSend("/channel/service/" + service.getId(), new ApiResponse(SUCCESS, service));
+        sendStatusUpdate(service, credentials);
         return new ApiResponse(SUCCESS, service);
     }
 
+    private void sendStatusUpdate(Service service, Credentials credentials) {
+        System.out.println("Uhhhh ya!");
+        AppUser user = userRepo.findByUin(credentials.getUin());
+        if (user.getRole() == ROLE_ANONYMOUS || user.getRole() == ROLE_USER) {
+            simpMessagingTemplate.convertAndSend("/private/queue/status/overall-public", new ApiResponse(SUCCESS, systemMonitorService.getOverallStatusPublic()));
+        } else {
+            simpMessagingTemplate.convertAndSend("/private/queue/status/overall-full", new ApiResponse(SUCCESS, systemMonitorService.getOverallStatus()));
+        }
+    }
+    
     @ApiMapping("/remove")
     @Auth(role = "ROLE_SERVICE_MANAGER")
     @ApiValidation(business = { @ApiValidation.Business(value = DELETE), @ApiValidation.Business(value = NONEXISTS) })

src/main/java/edu/tamu/app/service/SystemMonitorService.java

@@ -18,6 +18,7 @@
 
 import edu.tamu.app.enums.Status;
 import edu.tamu.app.model.OverallStatus;
+import edu.tamu.app.model.repo.AppUserRepo;
 import edu.tamu.app.model.repo.ServiceRepo;
 import edu.tamu.framework.util.HttpUtility;
 
@@ -32,6 +33,9 @@ public class SystemMonitorService implements MonitorService {
 	@Autowired
 	ObjectMapper objectMapper;
 
+	@Autowired
+	AppUserRepo userRepo;
+	
 	private static final String SUCCESS_MESSAGE = "All services are working.";
 	private static final String ERROR_MESSAGE = "Some services are experiencing problems.";
 
@@ -62,8 +66,10 @@ public void updateAll() {
 	public OverallStatus getOverallStatus() {
 		Long downCount = serviceRepo.countByStatus(Status.DOWN);
 		if (downCount == 0) {
+		    System.out.println("All up");
 			return new OverallStatus(SUCCESS, SUCCESS_MESSAGE);
 		}
+		System.out.println("error");
 		return new OverallStatus(ERROR, ERROR_MESSAGE);
 	}
 
@@ -79,5 +85,4 @@ protected Status getServiceStatus(String serviceUrl) throws MalformedURLExceptio
 		List<Map<String,String>> mappedStatusResponse  = objectMapper.readValue(rawStatusResponse, new TypeReference<List<Map<String, String>>>(){});
 		return Status.valueOf(mappedStatusResponse.get(0).get("service").toUpperCase());
 	}
-
 }