add try-finally for resources closing (#36)

nightblure · Dec 29, 2024 · bd119f9 · bd119f9
1 parent 5c5fdea
commit bd119f9
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 8 deletions.
diff --git a/src/injection/auto_inject.py b/src/injection/auto_inject.py
@@ -59,8 +59,10 @@ def wrapper(*args: P.args, **kwargs: P.kwargs) -> T:
             providers.append(provider)
             kwargs[param_name] = provider()
 
-        result = f(*args, **kwargs)
-        close_related_function_scope_resources_sync(providers)
+        try:
+            result = f(*args, **kwargs)
+        finally:
+            close_related_function_scope_resources_sync(providers)
         return result
 
     return wrapper
@@ -113,8 +115,10 @@ async def wrapper(*args: P.args, **kwargs: P.kwargs) -> T:
 
                 kwargs[param_name] = resolved_provide
 
-        result = await f(*args, **kwargs)
-        await close_related_function_scope_resources_async(providers)
+        try:
+            result = await f(*args, **kwargs)
+        finally:
+            await close_related_function_scope_resources_async(providers)
         return result
 
     return wrapper

diff --git a/src/injection/inject.py b/src/injection/inject.py
@@ -126,8 +126,10 @@ async def wrapper(*args: P.args, **kwargs: P.kwargs) -> T:
 
             kwargs[param_name] = resolved_provide
 
-        result = await f(*args, **kwargs)
-        await close_related_function_scope_resources_async(providers)
+        try:
+            result = await f(*args, **kwargs)
+        finally:
+            await close_related_function_scope_resources_async(providers)
         return result
 
     return wrapper
@@ -154,8 +156,10 @@ def wrapper(*args: P.args, **kwargs: P.kwargs) -> T:
             providers.append(provider)
             kwargs[param_name] = provider()
 
-        result = f(*args, **kwargs)
-        close_related_function_scope_resources_sync(providers)
+        try:
+            result = f(*args, **kwargs)
+        finally:
+            close_related_function_scope_resources_sync(providers)
         return result
 
     return wrapper