Merge pull request #260 from ianpyw/master

keshayad · web-flow · commit da593256513d · 2021-07-29T20:38:22.000+01:00
Add headers in KafkaEventRecord
diff --git a/aws-lambda-java-events/src/main/java/com/amazonaws/services/lambda/runtime/events/KafkaEvent.java b/aws-lambda-java-events/src/main/java/com/amazonaws/services/lambda/runtime/events/KafkaEvent.java
@@ -42,6 +42,7 @@ public static class KafkaEventRecord {
         private String timestampType;
         private String key;
         private String value;
+        private List<Map<String, byte[]>> headers;
     }
 
     @Data
diff --git a/aws-lambda-java-tests/src/test/java/com/amazonaws/services/lambda/runtime/tests/EventLoaderTest.java b/aws-lambda-java-tests/src/test/java/com/amazonaws/services/lambda/runtime/tests/EventLoaderTest.java
@@ -82,6 +82,9 @@ public void testLoadKafkaEvent() {
 
         KafkaEvent.KafkaEventRecord record = event.getRecords().get("mytopic-01").get(0);
         assertThat(record.getValue()).decodedAsBase64().asString().isEqualTo("Hello from Kafka !!");
+         
+        String headerValue = new String(record.getHeaders().get(0).get("headerKey"));    
+        assertThat(headerValue).isEqualTo("headerValue");
     }
 
     @Test
diff --git a/aws-lambda-java-tests/src/test/resources/kafka_event.json b/aws-lambda-java-tests/src/test/resources/kafka_event.json
@@ -10,7 +10,24 @@
         "offset": 15,
         "timestamp": 1596480920837,
         "timestampType": "CREATE_TIME",
-        "value": "SGVsbG8gZnJvbSBLYWZrYSAhIQ=="
+        "value": "SGVsbG8gZnJvbSBLYWZrYSAhIQ==",
+        "headers": [
+          {
+            "headerKey": [
+              104,
+              101,
+              97,
+              100,
+              101,
+              114,
+              86,
+              97,
+              108,
+              117,
+              101
+            ]
+          }
+        ]
       }
     ]
   }

Original file line number	Diff line number	Diff line change
`@@ -42,6 +42,7 @@ public static class KafkaEventRecord {`
`42`	`42`	`private String timestampType;`
`43`	`43`	`private String key;`
`44`	`44`	`private String value;`
	`45`	`+ private List<Map<String, byte[]>> headers;`
`45`	`46`	`}`
`46`	`47`
`47`	`48`	`@Data`
Original file line number	Diff line number	Diff line change
`@@ -82,6 +82,9 @@ public void testLoadKafkaEvent() {`
`82`	`82`
`83`	`83`	`KafkaEvent.KafkaEventRecord record = event.getRecords().get("mytopic-01").get(0);`
`84`	`84`	`assertThat(record.getValue()).decodedAsBase64().asString().isEqualTo("Hello from Kafka !!");`
	`85`	`+`
	`86`	`+ String headerValue = new String(record.getHeaders().get(0).get("headerKey"));`
	`87`	`+ assertThat(headerValue).isEqualTo("headerValue");`
`85`	`88`	`}`
`86`	`89`
`87`	`90`	`@Test`