Format

examples/powertools-examples-batch/README.md

@@ -1,6 +1,7 @@
                                          #  Powertools for AWS Lambda (Java) - Batch Example
 
-This project contains examples of Lambda function using the batch processing module of Powertools for AWS Lambda (Java). For more information on this module, please refer to the
+This project contains examples of Lambda function using the batch processing module of Powertools for AWS Lambda (Java).
+For more information on this module, please refer to the
 [documentation](https://docs.powertools.aws.dev/lambda-java/utilities/batch/).
 
 Three different examples and SAM deployments are included, covering each of the batch sources:
@@ -17,6 +18,7 @@ started with SAM in [the examples directory](../README.md)
 This sample contains three different deployments, depending on which batch processor you'd like to use, you can
 change to the subdirectory containing the example SAM template, and deploy. For instance, for the SQS batch
 deployment:
+
 ```bash
 cd deploy/sqs
 sam build
@@ -25,7 +27,7 @@ sam deploy --guided
 
 ## Test the application
 
-Each of the examples uses a Lambda scheduled every 5 minutes to push a batch, and a separate lambda to read it. To 
+Each of the examples uses a Lambda scheduled every 5 minutes to push a batch, and a separate lambda to read it. To
 see this in action, we can simply tail the logs of our stack:
 
 ```bash

examples/powertools-examples-batch/deploy/kinesis/template.yml

@@ -39,18 +39,18 @@ Resources:
           STREAM_NAME: !Ref DemoKinesisStream
       Policies:
         - Statement:
-          - Sid: WriteToKinesis
-            Effect: Allow
-            Action:
-              - kinesis:PutRecords
-              - kinesis:DescribeStream
-            Resource: !GetAtt DemoKinesisStream.Arn
+            - Sid: WriteToKinesis
+              Effect: Allow
+              Action:
+                - kinesis:PutRecords
+                - kinesis:DescribeStream
+              Resource: !GetAtt DemoKinesisStream.Arn
       Events:
         CWSchedule:
           Type: Schedule
           Properties:
             Schedule: 'rate(5 minutes)'
-            Name: !Join ["-", ["message-producer-schedule", !Select [0, !Split [-, !Select [2, !Split [/, !Ref AWS::StackId ]]]]]]
+            Name: !Join [ "-", [ "message-producer-schedule", !Select [ 0, !Split [ -, !Select [ 2, !Split [ /, !Ref AWS::StackId ] ] ] ] ] ]
             Description: Produce message to Kinesis via a Lambda function
             Enabled: true
 

examples/powertools-examples-batch/deploy/sqs/template.yml

@@ -72,24 +72,24 @@ Resources:
           QUEUE_URL: !Ref DemoSqsQueue
       Policies:
         - Statement:
-          - Sid: SQSSendMessageBatch
-            Effect: Allow
-            Action:
-              - sqs:SendMessageBatch
-              - sqs:SendMessage
-            Resource: !GetAtt DemoSqsQueue.Arn
-          - Sid: SQSKMSKey
-            Effect: Allow
-            Action:
-              - kms:GenerateDataKey
-              - kms:Decrypt
-            Resource: !GetAtt CustomerKey.Arn
+            - Sid: SQSSendMessageBatch
+              Effect: Allow
+              Action:
+                - sqs:SendMessageBatch
+                - sqs:SendMessage
+              Resource: !GetAtt DemoSqsQueue.Arn
+            - Sid: SQSKMSKey
+              Effect: Allow
+              Action:
+                - kms:GenerateDataKey
+                - kms:Decrypt
+              Resource: !GetAtt CustomerKey.Arn
       Events:
         CWSchedule:
           Type: Schedule
           Properties:
             Schedule: 'rate(5 minutes)'
-            Name: !Join ["-", ["message-producer-schedule", !Select [0, !Split [-, !Select [2, !Split [/, !Ref AWS::StackId ]]]]]]
+            Name: !Join [ "-", [ "message-producer-schedule", !Select [ 0, !Split [ -, !Select [ 2, !Split [ /, !Ref AWS::StackId ] ] ] ] ] ]
             Description: Produce message to SQS via a Lambda function
             Enabled: true
 

examples/powertools-examples-batch/pom.xml

@@ -1,6 +1,6 @@
 <?xml version="1.0" encoding="UTF-8"?>
-<project xmlns="http://maven.apache.org/POM/4.0.0"
-         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+<project xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xmlns="http://maven.apache.org/POM/4.0.0"
          xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
     <modelVersion>4.0.0</modelVersion>
 

examples/powertools-examples-batch/src/main/java/org/demo/batch/dynamo/DynamoDBStreamBatchHandler.java

@@ -6,7 +6,6 @@
 import com.amazonaws.services.lambda.runtime.events.StreamsEventResponse;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
-import org.demo.batch.model.Product;
 import software.amazon.lambda.powertools.batch.BatchMessageHandlerBuilder;
 import software.amazon.lambda.powertools.batch.handler.BatchMessageHandler;
 

examples/powertools-examples-batch/src/main/java/org/demo/batch/dynamo/DynamoDBWriter.java

@@ -3,6 +3,9 @@
 import com.amazonaws.services.lambda.runtime.Context;
 import com.amazonaws.services.lambda.runtime.RequestHandler;
 import com.amazonaws.services.lambda.runtime.events.ScheduledEvent;
+import java.security.SecureRandom;
+import java.util.UUID;
+import java.util.stream.IntStream;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 import org.demo.batch.model.DdbProduct;
@@ -14,10 +17,6 @@
 import software.amazon.awssdk.http.urlconnection.UrlConnectionHttpClient;
 import software.amazon.awssdk.services.dynamodb.DynamoDbClient;
 
-import java.security.SecureRandom;
-import java.util.UUID;
-import java.util.stream.IntStream;
-
 public class DynamoDBWriter implements RequestHandler<ScheduledEvent, String> {
 
     private static final Logger LOGGER = LogManager.getLogger(DynamoDBWriter.class);
@@ -54,9 +53,10 @@ public String handleRequest(ScheduledEvent scheduledEvent, Context context) {
             productBuilder.addPutItem(product);
         });
 
-        BatchWriteItemEnhancedRequest batchWriteItemEnhancedRequest = BatchWriteItemEnhancedRequest.builder().writeBatches(
-                productBuilder.build())
-                .build();
+        BatchWriteItemEnhancedRequest batchWriteItemEnhancedRequest =
+                BatchWriteItemEnhancedRequest.builder().writeBatches(
+                                productBuilder.build())
+                        .build();
 
         BatchWriteResult batchWriteResult = enhancedClient.batchWriteItem(batchWriteItemEnhancedRequest);
 

examples/powertools-examples-batch/src/main/java/org/demo/batch/kinesis/KinesisBatchSender.java

@@ -1,30 +1,24 @@
 package org.demo.batch.kinesis;
 
+import static java.util.stream.Collectors.toList;
+
 import com.amazonaws.services.lambda.runtime.Context;
 import com.amazonaws.services.lambda.runtime.RequestHandler;
 import com.amazonaws.services.lambda.runtime.events.ScheduledEvent;
 import com.fasterxml.jackson.core.JsonProcessingException;
 import com.fasterxml.jackson.databind.ObjectMapper;
+import java.security.SecureRandom;
+import java.util.List;
+import java.util.stream.IntStream;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 import org.demo.batch.model.Product;
-import software.amazon.awssdk.core.BytesWrapper;
 import software.amazon.awssdk.core.SdkBytes;
 import software.amazon.awssdk.http.urlconnection.UrlConnectionHttpClient;
 import software.amazon.awssdk.services.kinesis.KinesisClient;
 import software.amazon.awssdk.services.kinesis.model.PutRecordsRequest;
 import software.amazon.awssdk.services.kinesis.model.PutRecordsRequestEntry;
 import software.amazon.awssdk.services.kinesis.model.PutRecordsResponse;
-import software.amazon.awssdk.services.sqs.SqsClient;
-import software.amazon.awssdk.services.sqs.model.SendMessageBatchRequest;
-import software.amazon.awssdk.services.sqs.model.SendMessageBatchRequestEntry;
-import software.amazon.awssdk.services.sqs.model.SendMessageBatchResponse;
-
-import java.security.SecureRandom;
-import java.util.List;
-import java.util.stream.IntStream;
-
-import static java.util.stream.Collectors.toList;
 
 
 /**

examples/powertools-examples-batch/src/main/java/org/demo/batch/model/DdbProduct.java

@@ -11,13 +11,13 @@
  * limitations under the License.
  *
  */
+
 package org.demo.batch.model;
 
+import java.util.Objects;
 import software.amazon.awssdk.enhanced.dynamodb.mapper.annotations.DynamoDbBean;
 import software.amazon.awssdk.enhanced.dynamodb.mapper.annotations.DynamoDbPartitionKey;
 
-import java.util.Objects;
-
 @DynamoDbBean
 public class DdbProduct {
     private String id;
@@ -62,8 +62,12 @@ public void setPrice(double price) {
 
     @Override
     public boolean equals(Object o) {
-        if (this == o) return true;
-        if (o == null || getClass() != o.getClass()) return false;
+        if (this == o) {
+            return true;
+        }
+        if (o == null || getClass() != o.getClass()) {
+            return false;
+        }
         DdbProduct that = (DdbProduct) o;
         return Double.compare(that.price, price) == 0 && Objects.equals(id, that.id) && Objects.equals(name, that.name);
     }

examples/powertools-examples-batch/src/main/java/org/demo/batch/model/Product.java

@@ -11,6 +11,7 @@
  * limitations under the License.
  *
  */
+
 package org.demo.batch.model;
 
 import java.util.Objects;
@@ -57,8 +58,12 @@ public void setPrice(double price) {
 
     @Override
     public boolean equals(Object o) {
-        if (this == o) return true;
-        if (o == null || getClass() != o.getClass()) return false;
+        if (this == o) {
+            return true;
+        }
+        if (o == null || getClass() != o.getClass()) {
+            return false;
+        }
         Product product = (Product) o;
         return id == product.id && Double.compare(product.price, price) == 0 && Objects.equals(name, product.name);
     }

examples/powertools-examples-batch/src/main/java/org/demo/batch/sqs/SqsBatchHandler.java

@@ -4,14 +4,14 @@
 import com.amazonaws.services.lambda.runtime.RequestHandler;
 import com.amazonaws.services.lambda.runtime.events.SQSBatchResponse;
 import com.amazonaws.services.lambda.runtime.events.SQSEvent;
-import org.demo.batch.model.Product;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
+import org.demo.batch.model.Product;
 import software.amazon.lambda.powertools.batch.BatchMessageHandlerBuilder;
 import software.amazon.lambda.powertools.batch.handler.BatchMessageHandler;
 
 public class SqsBatchHandler implements RequestHandler<SQSEvent, SQSBatchResponse> {
-        private final static Logger LOGGER = LogManager.getLogger(SqsBatchHandler.class);
+    private final static Logger LOGGER = LogManager.getLogger(SqsBatchHandler.class);
     private final BatchMessageHandler<SQSEvent, SQSBatchResponse> handler;
 
     public SqsBatchHandler() {

examples/powertools-examples-batch/src/main/java/org/demo/batch/sqs/SqsBatchSender.java

@@ -1,10 +1,15 @@
 package org.demo.batch.sqs;
 
+import static java.util.stream.Collectors.toList;
+
 import com.amazonaws.services.lambda.runtime.Context;
 import com.amazonaws.services.lambda.runtime.RequestHandler;
 import com.amazonaws.services.lambda.runtime.events.ScheduledEvent;
 import com.fasterxml.jackson.core.JsonProcessingException;
 import com.fasterxml.jackson.databind.ObjectMapper;
+import java.security.SecureRandom;
+import java.util.List;
+import java.util.stream.IntStream;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 import org.demo.batch.model.Product;
@@ -14,12 +19,6 @@
 import software.amazon.awssdk.services.sqs.model.SendMessageBatchRequestEntry;
 import software.amazon.awssdk.services.sqs.model.SendMessageBatchResponse;
 
-import java.security.SecureRandom;
-import java.util.List;
-import java.util.stream.IntStream;
-
-import static java.util.stream.Collectors.toList;
-
 
 /**
  * A Lambda handler used to send message batches to SQS. This is only here

examples/powertools-examples-batch/src/main/resources/log4j2.xml

@@ -2,7 +2,7 @@
 <Configuration>
     <Appenders>
         <Console name="JsonAppender" target="SYSTEM_OUT">
-            <JsonTemplateLayout eventTemplateUri="classpath:LambdaJsonLayout.json" />
+            <JsonTemplateLayout eventTemplateUri="classpath:LambdaJsonLayout.json"/>
         </Console>
     </Appenders>
     <Loggers>

examples/powertools-examples-batch/template-dynamodb.yaml

@@ -66,7 +66,7 @@ Resources:
           Type: Schedule
           Properties:
             Schedule: 'rate(5 minutes)'
-            Name: !Join ["-", ["ddb-writer-schedule", !Select [0, !Split [-, !Select [2, !Split [/, !Ref AWS::StackId ]]]]]]
+            Name: !Join [ "-", [ "ddb-writer-schedule", !Select [ 0, !Split [ -, !Select [ 2, !Split [ /, !Ref AWS::StackId ] ] ] ] ] ]
             Description: Produce message to SQS via a Lambda function
             Enabled: true
 

powertools-batch/pom.xml

@@ -1,6 +1,6 @@
 <?xml version="1.0" encoding="UTF-8"?>
-<project xmlns="http://maven.apache.org/POM/4.0.0"
-         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+<project xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xmlns="http://maven.apache.org/POM/4.0.0"
          xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
     <modelVersion>4.0.0</modelVersion>
     <parent>

powertools-batch/src/main/java/software/amazon/lambda/powertools/batch/BatchMessageHandlerBuilder.java

@@ -1,6 +1,8 @@
 package software.amazon.lambda.powertools.batch;
 
-import software.amazon.lambda.powertools.batch.builder.*;
+import software.amazon.lambda.powertools.batch.builder.DynamoDbBatchMessageHandlerBuilder;
+import software.amazon.lambda.powertools.batch.builder.KinesisBatchMessageHandlerBuilder;
+import software.amazon.lambda.powertools.batch.builder.SqsBatchMessageHandlerBuilder;
 
 /**
  * A builder-style interface we can use to build batch processing handlers for SQS, Kinesis Streams,

powertools-batch/src/main/java/software/amazon/lambda/powertools/batch/builder/AbstractBatchMessageHandlerBuilder.java

@@ -1,17 +1,15 @@
 package software.amazon.lambda.powertools.batch.builder;
 
 import com.amazonaws.services.lambda.runtime.Context;
-import software.amazon.lambda.powertools.batch.handler.BatchMessageHandler;
-
 import java.util.function.BiConsumer;
 import java.util.function.Consumer;
+import software.amazon.lambda.powertools.batch.handler.BatchMessageHandler;
 
 /**
- *
  * An abstract class to capture common arguments used across all the message-binding-specific batch processing
  * builders. The builders provide a fluent interface to configure the batch processors. Any arguments specific
  * to a particular batch binding can be added to the child builder.
- *
+ * <p>
  * We capture types for the various messages involved, so that we can provide an interface that makes
  * sense for the concrete child.
  *
@@ -28,7 +26,7 @@ abstract class AbstractBatchMessageHandlerBuilder<T, C, E, R> {
      * Provides an (Optional!) success handler. A success handler is invoked
      * once for each message after it has been processed by the user-provided
      * handler.
-     *
+     * <p>
      * If the success handler throws, the item in the batch will be
      * marked failed.
      *
@@ -45,7 +43,7 @@ public C withSuccessHandler(Consumer<T> handler) {
      * user-provided handler. This gives the user's code a useful hook to do
      * anything else that might have to be done in response to a failure - for
      * instance, updating a metric, or writing a detailed log.
-     *
+     * <p>
      * Please note that this method has nothing to do with the partial batch
      * failure mechanism. Regardless of whether a failure handler is
      * specified, partial batch failures and responses to the Lambda environment
@@ -64,12 +62,12 @@ public C withFailureHandler(BiConsumer<T, Throwable> handler) {
      * takes a function that consumes a raw message and the Lambda context. This
      * is useful for handlers that need access to the entire message object, not
      * just the deserialized contents of the body.
-     *
+     * <p>
      * Note:  If you don't need the Lambda context, use the variant of this function
      * that does not require it.
      *
      * @param handler Takes a raw message - the underlying AWS Events Library event - to process.
-     *                  For instance for SQS this would be an SQSMessage.
+     *                For instance for SQS this would be an SQSMessage.
      * @return A BatchMessageHandler for processing the batch
      */
     public abstract BatchMessageHandler<E, R> buildWithRawMessageHandler(BiConsumer<T, Context> handler);
@@ -82,7 +80,7 @@ public C withFailureHandler(BiConsumer<T, Throwable> handler) {
      * just the deserialized contents of the body.
      *
      * @param handler Takes a raw message - the underlying AWS Events Library event - to process.
-     *                  For instance for SQS this would be an SQSMessage.
+     *                For instance for SQS this would be an SQSMessage.
      * @return A BatchMessageHandler for processing the batch
      */
     public BatchMessageHandler<E, R> buildWithRawMessageHandler(Consumer<T> handler) {
@@ -101,13 +99,14 @@ public BatchMessageHandler<E, R> buildWithRawMessageHandler(Consumer<T> handler)
      * @param handler Processes the deserialized body of the message
      * @return A BatchMessageHandler for processing the batch
      */
-    public abstract <M> BatchMessageHandler<E, R> buildWithMessageHandler(BiConsumer<M, Context> handler, Class<M> messageClass);
+    public abstract <M> BatchMessageHandler<E, R> buildWithMessageHandler(BiConsumer<M, Context> handler,
+                                                                          Class<M> messageClass);
 
     /**
      * Builds a BatchMessageHandler that can be used to process batches, given
      * a user-defined handler to process each item in the batch. This variant
      * takes a function that consumes the deserialized body of the given message
-     *  If deserialization fails, it will be treated as
+     * If deserialization fails, it will be treated as
      * failure of the processing of that item in the batch.
      * Note:  If you don't need the Lambda context, use the variant of this function
      * that does not require it.