wsgi: ensure to close the app response iterator (#29)

emmett-framework · Jan 16, 2023 · e9925c9 · e9925c9
1 parent ed0e12a
commit e9925c9
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/granian/wsgi.py b/granian/wsgi.py
@@ -62,9 +62,13 @@ def start_response(
             resp.headers = headers
 
         rv = callback(environ, start_response)
-        body = b"".join(rv)
-        if hasattr(rv, "close"):
-            rv.close()
+
+        try:
+            body = b"".join(rv)
+        finally:
+            if hasattr(rv, "close"):
+                rv.close()
+
         return (resp.status, resp.headers, body)
 
     return wrapper