Handle request body read timeouts

raphaelm · Apr 3, 2016 · 3bb5ee1 · 3bb5ee1
1 parent c59d3a5
commit 3bb5ee1
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/cockatiel/handlers.py b/cockatiel/handlers.py
@@ -61,6 +61,11 @@ def put_file(request: web.Request):
 
     with tempfile.SpooledTemporaryFile(max_size=1024 * 1024) as tmpfile:
         for chunk in request_chunks(request):
+            if isinstance(chunk, asyncio.Future):
+                try:
+                    chunk = yield from asyncio.wait_for(chunk, timeout=60)
+                except asyncio.TimeoutError:
+                    raise web.HTTPRequestTimeout()
             checksum.update(chunk)
             tmpfile.write(chunk)