Merge pull request #121 from alexanderson1993/master

Clears the invocation timeout if the invocation is a success.
netlify · Feb 13, 2019 · 879211d · 879211d
2 parents 6f470ae + 20cb467
commit 879211d
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/lib/serve.js b/lib/serve.js
@@ -135,12 +135,20 @@ function createHandler(dir, static, timeout) {
       { clientContext: buildClientContext(request.headers) || {} },
       callback
     );
+
+    var invocationTimeoutRef = null
+
     Promise.race([
       promiseCallback(promise, callback),
-      setTimeout(function() {
-        handleInvocationTimeout(response, timeout)
-      }, timeout * 1000)
-    ])
+      new Promise(function(resolve) {
+        invocationTimeoutRef = setTimeout(function() {
+          handleInvocationTimeout(response, timeout)
+          resolve()
+        }, timeout * 1000)
+      })
+    ]).finally(() => {
+      clearTimeout(invocationTimeoutRef)
+    })
   };
 }