From d939c74ba7dd8751970d81ee3d96ff2d2ccf02c5 Mon Sep 17 00:00:00 2001
From: Ershad T J <ershad@paladincloud.io>
Date: Tue, 17 Dec 2024 17:19:11 +0530
Subject: [PATCH] fix:asset-state-event queue url and fixed rabbit suggestions

---
 .../pacman/commons/aws/sqs/SQSManager.java    |  2 +-
 .../pacman/common/PacmanSdkConstants.java     |  2 +-
 .../pacman/executor/PolicyExecutor.java       | 26 ++++++++++++-------
 3 files changed, 19 insertions(+), 11 deletions(-)

diff --git a/commons/pac-batch-commons/src/main/java/com/tmobile/pacman/commons/aws/sqs/SQSManager.java b/commons/pac-batch-commons/src/main/java/com/tmobile/pacman/commons/aws/sqs/SQSManager.java
index 7e261050b..03b9d0518 100644
--- a/commons/pac-batch-commons/src/main/java/com/tmobile/pacman/commons/aws/sqs/SQSManager.java
+++ b/commons/pac-batch-commons/src/main/java/com/tmobile/pacman/commons/aws/sqs/SQSManager.java
@@ -57,7 +57,7 @@ public String sendSQSMessage(SQSBaseMessage sqsMessage, String url) {
             String message = objectMapper.writeValueAsString(sqsMessage);
             return sendMessage(message, url);
         } catch (Exception e) {
-            LOGGER.error("Unable to send SQS message", e);
+            LOGGER.error("Unable to send SQS message to URL {}: {}", url, e.getMessage(), e);
         }
         return null;
     }
diff --git a/jobs/pacman-rule-engine-2.0/src/main/java/com/tmobile/pacman/common/PacmanSdkConstants.java b/jobs/pacman-rule-engine-2.0/src/main/java/com/tmobile/pacman/common/PacmanSdkConstants.java
index 947c877b1..3a4eb23f0 100644
--- a/jobs/pacman-rule-engine-2.0/src/main/java/com/tmobile/pacman/common/PacmanSdkConstants.java
+++ b/jobs/pacman-rule-engine-2.0/src/main/java/com/tmobile/pacman/common/PacmanSdkConstants.java
@@ -701,7 +701,7 @@ public interface PacmanSdkConstants {
     String POLICY_DISABLED_MSG = "Policy has been disabled";
     String TENANT_ID = "TENANT_ID";
     String TENANT_NAME = "TENANT_NAME";
-    String ASSET_STATE_SQS_QUEUE_URL = "ASSET_STATE_SQS_URL";
+    String ASSET_STATE_TRIGGER_EVENT = "ASSET_STATE_TRIGGER_EVENT";
     String API_FEATURE_FLAGS = "api_feature_flags";
     String ENABLE_ASSET_STATE_SERVICE_FLAG_NAME = "enableAssetStateService";
     String ASSET_STATE_JOB = "asset-state-job";
diff --git a/jobs/pacman-rule-engine-2.0/src/main/java/com/tmobile/pacman/executor/PolicyExecutor.java b/jobs/pacman-rule-engine-2.0/src/main/java/com/tmobile/pacman/executor/PolicyExecutor.java
index aa7b63d3d..608d1a509 100755
--- a/jobs/pacman-rule-engine-2.0/src/main/java/com/tmobile/pacman/executor/PolicyExecutor.java
+++ b/jobs/pacman-rule-engine-2.0/src/main/java/com/tmobile/pacman/executor/PolicyExecutor.java
@@ -117,14 +117,18 @@ public PolicyExecutor (String jsonString) {
             logger.error("can't get the temp credentials");
         }
 
-        try {
-            amazonDynamoDB = AmazonDynamoDBClientBuilder
-                    .standard()
-                    .withRegion(region)
-                    .withCredentials(new AWSStaticCredentialsProvider(tempCredentials))
-                    .build();
-        } catch (Exception e) {
-            logger.error("Error initializing dynamoDb client for account {} , cause : {}", baseAccount, e.getMessage(), e);
+        if (tempCredentials != null) {
+            try {
+                amazonDynamoDB = AmazonDynamoDBClientBuilder
+                        .standard()
+                        .withRegion(region)
+                        .withCredentials(new AWSStaticCredentialsProvider(tempCredentials))
+                        .build();
+            } catch (Exception e) {
+                logger.error("Error initializing dynamoDb client for account {} , cause : {}", baseAccount, e.getMessage(), e);
+            }
+        } else {
+            logger.error("Cannot initialize AmazonDynamoDB client because tempCredentials is null");
         }
     }
 
@@ -195,6 +199,10 @@ public static void main(String[] args) {
     }
 
     private boolean isAssetStateSvcEnabled(String tenantId) {
+        if (amazonDynamoDB == null) {
+            logger.error("not able to fetch feature flag since, AmazonDynamoDB client is not initialized");
+            return false;
+        }
         Map<String, AttributeValue> expressionAttributeValues = new HashMap<>();
         expressionAttributeValues.put(":tenantId", new AttributeValue().withS(tenantId));
         QueryRequest queryRequest = new QueryRequest()
@@ -218,7 +226,7 @@ private boolean isAssetStateSvcEnabled(String tenantId) {
 
     private void sendAssetStateEvent(String source, String targetType) {
         logger.info("sending asset state event for source: {} and target type: {}");
-        String queueUrl = System.getenv(ASSET_STATE_SQS_QUEUE_URL);
+        String queueUrl = System.getenv(ASSET_STATE_TRIGGER_EVENT);
         String tenantId = System.getenv(TENANT_ID);
         String tenantName = System.getenv(TENANT_NAME);
         SQSManager sqsManager = SQSManager.getInstance();