Improved lambda timings

This commit is contained in:
Terry Cain 2017-10-25 20:04:00 +01:00
commit 62a2d8f756
No known key found for this signature in database
GPG key ID: 14D90844E4E9B9F3
3 changed files with 36 additions and 8 deletions

View file

@ -298,7 +298,12 @@ class LambdaFunction(BaseModel):
volumes=["{}:/var/task".format(data_vol.name)], environment=env_vars, detach=True, **run_kwargs)
finally:
if container:
exit_code = container.wait()
try:
exit_code = container.wait(timeout=300)
except requests.exceptions.ReadTimeout:
exit_code = -1
container.stop()
container.kill()
output = container.logs(stdout=False, stderr=True)
output += container.logs(stdout=True, stderr=False)
container.remove()