Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions diracx-tasks/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,9 @@ diracx-task-run = "diracx.tasks.task_run:main"
[project.entry-points."diracx.dbs.sql"]
TaskDB = "diracx.tasks.plumbing.persistence:TaskDB"

[project.entry-points."diracx.tasks.jobs"]
CleanSandboxStoreTask = "diracx.tasks.jobs:CleanSandboxStoreTask"

[build-system]
requires = ["hatchling", "hatch-vcs"]
build-backend = "hatchling.build"
Expand Down
31 changes: 31 additions & 0 deletions diracx-tasks/src/diracx/tasks/jobs/clean_sandbox_store.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
"""Task definition to clean Sandbox Store."""

from __future__ import annotations

import dataclasses
from typing import Annotated

from diracx.core.settings import SandboxStoreSettings
from diracx.db.sql import SandboxMetadataDB
from diracx.logic.jobs.sandboxes import clean_sandboxes
from diracx.tasks.plumbing.base_task import PeriodicBaseTask
from diracx.tasks.plumbing.enums import Priority, Size
from diracx.tasks.plumbing.retry_policies import ExponentialBackoff
from diracx.tasks.plumbing.schedules import CronSchedule

from .depends import NoTransaction


@dataclasses.dataclass
class CleanSandboxStoreTask(PeriodicBaseTask):
priority = Priority.BACKGROUND
size = Size.MEDIUM
retry_policy = ExponentialBackoff(base_delay_seconds=300, max_retries=3)
default_schedule = CronSchedule("0 6 * * *")

async def execute(
self,
sandbox_metadata_db: Annotated[SandboxMetadataDB, NoTransaction()],
settings: SandboxStoreSettings,
) -> int:
return await clean_sandboxes(sandbox_metadata_db, settings)
52 changes: 26 additions & 26 deletions pixi.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Loading