Merge pull request #115 from tgsong/patch-1

Fix runtime_client blocking main thread

Author: pushkarchawda

awslambdaric/lambda_runtime_client.py

@@ -3,6 +3,7 @@
 """
 
 import sys
+from concurrent.futures import ThreadPoolExecutor
 from awslambdaric import __version__
 
 
@@ -68,7 +69,9 @@ def post_init_error(self, error_response_data):
             raise LambdaRuntimeClientError(endpoint, response.code, response_body)
 
     def wait_next_invocation(self):
-        response_body, headers = runtime_client.next()
+        with ThreadPoolExecutor() as e:
+            fut = e.submit(runtime_client.next)
+        response_body, headers = fut.result()
         return InvocationRequest(
             invoke_id=headers.get("Lambda-Runtime-Aws-Request-Id"),
             x_amzn_trace_id=headers.get("Lambda-Runtime-Trace-Id"),