Replace GatewayResponse.java with APIGatewayProxyResponseEvent

Author: sodre

src/main/java/co/zeroae/gate/App.java

@@ -16,15 +16,14 @@
 import java.io.File;
 import java.net.URL;
 import java.util.HashMap;
-import java.util.Map;
 import java.util.Objects;
 
 /**
  * This class implements a GATE application using AWS Lambda.
  * It loads the application from the .gapp file defined by the GATE_APP_FILE environment variable.
  * For every lambda invocation, it runs the application and outputs the result in GateXML format.
  */
-public class App implements RequestHandler<APIGatewayProxyRequestEvent, GatewayResponse> {
+public class App implements RequestHandler<APIGatewayProxyRequestEvent, APIGatewayProxyResponseEvent> {
     static {
         try {
             Gate.init();
@@ -36,12 +35,13 @@ public class App implements RequestHandler<APIGatewayProxyRequestEvent, GatewayR
     private static final Logger logger = LogManager.getLogger(App.class);
     private static final CorpusController application = loadApplication();
 
-    public GatewayResponse handleRequest(APIGatewayProxyRequestEvent input, final Context context) {
-        final Map<String, String> headers = new HashMap<>();
-        headers.put("Content-Type", "text/plain");
+    public APIGatewayProxyResponseEvent handleRequest(APIGatewayProxyRequestEvent input, final Context context) {
+        final APIGatewayProxyResponseEvent response = new APIGatewayProxyResponseEvent()
+                .withHeaders(new HashMap<>());
+        response.getHeaders().put("Content-Type", "text/plain");
 
         if (!input.getHeaders().get("Content-Type").equalsIgnoreCase("text/plain")) {
-            return new GatewayResponse("We only support text/plain input.", headers, 400);
+            return response.withBody("We only support text/plain input.").withStatusCode(400);
         }
 
         try {
@@ -50,18 +50,18 @@ public GatewayResponse handleRequest(APIGatewayProxyRequestEvent input, final Co
             corpus.add(doc);
             try {
                 application.execute();
-                headers.put("Content-Type", "application/xml");
-                return new GatewayResponse(doc.toXml(), headers, 200);
+                response.getHeaders().put("Content-Type", "application/xml");
+                return response.withBody(doc.toXml()).withStatusCode(200);
             } catch (ExecutionException e) {
                 logger.error(e);
-                return new GatewayResponse(e.getMessage(), headers, 500);
+                return response.withBody(e.getMessage()).withStatusCode(500);
             } finally {
                 corpus.clear();
                 Factory.deleteResource(doc);
             }
         } catch (ResourceInstantiationException e) {
             logger.warn(e);
-            return new GatewayResponse(e.getMessage(), headers, 400);
+            return response.withBody(e.getMessage()).withStatusCode(400);
         }
     }
 

src/main/java/co/zeroae/gate/GatewayResponse.java

@@ -1,6 +1,7 @@
 package co.zeroae.gate;
 
 import com.amazonaws.services.lambda.runtime.events.APIGatewayProxyRequestEvent;
+import com.amazonaws.services.lambda.runtime.events.APIGatewayProxyResponseEvent;
 import org.junit.Test;
 
 import java.util.Collections;
@@ -14,7 +15,7 @@ public class AppTest {
   @Test
   public void successfulResponse() throws Exception {
     App app = withEnvironmentVariable("GATE_APP_NAME", "annie")
-      .execute(() -> new App());
+      .execute(App::new);
 
     // Create the Input
     final HashMap<String, String> input_headers = new HashMap<>();
@@ -28,12 +29,12 @@ public void successfulResponse() throws Exception {
     final TestContext context = new TestContext();
 
     // Invoke the App
-    final GatewayResponse result = app.handleRequest(input, context);
+    final APIGatewayProxyResponseEvent result = app.handleRequest(input, context);
 
     // Assert Results
-    assertEquals(result.getStatusCode(), 200);
+    assertEquals(result.getStatusCode().intValue(), 200);
     assertEquals(result.getHeaders().get("Content-Type"), "application/xml");
-    String content = result.getBody();
+    final String content = result.getBody();
     assertNotNull(content);
     assertTrue(content.contains("GateDocument version=\"3\""));
   }