diff --git a/docs/index.md b/docs/index.md index 4f05fd2..7d48192 100644 --- a/docs/index.md +++ b/docs/index.md @@ -60,6 +60,7 @@ class SimpleFactory: @dataclasses.dataclass(kw_only=True, slots=True) class DependentFactory: + simple_factory: SimpleFactory singleton: str ``` @@ -99,7 +100,7 @@ For now there are integrations for the following frameworks: Create a container and resolve dependencies in your code ```python -from modern_di import Container +from modern_di import Container, Scope ALL_GROUPS = [Dependencies] @@ -123,8 +124,8 @@ try: finally: # Close container when done # For async usage: - # await request_container.close_async() + await request_container.close_async() # For sync usage: - # request_container.close_sync() + request_container.close_sync() ``` diff --git a/packages/modern-di-faststream/modern_di_faststream/main.py b/packages/modern-di-faststream/modern_di_faststream/main.py index 0dee223..87f4216 100644 --- a/packages/modern-di-faststream/modern_di_faststream/main.py +++ b/packages/modern-di-faststream/modern_di_faststream/main.py @@ -80,6 +80,7 @@ def setup_di( container.providers_registry.add_providers(faststream_message=faststream_message) app.context.set_global("di_container", container) + app.after_shutdown(container.close_async) app.broker.add_middleware(_DIMiddlewareFactory(container)) return container