optimize

nggit · Dec 30, 2024 · 8f3ad09 · 8f3ad09
1 parent 467eccd
commit 8f3ad09
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/tremolo/lib/http_protocol.py b/tremolo/lib/http_protocol.py
@@ -281,15 +281,15 @@ async def _handle_request(self, header):
             if not self._waiters['request'].done():
                 self._waiters['request'].set_result(None)
 
-            if self.request is not None:
-                timer = self.set_handler_timeout(
-                    self.options['app_handler_timeout']
-                )
+            timer = self.set_handler_timeout(
+                self.options['app_handler_timeout']
+            )
 
-                try:
+            try:
+                if self.request is not None:
                     await self.headers_received(response)
-                finally:
-                    timer.cancel()
+            finally:
+                timer.cancel()
         except (asyncio.CancelledError, Exception) as exc:
             data = None
 
@@ -460,9 +460,6 @@ async def _handle_keepalive(self):
         self.transport.resume_reading()
 
     def connection_lost(self, _):
-        self.set_handler_timeout(self.options['app_close_timeout'])
-        self.globals.connections.discard(self)
-
         while self.tasks:
             task = self.tasks.pop()
 
@@ -485,3 +482,6 @@ def connection_lost(self, _):
         self.context.update(transport=None)
         self.request = None
         self._header_buf = None
+
+        self.set_handler_timeout(self.options['app_close_timeout'])
+        self.globals.connections.discard(self)