laipeng/reflex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

capture exceptions thrown by any process in run_concurrently_context (#2296)

Browse Source
This commit is contained in:
jackie-pc 2023-12-14 14:41:45 -08:00 committed by GitHub
parent 9da65b9a9a
commit 92bd3da676
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
reflex/utils/processes.py
View File

@ -171,7 +171,9 @@ def run_concurrently_context(
fns = [fn if isinstance(fn, tuple) else (fn,) for fn in fns] # type: ignore
# Run the functions concurrently.
with futures.ThreadPoolExecutor(max_workers=len(fns)) as executor:
executor = None
try:
executor = futures.ThreadPoolExecutor(max_workers=len(fns))
# Submit the tasks.
tasks = [executor.submit(*fn) for fn in fns] # type: ignore
@ -180,7 +182,12 @@ def run_concurrently_context(
# Get the results in the order completed to check any exceptions.
for task in futures.as_completed(tasks):
# if task throws something, we let it bubble up immediately
task.result()
finally:
# Shutdown the executor
if executor:
executor.shutdown(wait=False)
def run_concurrently(*fns: Union[Callable, Tuple]) -> None: