From 8409746072fc90dad4141fc4e99518dd9aa38216 Mon Sep 17 00:00:00 2001
From: vivek <konavivekramakrishna@gmail.com>
Date: Fri, 30 Aug 2024 14:36:58 +0530
Subject: [PATCH] add additional trycatch

---
 .../rest/NotificationRestController.java      | 44 +++++++++++++------
 1 file changed, 31 insertions(+), 13 deletions(-)

diff --git a/src/main/java/org/openelisglobal/notifications/rest/NotificationRestController.java b/src/main/java/org/openelisglobal/notifications/rest/NotificationRestController.java
index a45f3f2a3..2c8da886e 100644
--- a/src/main/java/org/openelisglobal/notifications/rest/NotificationRestController.java
+++ b/src/main/java/org/openelisglobal/notifications/rest/NotificationRestController.java
@@ -62,21 +62,39 @@ public List<Notification> getNotificationsByUserId(HttpServletRequest request) {
 
     @PostMapping("/notification/{userId}")
     public ResponseEntity<?> saveNotification(@PathVariable String userId, @RequestBody Notification notification) {
-        notification.setUser(systemUserService.getUserById(userId));
-        notification.setCreatedDate(OffsetDateTime.now());
-        notification.setReadAt(null);
-        notificationDAO.save(notification);
-
-        // Ensure BouncyCastleProvider is added for cryptographic operations
-        if (Security.getProvider(BouncyCastleProvider.PROVIDER_NAME) == null) {
-            Security.addProvider(new BouncyCastleProvider());
+        try {
+            // Set user and created date
+            notification.setUser(systemUserService.getUserById(userId));
+            notification.setCreatedDate(OffsetDateTime.now());
+            notification.setReadAt(null);
+
+            // Save notification
+            notificationDAO.save(notification);
+        } catch (Exception e) {
+            return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
+                    .body("Failed to save notification: " + e.getMessage());
         }
 
-        NotificationSubscriptions ns = notificationSubscriptionDAO
-                .getNotificationSubscriptionByUserId(Long.valueOf(userId));
+        try {
+            // Ensure BouncyCastleProvider is added for cryptographic operations
+            if (Security.getProvider(BouncyCastleProvider.PROVIDER_NAME) == null) {
+                Security.addProvider(new BouncyCastleProvider());
+            }
+        } catch (Exception e) {
+            return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
+                    .body("Failed to add BouncyCastleProvider: " + e.getMessage());
+        }
 
-        if (ns == null) {
-            return ResponseEntity.status(HttpStatus.NOT_FOUND).body("Subscription not found");
+        NotificationSubscriptions ns;
+        try {
+            // Get notification subscription by userId
+            ns = notificationSubscriptionDAO.getNotificationSubscriptionByUserId(Long.valueOf(userId));
+            if (ns == null) {
+                return ResponseEntity.status(HttpStatus.NOT_FOUND).body("Subscription not found for userId: " + userId);
+            }
+        } catch (Exception e) {
+            return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
+                    .body("Failed to retrieve notification subscription: " + e.getMessage());
         }
 
         try {
@@ -101,6 +119,7 @@ public ResponseEntity<?> saveNotification(@PathVariable String userId, @RequestB
             nl.martijndwars.webpush.Notification webPushNotification = new nl.martijndwars.webpush.Notification(
                     ns.getPfEndpoint(), ns.getPfP256dh(), ns.getPfAuth(), payloadJson);
 
+            // Send push notification
             HttpResponse response = pushService.send(webPushNotification);
 
             return ResponseEntity.ok()
@@ -109,7 +128,6 @@ public ResponseEntity<?> saveNotification(@PathVariable String userId, @RequestB
             return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
                     .body("Failed to send push notification: " + e.getMessage());
         }
-
     }
 
     @GetMapping("/notification/pnconfig")