ets-labs/python-dependency-injector
1
1
Fork 0
mirror of https://github.com/ets-labs/python-dependency-injector.git synced 2025-05-22 21:46:17 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add PoC

Browse Source
This commit is contained in:
Roman Mogylatov 2021-07-16 20:03:56 -04:00
parent 54de3a9d2c
commit f5e76aef28
2 changed files with 4898 additions and 2220 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7067
src/dependency_injector/containers.c
View File

File diff suppressed because it is too large Load Diff

51
src/dependency_injector/containers.pyx
View File

@ -290,16 +290,53 @@ class DynamicContainer(Container):
def shutdown_resources(self):
"""Shutdown all container resources."""
futures = []
def _is_safe_to_shutdown(resource):
for related in resource.related:
if isinstance(related, providers.Resource) and related.initialized:
return False
return True
for provider in self.traverse(types=[providers.Resource]):
shutdown = provider.shutdown()
def _resources_without_dependencies(resources):
for resource in resources:
if _is_safe_to_shutdown(resource):
yield resource
if __is_future_or_coroutine(shutdown):
futures.append(shutdown)
async def _async_shutdown(initialized_resources):
initialized_resources = set(initialized_resources)
while initialized_resources:
resources_to_shutdown = set(_resources_without_dependencies(initialized_resources))
if not resources_to_shutdown:
raise RuntimeError('Unable to resolve resources shutdown order')
await asyncio.gather(
*(
resource.shutdown()
for resource in resources_to_shutdown
),
)
initialized_resources -= resources_to_shutdown
if futures:
return asyncio.gather(*futures)
def _sync_shutdown(initialized_resources):
initialized_resources = set(initialized_resources)
while initialized_resources:
resources_to_shutdown = set(_resources_without_dependencies(initialized_resources))
if not resources_to_shutdown:
raise RuntimeError('Unable to resolve resources shutdown order')
for resource in resources_to_shutdown:
resource.shutdown()
initialized_resources -= resources_to_shutdown
initialized_resources = list(
filter(
lambda resource: resource.initialized,
self.traverse(types=[providers.Resource])
),
)
is_async = any((resource.is_async_mode_enabled() for resource in initialized_resources))
if is_async:
return _async_shutdown(initialized_resources)
else:
return _sync_shutdown(initialized_resources)
def apply_container_providers_overridings(self):
"""Apply container providers' overridings."""