fix : update notification-invoke-service to accept list of notificati…

…ons or single notification in the payload
PaladinCloud · Apr 20, 2023 · 1e9792b · 1e9792b
1 parent 1f14989
commit 1e9792b
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/installer/settings/common.py b/installer/settings/common.py
@@ -75,7 +75,8 @@
     'notification.appsync' : {'tags' : ["notification"]}
 }
 
-LAMBDA_PATH = "V3"
+
+LAMBDA_PATH = "V4"
 DATA_DIR = os.path.join(BASE_APP_DIR, 'data')
 LOG_DIR = os.path.join(BASE_APP_DIR, 'log')
 PROVISIONER_FILES_DIR_TO_COPY = os.path.join(BASE_APP_DIR, 'files')

diff --git a/...unctions/notification-invoke-service/src/main/java/com/paladincloud/SendNotification.java b/...unctions/notification-invoke-service/src/main/java/com/paladincloud/SendNotification.java
@@ -40,6 +40,26 @@ public String handleRequest(Map<String,Object> event, Context context)
                 System.out.println("bodyJson--"+bodyJson.toString());
                 payloadString = bodyJson.getAsJsonObject().get("Sns").getAsJsonObject().get("Message").getAsString();
             }
+
+            Object payloadObj = gson.fromJson(payloadString,Object.class);
+            if(payloadObj instanceof List){
+                List<Object> notificationRequestList = gson.fromJson(payloadString,List.class);
+                notificationRequestList.stream().forEach(obj -> {
+                    String notificationRequestStr = gson.toJson(obj);
+                    System.out.println("notificationsrequeststr for list---"+notificationRequestStr);
+                    PublishRequest request = new PublishRequest(snsTopicArn,notificationRequestStr);
+                    PublishResult result = client.publish(request);
+                    logger.log("message sent with id "+result.getMessageId());
+                });
+            }
+            else{
+                System.out.println("notificationsrequeststr for single event---"+payloadObj.toString());
+                PublishRequest request = new PublishRequest(snsTopicArn,payloadObj.toString());
+                PublishResult result = client.publish(request);
+                logger.log("message sent with id "+result.getMessageId());
+            }
+
+
             List<Object> notificationRequestList = gson.fromJson(payloadString,List.class);
             notificationRequestList.stream().forEach(obj -> {
                 String notificationRequestStr = gson.toJson(obj);