Merge pull request #44 from klarna-incubator/sink-refactor

Sink refactor
klarna-incubator · Feb 22, 2021 · 5f91906 · 5f91906
2 parents 03e761a + 5b3e6b9
commit 5f91906
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pom.xml b/pom.xml
@@ -6,7 +6,7 @@
     <modelVersion>4.0.0</modelVersion>
     <packaging>jar</packaging>
     <groupId>com.klarna</groupId>
-    <version>1.1.0</version>
+    <version>1.1.1</version>
     <artifactId>flink-connector-dynamodb</artifactId>
 
     <licenses>

diff --git a/src/main/java/com/klarna/flink/connectors/dynamodb/DynamoDBProducer.java b/src/main/java/com/klarna/flink/connectors/dynamodb/DynamoDBProducer.java
@@ -241,7 +241,8 @@ protected Callable<BatchResponse> batchWrite(final BatchRequest batchRequest) {
                     t = null;
                     try {
                         final BatchWriteItemResponse batchWriteItemResponse = dynamoDbClient.batchWriteItem(batchWriteItemRequest);
-                        if (batchWriteItemResponse.hasUnprocessedItems()) {
+                        Map<String, List<WriteRequest>> unprocessedItems = batchWriteItemResponse.unprocessedItems();
+                        if (!unprocessedItems.isEmpty()) {
                             retry = true;
                             batchWriteItemRequest = BatchWriteItemRequest.builder()
                                     .requestItems(batchWriteItemResponse.unprocessedItems())