Add bootstrap servers to Kafka Event (aws#216)

zabrip · carlzogh · web-flow · commit 2eb3f70503d1 · 2021-03-23T13:31:09.000Z
Co-authored-by: Carl Zogheib &lt;11421173+carlzogh@users.noreply.github.com&gt;
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
@@ -26,8 +26,9 @@
 /** Represents a Kafka Event. **/
 public class KafkaEvent {
     private Map<String, List<KafkaEventRecord>> records;
-    private  String eventSource;
+    private String eventSource;
     private String eventSourceArn;
+    private String bootstrapServers;
     
     @Data
     @NoArgsConstructor
diff --git a/aws-lambda-java-tests/pom.xml b/aws-lambda-java-tests/pom.xml
@@ -44,7 +44,7 @@
         <dependency>
             <groupId>com.amazonaws</groupId>
             <artifactId>aws-lambda-java-events</artifactId>
-            <version>3.6.0</version>
+            <version>3.7.0</version>
         </dependency>
         <dependency>
             <groupId>org.junit.jupiter</groupId>
@@ -253,4 +253,4 @@
             </plugin>
         </plugins>
     </build>
-</project>
+</project>
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
@@ -78,6 +78,7 @@ public void testLoadKafkaEvent() {
 
         assertThat(event).isNotNull();
         assertThat(event.getEventSourceArn()).isEqualTo("arn:aws:kafka:us-east-1:123456789012:cluster/vpc-3432434/4834-3547-3455-9872-7929");
+        assertThat(event.getBootstrapServers()).isEqualTo("b-2.demo-cluster-1.a1bcde.c1.kafka.us-east-1.amazonaws.com:9092,b-1.demo-cluster-1.a1bcde.c1.kafka.us-east-1.amazonaws.com:9092");
 
         KafkaEvent.KafkaEventRecord record = event.getRecords().get("mytopic-01").get(0);
         assertThat(record.getValue()).decodedAsBase64().asString().isEqualTo("Hello from Kafka !!");
diff --git a/aws-lambda-java-tests/src/test/resources/kafka_event.json b/aws-lambda-java-tests/src/test/resources/kafka_event.json
@@ -1,6 +1,7 @@
 {
   "eventSource": "aws:kafka",
   "eventSourceArn": "arn:aws:kafka:us-east-1:123456789012:cluster/vpc-3432434/4834-3547-3455-9872-7929",
+  "bootstrapServers": "b-2.demo-cluster-1.a1bcde.c1.kafka.us-east-1.amazonaws.com:9092,b-1.demo-cluster-1.a1bcde.c1.kafka.us-east-1.amazonaws.com:9092",
   "records": {
     "mytopic-01": [
       {
@@ -13,4 +14,4 @@
       }
     ]
   }
-}
+}

Original file line number	Diff line number	Diff line change
`@@ -1,6 +1,7 @@`
`1`	`1`	`{`
`2`	`2`	`"eventSource": "aws:kafka",`
`3`	`3`	`"eventSourceArn": "arn:aws:kafka:us-east-1:123456789012:cluster/vpc-3432434/4834-3547-3455-9872-7929",`
	`4`	`+ "bootstrapServers": "b-2.demo-cluster-1.a1bcde.c1.kafka.us-east-1.amazonaws.com:9092,b-1.demo-cluster-1.a1bcde.c1.kafka.us-east-1.amazonaws.com:9092",`
`4`	`5`	`"records": {`
`5`	`6`	`"mytopic-01": [`
`6`	`7`	`{`
`@@ -13,4 +14,4 @@`
`13`	`14`	`}`
`14`	`15`	`]`
`15`	`16`	`}`
`16`		`-}`
	`17`	`+}`