diff --git a/uncloud/celery.py b/uncloud/celery.py index 7bcaaae..3408634 100644 --- a/uncloud/celery.py +++ b/uncloud/celery.py @@ -15,8 +15,3 @@ app.config_from_object('django.conf:settings', namespace='CELERY') # Load task modules from all registered Django app configs. app.autodiscover_tasks() - - -@app.task(bind=True) -def debug_task(self): - print(f'Request: {self.request!r}') diff --git a/uncloud/settings.py b/uncloud/settings.py index ea90c22..ae734dc 100644 --- a/uncloud/settings.py +++ b/uncloud/settings.py @@ -224,6 +224,13 @@ CELERY_TASK_ROUTES = { re.compile(r'.*.tasks.cdist.*'): { 'queue': 'cdist' } # cdist tasks go into cdist queue } +CELERY_BEAT_SCHEDULE = { + 'cleanup_tasks': { + 'task': 'uncloud.tasks.cleanup_tasks', + 'schedule': 10 + } +} + # CELERY_TASK_CREATE_MISSING_QUEUES = False # Overwrite settings with local settings, if existing diff --git a/uncloud/tasks.py b/uncloud/tasks.py new file mode 100644 index 0000000..8350354 --- /dev/null +++ b/uncloud/tasks.py @@ -0,0 +1,14 @@ +from celery import shared_task +from celery.result import AsyncResult + +from .models import UncloudTask + +@shared_task +def cleanup_tasks(): + print("Cleanup time") + for task in UncloudTask.objects.all(): + print(f"Pruning {task}...") + res = AsyncResult(id=str(task.task_id)) + if res.ready(): + print(res.get()) + task.delete() diff --git a/uncloud_net/tasks.py b/uncloud_net/tasks.py index 0dcf6f3..529c525 100644 --- a/uncloud_net/tasks.py +++ b/uncloud_net/tasks.py @@ -29,7 +29,7 @@ def configure_wireguard_server(wireguardvpnpool): log.info(f"Configuring VPN server {server} (async)") task_id = uuid.UUID(cdist_configure_wireguard_server.apply_async((config, server)).id) - UncloudTasks.objects.create(task_id=task_id) + UncloudTask.objects.create(task_id=task_id) @shared_task