From 7ca6aa2eb742925927e5233d224ca742f8f838fe Mon Sep 17 00:00:00 2001 From: Stepan Burlakov Date: Wed, 29 Dec 2021 15:43:29 +0200 Subject: [PATCH] set global loop if none --- src/firebolt/common/util.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/src/firebolt/common/util.py b/src/firebolt/common/util.py index dfeb7313b40..c9507f0d7f8 100644 --- a/src/firebolt/common/util.py +++ b/src/firebolt/common/util.py @@ -1,4 +1,4 @@ -from asyncio import get_event_loop, new_event_loop +from asyncio import get_event_loop, new_event_loop, set_event_loop from functools import lru_cache, wraps from typing import TYPE_CHECKING, Any, Callable, Type, TypeVar @@ -40,15 +40,12 @@ def fix_url_schema(url: str) -> str: def async_to_sync(f: Callable) -> Callable: @wraps(f) def sync(*args: Any, **kwargs: Any) -> Any: - close = False try: loop = get_event_loop() except RuntimeError: loop = new_event_loop() - close = True + set_event_loop(loop) res = loop.run_until_complete(f(*args, **kwargs)) - if close: - loop.close() return res return sync