Common response class for ApiGateway and ALB

aws-lambda-java-events/src/main/java/com/amazonaws/services/lambda/runtime/events/ApplicationLoadBalancerRequestEvent.java

@@ -13,6 +13,9 @@
 
 package com.amazonaws.services.lambda.runtime.events;
 
+import com.amazonaws.services.lambda.runtime.events.helper.AwsLambdaRequestEvent;
+import com.amazonaws.services.lambda.runtime.events.helper.RequestSource;
+import com.fasterxml.jackson.annotation.JsonIgnore;
 import lombok.AllArgsConstructor;
 import lombok.Builder;
 import lombok.Data;
@@ -34,7 +37,13 @@
 @Builder(setterPrefix = "with")
 @NoArgsConstructor
 @AllArgsConstructor
-public class ApplicationLoadBalancerRequestEvent implements Serializable  {
+public class ApplicationLoadBalancerRequestEvent implements Serializable, AwsLambdaRequestEvent {
+
+    @Override
+    @JsonIgnore
+    public RequestSource getRequestSource() {
+        return RequestSource.ALB;
+    }
 
     @Data
     @Builder(setterPrefix = "with")

aws-lambda-java-events/src/main/java/com/amazonaws/services/lambda/runtime/events/AwsProxyResponseEvent.java

@@ -0,0 +1,33 @@
+package com.amazonaws.services.lambda.runtime.events;
+
+
+import com.fasterxml.jackson.annotation.JsonInclude;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+import java.util.List;
+import java.util.Map;
+
+/**
+ Common response class for APIGateway and ALB
+ */
+@NoArgsConstructor
+@Data
+@JsonInclude(JsonInclude.Include.NON_NULL)
+public class AwsProxyResponseEvent {
+    private static final long serialVersionUID = 2263167344670024138L;
+
+    private Integer statusCode;
+
+    private String statusDescription;
+
+    private List<String> cookies;
+
+    private Map<String, String> headers;
+
+    private Map<String, List<String>> multiValueHeaders;
+
+    private String body;
+
+    private Boolean isBase64Encoded;
+}

aws-lambda-java-events/src/main/java/com/amazonaws/services/lambda/runtime/events/apigateway/APIGatewayProxyRequestEvent.java

@@ -13,6 +13,9 @@
 
 package com.amazonaws.services.lambda.runtime.events.apigateway;
 
+import com.amazonaws.services.lambda.runtime.events.helper.AwsLambdaRequestEvent;
+import com.amazonaws.services.lambda.runtime.events.helper.RequestSource;
+import com.fasterxml.jackson.annotation.JsonIgnore;
 import lombok.AllArgsConstructor;
 import lombok.Builder;
 import lombok.Data;
@@ -30,7 +33,7 @@
 @Builder(setterPrefix = "with")
 @NoArgsConstructor
 @AllArgsConstructor
-public class APIGatewayProxyRequestEvent implements Serializable {
+public class APIGatewayProxyRequestEvent implements Serializable, AwsLambdaRequestEvent {
 
     private static final long serialVersionUID = 4189228800688527467L;
 
@@ -47,6 +50,12 @@ public class APIGatewayProxyRequestEvent implements Serializable {
     private String body;
     private Boolean isBase64Encoded;
 
+    @Override
+    @JsonIgnore
+    public RequestSource getRequestSource() {
+        return RequestSource.API_GATEWAY_REST;
+    }
+
     /**
      * class that represents proxy request context
      */

aws-lambda-java-events/src/main/java/com/amazonaws/services/lambda/runtime/events/apigateway/APIGatewayV2HTTPEvent.java

@@ -13,6 +13,9 @@
 
 package com.amazonaws.services.lambda.runtime.events.apigateway;
 
+import com.amazonaws.services.lambda.runtime.events.helper.AwsLambdaRequestEvent;
+import com.amazonaws.services.lambda.runtime.events.helper.RequestSource;
+import com.fasterxml.jackson.annotation.JsonIgnore;
 import lombok.AllArgsConstructor;
 import lombok.Builder;
 import lombok.Data;
@@ -30,7 +33,7 @@
 @Builder(setterPrefix = "with")
 @Data
 @NoArgsConstructor
-public class APIGatewayV2HTTPEvent {
+public class APIGatewayV2HTTPEvent implements AwsLambdaRequestEvent {
 
     private String version;
     private String routeKey;
@@ -45,6 +48,12 @@ public class APIGatewayV2HTTPEvent {
     private boolean isBase64Encoded;
     private RequestContext requestContext;
 
+    @Override
+    @JsonIgnore
+    public RequestSource getRequestSource() {
+        return RequestSource.API_GATEWAY_HTTP;
+    }
+
     @AllArgsConstructor
     @Builder(setterPrefix = "with")
     @Data

aws-lambda-java-events/src/main/java/com/amazonaws/services/lambda/runtime/events/helper/AwsLambdaRequestEvent.java

@@ -0,0 +1,8 @@
+package com.amazonaws.services.lambda.runtime.events.helper;
+
+/**
+ * Interface to abstract APIGWY and ALB request classes, reduce complexity and increase efficiency.
+ */
+public interface AwsLambdaRequestEvent {
+    RequestSource getRequestSource();
+}