From 4a1f8524dd9c4e5b3124c1bc5ef63589de09b00a Mon Sep 17 00:00:00 2001
From: Jeremy Giberson <jeremyg@webpt.com>
Date: Sun, 8 Oct 2017 11:47:50 -0700
Subject: [PATCH] close server after context succeed

---
 index.js | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/index.js b/index.js
index 6eb3de7d..d056f56d 100644
--- a/index.js
+++ b/index.js
@@ -84,7 +84,10 @@ function forwardResponseToApiGateway(server, response, context) {
             const body = bodyBuffer.toString(isBase64Encoded ? 'base64' : 'utf8')
             const successResponse = {statusCode, body, headers, isBase64Encoded}
 
-            context.succeed(successResponse)
+            context.succeed(successResponse);
+            if (server && server.close) {
+                server.close();
+            }
         })
 }
 
@@ -98,6 +101,9 @@ function forwardConnectionErrorResponseToApiGateway(server, error, context) {
     }
 
     context.succeed(errorResponse)
+    if (server && server.close) {
+        server.close();
+    }
 }
 
 function forwardLibraryErrorResponseToApiGateway(server, error, context) {
@@ -110,6 +116,9 @@ function forwardLibraryErrorResponseToApiGateway(server, error, context) {
     }
 
     context.succeed(errorResponse)
+    if (server && server.close) {
+        server.close();
+    }
 }
 
 function forwardRequestToNodeServer(server, event, context) {