omserv 0.0.0.dev177__py3-none-any.whl → 0.0.0.dev179__py3-none-any.whl

omserv/server/__init__.py

@@ -15,4 +15,4 @@ See:
  - https://github.com/tiangolo/fastapi
 """
 from .config import Config  # noqa
-from .workers import serve  # noqa
+from .default import serve  # noqa

omserv/server/default.py

@@ -0,0 +1,34 @@
+import typing as ta
+
+import anyio.abc
+
+from omlish import inject as inj
+
+from .config import Config
+from .inject import bind_server
+from .listener import Listener
+from .sockets import Sockets
+from .types import AppWrapper
+from .types import AsgiFramework
+from .types import wrap_app
+
+
+async def serve(
+        app: AsgiFramework | AppWrapper,
+        config: Config,
+        *,
+        sockets: Sockets | None = None,
+        shutdown_trigger: ta.Callable[..., ta.Awaitable[None]] | None = None,
+        handle_shutdown_signals: bool = False,
+        task_status: anyio.abc.TaskStatus[ta.Sequence[str]] = anyio.TASK_STATUS_IGNORED,
+) -> None:
+    injector = inj.create_injector(bind_server(config))
+    listener = injector.inject(Listener)
+    await listener.listen(
+        wrap_app(app),
+        config,
+        sockets=sockets,
+        shutdown_trigger=shutdown_trigger,
+        handle_shutdown_signals=handle_shutdown_signals,
+        task_status=task_status,
+    )

omserv/server/inject.py

@@ -0,0 +1,21 @@
+import functools
+
+from omlish import inject as inj
+
+from .config import Config
+from .server import Server
+from .server import ServerFactory
+
+
+def _provide_server_factory(config: Config) -> ServerFactory:
+    return ServerFactory(functools.partial(Server, config=config))
+
+
+def bind_server(
+        config: Config,
+) -> inj.Elemental:
+    return inj.as_elements(
+        inj.bind(config),
+
+        inj.bind(_provide_server_factory, singleton=True),
+    )

omserv/server/{workers.py → listener.py}

@@ -11,10 +11,10 @@ import anyio.abc
 
 from .config import Config
 from .lifespans import Lifespan
+from .server import ServerFactory
 from .sockets import Sockets
 from .sockets import create_sockets
 from .sockets import repr_socket_addr
-from .tcpserver import TcpServer
 from .types import AppWrapper
 from .types import AsgiFramework
 from .types import wrap_app
@@ -125,69 +125,80 @@ async def _install_signal_handler(
     return signal_event.wait
 
 
-async def serve(
-        app: AsgiFramework | AppWrapper,
-        config: Config,
-        *,
-        sockets: Sockets | None = None,
-        shutdown_trigger: ta.Callable[..., ta.Awaitable[None]] | None = None,
-        handle_shutdown_signals: bool = False,
-        task_status: anyio.abc.TaskStatus[ta.Sequence[str]] = anyio.TASK_STATUS_IGNORED,
-) -> None:
-    app = wrap_app(app)
-
-    lifespan = Lifespan(app, config)
-    max_requests = None
-    if config.max_requests is not None:
-        max_requests = config.max_requests + random.randint(0, config.max_requests_jitter)
-    context = WorkerContext(max_requests)
+class Listener:
+    def __init__(
+            self,
+            *,
+            server_factory: ServerFactory,
+    ) -> None:
+        super().__init__()
+
+        self._server_factory = server_factory
+
+    async def listen(
+            self,
+            app: AsgiFramework | AppWrapper,
+            config: Config,
+            *,
+            sockets: Sockets | None = None,
+            shutdown_trigger: ta.Callable[..., ta.Awaitable[None]] | None = None,
+            handle_shutdown_signals: bool = False,
+            task_status: anyio.abc.TaskStatus[ta.Sequence[str]] = anyio.TASK_STATUS_IGNORED,
+    ) -> None:
+        app = wrap_app(app)
+
+        lifespan = Lifespan(app, config)
+        max_requests = None
+        if config.max_requests is not None:
+            max_requests = config.max_requests + random.randint(0, config.max_requests_jitter)
+        context = WorkerContext(max_requests)
+
+        async with anyio.create_task_group() as lifespan_task_group:
+            if shutdown_trigger is None and handle_shutdown_signals:
+                shutdown_trigger = await _install_signal_handler(lifespan_task_group)
+
+            await lifespan_task_group.start(lifespan.handle_lifespan)
+            await lifespan.wait_for_startup()
+
+            async with anyio.create_task_group() as server_task_group:
+                if sockets is None:
+                    sockets = create_sockets(config)
+                    for sock in sockets.insecure_sockets:
+                        sock.listen(config.backlog)
+
+                listeners = []
+                binds = []
 
-    async with anyio.create_task_group() as lifespan_task_group:
-        if shutdown_trigger is None and handle_shutdown_signals:
-            shutdown_trigger = await _install_signal_handler(lifespan_task_group)
-
-        await lifespan_task_group.start(lifespan.handle_lifespan)
-        await lifespan.wait_for_startup()
-
-        async with anyio.create_task_group() as server_task_group:
-            if sockets is None:
-                sockets = create_sockets(config)
                 for sock in sockets.insecure_sockets:
-                    sock.listen(config.backlog)
-
-            listeners = []
-            binds = []
-
-            for sock in sockets.insecure_sockets:
-                listeners.append(anyio._core._eventloop.get_async_backend().create_tcp_listener(sock))  # noqa
-                bind = repr_socket_addr(sock.family, sock.getsockname())
-                binds.append(f'http://{bind}')
-                log.info('Running on http://%s (CTRL + C to quit)', bind)
-
-            task_status.started(binds)
-            try:
-                async with anyio.create_task_group() as task_group:
-                    if shutdown_trigger is not None:
-                        task_group.start_soon(raise_shutdown, shutdown_trigger)
-                    task_group.start_soon(raise_shutdown, context.terminate.wait)
-
-                    task_group.start_soon(
-                        functools.partial(
-                            serve_listeners,
-                            functools.partial(TcpServer, app, config, context),
-                            listeners,
-                            handler_task_group=server_task_group,
-                        ),
-                    )
-
-                    await anyio.sleep_forever()
-            except BaseExceptionGroup as error:
-                _, other_errors = error.split((ShutdownError, KeyboardInterrupt))  # noqa
-                if other_errors is not None:
-                    raise other_errors  # noqa
-            finally:
-                await context.terminated.set()
-                server_task_group.cancel_scope.deadline = anyio.current_time() + config.graceful_timeout
-
-        await lifespan.wait_for_shutdown()
-        lifespan_task_group.cancel_scope.cancel()
+                    listeners.append(anyio._core._eventloop.get_async_backend().create_tcp_listener(sock))  # noqa
+                    bind = repr_socket_addr(sock.family, sock.getsockname())
+                    binds.append(f'http://{bind}')
+                    log.info('Running on http://%s (CTRL + C to quit)', bind)
+
+                task_status.started(binds)
+                try:
+                    async with anyio.create_task_group() as task_group:
+                        if shutdown_trigger is not None:
+                            task_group.start_soon(raise_shutdown, shutdown_trigger)
+                        task_group.start_soon(raise_shutdown, context.terminate.wait)
+
+                        task_group.start_soon(
+                            functools.partial(
+                                serve_listeners,
+                                functools.partial(self._server_factory, app, context),
+                                listeners,
+                                handler_task_group=server_task_group,
+                            ),
+                        )
+
+                        await anyio.sleep_forever()
+                except BaseExceptionGroup as error:
+                    _, other_errors = error.split((ShutdownError, KeyboardInterrupt))  # noqa
+                    if other_errors is not None:
+                        raise other_errors  # noqa
+                finally:
+                    await context.terminated.set()
+                    server_task_group.cancel_scope.deadline = anyio.current_time() + config.graceful_timeout
+
+            await lifespan.wait_for_shutdown()
+            lifespan_task_group.cancel_scope.cancel()

omserv/server/multiprocess.py

@@ -12,11 +12,11 @@ import typing as ta
 import anyio
 
 from .config import Config
+from .default import serve
 from .sockets import Sockets
 from .sockets import create_sockets
 from .types import AsgiFramework
 from .types import wrap_app
-from .workers import serve
 
 
 async def check_multiprocess_shutdown_event(

omserv/server/{tcpserver.py → server.py}

@@ -6,6 +6,7 @@ import typing as ta
 import anyio.abc
 
 from omlish import check
+from omlish import lang
 
 from .config import Config
 from .events import Closed
@@ -25,13 +26,18 @@ log = logging.getLogger(__name__)
 MAX_RECV = 2 ** 16
 
 
-class TcpServer:
+class ServerFactory(lang.Func3[AppWrapper, WorkerContext, anyio.abc.SocketStream, 'Server']):
+    pass
+
+
+class Server:
     def __init__(
             self,
             app: AppWrapper,
-            config: Config,
             context: WorkerContext,
             stream: anyio.abc.SocketStream,
+            *,
+            config: Config,
     ) -> None:
         super().__init__()
 

{omserv-0.0.0.dev177.dist-info → omserv-0.0.0.dev179.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omserv
-Version: 0.0.0.dev177
+Version: 0.0.0.dev179
 Summary: omserv
 Author: wrmsr
 License: BSD-3-Clause
@@ -12,7 +12,7 @@ Classifier: Operating System :: OS Independent
 Classifier: Operating System :: POSIX
 Requires-Python: >=3.12
 License-File: LICENSE
-Requires-Dist: omlish==0.0.0.dev177
+Requires-Dist: omlish==0.0.0.dev179
 Provides-Extra: all
 Requires-Dist: h11~=0.14; extra == "all"
 Requires-Dist: h2~=4.1; extra == "all"

{omserv-0.0.0.dev177.dist-info → omserv-0.0.0.dev179.dist-info}/RECORD

@@ -21,20 +21,22 @@ omserv/node/models.py,sha256=EOaq-aW1rbGOzHTmTULhbVwV5j_burL07qh2QO9smdM,1273
 omserv/node/registry.py,sha256=y67VMowll9IuLiTVgauAcbP8-evFJNqpwocwwt7kZL4,3478
 omserv/node/sql.py,sha256=vy7RP50JiH3jQHMVa7Hxk0pFJK3QcbGeTvyNppB1W4I,2826
 omserv/server/LICENSE,sha256=VKPNmbyrS9wcwcx20hBlVtLP01brb2dByHrWHeNLPag,1050
-omserv/server/__init__.py,sha256=7c9pUvoC-3lByJeGoQfAcp78JCZ9oA5wP0Bzpm87D7Y,372
+omserv/server/__init__.py,sha256=jrFdrxfnQwjK5DW5y9w0alw6IKToGn2IQyZDhsuGHv4,372
 omserv/server/config.py,sha256=oGWL1kuk45bJ6sVr8n3ow5Q-1nz9EqByjoykU2iOHIY,1189
 omserv/server/debug.py,sha256=N7RI0Jj-ttmys3DJD0RREmGG5XZpTCp6y9Yu0x98Agg,299
+omserv/server/default.py,sha256=hmfy--Q35QFMU8oTf4uHwVM2qBFG8mQDR9Wik2f1yZk,970
 omserv/server/events.py,sha256=VMr_rArsVjJYnyH9SqLWtOLUg18vSu1O0ep9gNBGR_c,1369
 omserv/server/headers.py,sha256=3H-NxMMQg5WuF5wF4AWFUEqkToh4NqNqHouavzbOQok,1188
+omserv/server/inject.py,sha256=mD3MIDd44_nr1mIiDfjzgJ4DCBeD3JILN6STTDNReGc,454
 omserv/server/lifespans.py,sha256=kRVxDQM18jCBzRUpafyb69q_bGSCyxxjAtrkxjqcZdE,4607
-omserv/server/multiprocess.py,sha256=jKmQzj_Nrwxw2mM-Tf3XVf6vPYMbfyCkPOUWVe028Q8,4253
+omserv/server/listener.py,sha256=t_wUlS6pXnRn-aEXV8D8jhWAeIlQ3a3Sx9i_0BOKV70,7033
+omserv/server/multiprocess.py,sha256=qvNFQEMpTXXR5ACdja4JVC0HFgdvb8CL94iKLDYvZ48,4253
+omserv/server/server.py,sha256=hgUTfZAUwF7V-HdsdV98KbYmE1IRKmEO3B1LlKTiPWc,5223
 omserv/server/sockets.py,sha256=lwqNP7URlp605ibsjHzp0pc-lyjcyTu-hD-uyojLUYk,3389
 omserv/server/ssl.py,sha256=gmB5ecM8Mck-YtGYF8pb2dwFdjABVGzERFCDzM9lBck,1483
 omserv/server/taskspawner.py,sha256=ljzF26UPtnp7GLAY_BvjzuwCoCO9aL7TKLwRNTmUy1M,3008
-omserv/server/tcpserver.py,sha256=akC-2WOhmoIiJBH0Ti0m1uK_sOTBYGie0CoRkEcUmkA,5082
 omserv/server/types.py,sha256=XXY5py8RYlEeD4FZrWNqSyX7DD-ffSlcG-T2s9BY9eI,2017
 omserv/server/workercontext.py,sha256=4rcLuGsyiU7URO7T_eHylOBPPNUS9C23QfEUVyJUtIY,1200
-omserv/server/workers.py,sha256=rdV8qEzWKMZ6HnN9nUoGS58U9LRsrsqOcAd_7yl73Y0,6586
 omserv/server/protocols/__init__.py,sha256=Ryu2PDZ1TUI6F2l-HBEYgyzZ7wHqE6VmjqnS0tIvmQI,47
 omserv/server/protocols/h11.py,sha256=_q_paD-ff0AWJEPaNK-6MUsQVtYRiALnWGwFyM3D0KU,11976
 omserv/server/protocols/h2.py,sha256=bC-qmRQqgLASL1DtF3UX1ozximHH4xtae1d_vN2PffY,15345
@@ -46,9 +48,9 @@ omserv/server/streams/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3h
 omserv/server/streams/httpstream.py,sha256=0DeiAPLGbEGNa0fHTs8lUpi_CFZs4M5_QB-TiS8mobQ,8015
 omserv/server/streams/utils.py,sha256=aMOrqWIg_Hht5W4kLg3y7oR5AEkVvMrZhyjzo6U5owE,1527
 omserv/server/streams/wsstream.py,sha256=3Vyzox7dCE1tDSXjb6xBubWo41ZF9d38Hrsrlj6h1J8,15482
-omserv-0.0.0.dev177.dist-info/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
-omserv-0.0.0.dev177.dist-info/METADATA,sha256=8kgd6arWwSMcvrCNTckJjqUob39ux5K3iGn3axvRmDA,983
-omserv-0.0.0.dev177.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-omserv-0.0.0.dev177.dist-info/entry_points.txt,sha256=ivSrdA_ahEbI-eVMu-XZS-z4VrnQISvpecIkOqC9zFM,35
-omserv-0.0.0.dev177.dist-info/top_level.txt,sha256=HXehpnxeKscKNULzKNzZ27oNawBrsh1PaNAirbX-XNA,7
-omserv-0.0.0.dev177.dist-info/RECORD,,
+omserv-0.0.0.dev179.dist-info/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
+omserv-0.0.0.dev179.dist-info/METADATA,sha256=fd2btbHerGBDD_T9ETQdAMBb8XAoY3eo1tVDqQHIrbQ,983
+omserv-0.0.0.dev179.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+omserv-0.0.0.dev179.dist-info/entry_points.txt,sha256=ivSrdA_ahEbI-eVMu-XZS-z4VrnQISvpecIkOqC9zFM,35
+omserv-0.0.0.dev179.dist-info/top_level.txt,sha256=HXehpnxeKscKNULzKNzZ27oNawBrsh1PaNAirbX-XNA,7
+omserv-0.0.0.dev179.dist-info/RECORD,,