set shutdown timeout to AppRunner instead of site

aiokitchen · Jul 19, 2024 · 8ffc5a0 · 8ffc5a0
1 parent c03dd1e
commit 8ffc5a0
Show file tree

Hide file tree

Showing 3 changed files with 313 additions and 304 deletions.
diff --git a/aiomisc/service/aiohttp.py b/aiomisc/service/aiohttp.py
@@ -74,14 +74,12 @@ async def create_site(self) -> SockSite:
         if getattr(self, "runner", None) is None:
             raise RuntimeError("runner already created")
 
-        return SockSite(
-            self.runner, self.socket,
-            shutdown_timeout=self.shutdown_timeout,
-        )
+        return SockSite(self.runner, self.socket)
 
     async def make_runner(self, application: Application) -> AppRunner:
         return AppRunner(
             application,
+            shutdown_timeout=self.shutdown_timeout,
             **self.runner_kwargs,
         )
 
@@ -129,7 +127,6 @@ async def create_site(self) -> SockSite:
 
         return SockSite(
             self.runner, self.socket,
-            shutdown_timeout=self.shutdown_timeout,
             ssl_context=await self.loop.run_in_executor(
                 None, self.__ssl_options.create_context,
             ),