more logs

CMSgov · Apr 3, 2024 · 979c37a · 979c37a
1 parent 79da7be
commit 979c37a
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/optout/src/main/java/gov/cms/ab2d/optout/OptOutHandler.java b/optout/src/main/java/gov/cms/ab2d/optout/OptOutHandler.java
@@ -8,6 +8,8 @@
 import org.json.simple.JSONObject;
 import org.json.simple.parser.JSONParser;
 
+import java.sql.Timestamp;
+
 import static gov.cms.ab2d.optout.OptOutConstants.ENDPOINT;
 
 public class OptOutHandler implements RequestHandler<SQSEvent, Void> {
@@ -25,6 +27,7 @@ public void processSQSMessage(SQSEvent.SQSMessage msg, Context context) {
         var logger = context.getLogger();
         try {
             logger.log("OptOut Lambda started. Processing message from SQS " + msg.getBody());
+            var start = System.nanoTime();
 
             JSONParser parser = new JSONParser();
             JSONObject json = (JSONObject) parser.parse(msg.getBody());
@@ -33,6 +36,8 @@ public void processSQSMessage(SQSEvent.SQSMessage msg, Context context) {
 
             var optOutProcessing = processorInit(logger);
             optOutProcessing.process(getFileName(notification), getBucketName(notification), ENDPOINT);
+            var stop = System.nanoTime();
+            logger.log("TIME   " + (stop - start));
         } catch (Exception ex) {
             logger.log("An error occurred");
             throw new OptOutException("An error occurred", ex);

diff --git a/optout/src/main/java/gov/cms/ab2d/optout/OptOutProcessor.java b/optout/src/main/java/gov/cms/ab2d/optout/OptOutProcessor.java
@@ -78,7 +78,9 @@ public void processFileFromS3(BufferedReader reader) {
         String line;
         var lineNumber = 0L;
         try (var dbConnection = DriverManager.getConnection(parameterStore.getDbHost(), parameterStore.getDbUser(), parameterStore.getDbPassword())){
+            logger.log("Connection: " + dbConnection);
             while ((line = reader.readLine()) != null) {
+                logger.log("lineNumber " + lineNumber);
                 if (!line.startsWith(HEADER_RESP) && !line.startsWith(TRAILER_RESP)) {
                     var optOutInformation = createOptOutInformation(line);
                     optOutInformationMap.put(lineNumber, optOutInformation);
@@ -94,6 +96,7 @@ public void processFileFromS3(BufferedReader reader) {
     public OptOutInformation createOptOutInformation(String information) {
         var mbi = information.substring(MBI_INDEX_START, MBI_INDEX_END).trim();
         var optOutFlag = (information.charAt(OPTOUT_FLAG_INDEX) == 'Y');
+        logger.log("OptOutInformation " + optOutFlag);
         return new OptOutInformation(mbi, optOutFlag);
     }
 
@@ -103,6 +106,7 @@ public void updateOptOut(long lineNumber, OptOutInformation optOutInformation, C
             statement.execute();
         } catch (SQLException ex) {
             logger.log("There is an insertion error on the line " + lineNumber);
+            logger.log(ex.getMessage());
             isRejected = true;
         }
     }

diff --git a/optout/src/main/java/gov/cms/ab2d/optout/OptOutS3.java b/optout/src/main/java/gov/cms/ab2d/optout/OptOutS3.java
@@ -69,6 +69,7 @@ public String createResponseOptOutFile(String responseContent) {
                     .build();
 
             s3Client.putObject(objectRequest, RequestBody.fromString(responseContent));
+            logger.log("Response file done");
             return key;
         } catch (AmazonS3Exception ex) {
             var errorMessage = "Response OptOut file cannot be created. ";
@@ -85,6 +86,7 @@ public void deleteFileFromS3() {
                     .build();
 
             s3Client.deleteObject(request);
+            logger.log("delete file done");
         } catch (SdkClientException ex) {
             logger.log(ex.getMessage());
         }