Skip to content

Commit 9d02f48

Browse files
committed
♻️ Refactor periodic task setup to use periodic_task_lifespan utility
1 parent a9d6b6a commit 9d02f48

File tree

5 files changed

+12
-10
lines changed

5 files changed

+12
-10
lines changed

services/web/server/src/simcore_service_webserver/garbage_collector/_tasks_api_keys.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
from simcore_service_webserver.redis import get_redis_lock_manager_client_sdk
1414

1515
from ..api_keys import api_keys_service
16-
from ._tasks_utils import CleanupContextFunc, setup_periodic_task
16+
from ._tasks_utils import CleanupContextFunc, periodic_task_lifespan
1717

1818
_logger = logging.getLogger(__name__)
1919

@@ -45,7 +45,9 @@ async def _prune_expired_api_keys_periodically() -> None:
4545
with log_context(_logger, logging.INFO, "Pruning expired API keys"):
4646
await _prune_expired_api_keys(app)
4747

48-
async for _ in setup_periodic_task(app, _prune_expired_api_keys_periodically):
48+
async for _ in periodic_task_lifespan(
49+
app, _prune_expired_api_keys_periodically
50+
):
4951
yield
5052

5153
return _cleanup_ctx_fun

services/web/server/src/simcore_service_webserver/garbage_collector/_tasks_core.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
from simcore_service_webserver.redis import get_redis_lock_manager_client_sdk
1515

1616
from ._core import collect_garbage
17-
from ._tasks_utils import CleanupContextFunc, setup_periodic_task
17+
from ._tasks_utils import CleanupContextFunc, periodic_task_lifespan
1818
from .settings import GarbageCollectorSettings, get_plugin_settings
1919

2020
_logger = logging.getLogger(__name__)
@@ -38,7 +38,7 @@ async def _collect_garbage_periodically() -> None:
3838
with log_context(_logger, logging.INFO, "Garbage collect cycle"):
3939
await collect_garbage(app)
4040

41-
async for _ in setup_periodic_task(
41+
async for _ in periodic_task_lifespan(
4242
app, _collect_garbage_periodically, task_name=_GC_TASK_NAME
4343
):
4444
yield

services/web/server/src/simcore_service_webserver/garbage_collector/_tasks_trash.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
from simcore_service_webserver.redis import get_redis_lock_manager_client_sdk
1414

1515
from ..trash import trash_service
16-
from ._tasks_utils import CleanupContextFunc, setup_periodic_task
16+
from ._tasks_utils import CleanupContextFunc, periodic_task_lifespan
1717

1818
_logger = logging.getLogger(__name__)
1919

@@ -33,7 +33,7 @@ async def _prune_trash_periodically() -> None:
3333
with log_context(_logger, logging.INFO, "Deleting expired trashed items"):
3434
await trash_service.safe_delete_expired_trash_as_admin(app)
3535

36-
async for _ in setup_periodic_task(app, _prune_trash_periodically):
36+
async for _ in periodic_task_lifespan(app, _prune_trash_periodically):
3737
yield
3838

3939
return _cleanup_ctx_fun

services/web/server/src/simcore_service_webserver/garbage_collector/_tasks_users.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
from ..login import login_service
1717
from ..security import security_service
1818
from ..users.api import update_expired_users
19-
from ._tasks_utils import CleanupContextFunc, setup_periodic_task
19+
from ._tasks_utils import CleanupContextFunc, periodic_task_lifespan
2020

2121
_logger = logging.getLogger(__name__)
2222

@@ -82,7 +82,7 @@ async def _update_expired_users_periodically() -> None:
8282
with log_context(_logger, logging.INFO, "Updating expired users"):
8383
await _update_expired_users(app)
8484

85-
async for _ in setup_periodic_task(app, _update_expired_users_periodically):
85+
async for _ in periodic_task_lifespan(app, _update_expired_users_periodically):
8686
yield
8787

8888
return _cleanup_ctx_fun

services/web/server/src/simcore_service_webserver/garbage_collector/_tasks_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from collections.abc import AsyncIterator, Callable, Coroutine
77

88
from aiohttp import web
9-
from servicelib.async_utils import cancel_wait_task
9+
from common_library.async_tools import cancel_wait_task
1010

1111
CleanupContextFunc = Callable[[web.Application], AsyncIterator[None]]
1212

@@ -19,7 +19,7 @@ def create_task_name(coro: Callable) -> str:
1919
return f"{coro.__module__}.{coro.__name__}"
2020

2121

22-
async def setup_periodic_task(
22+
async def periodic_task_lifespan(
2323
app: web.Application,
2424
periodic_task_coro: Callable[[], Coroutine[None, None, None]],
2525
*,

0 commit comments

Comments
 (0)