6b40f95033
Co-authored-by: Eve832 <eve.meelan@grafana.com>
Co-authored-by: Francisco Montes de Oca <nevermind89x@gmail.com>
Co-authored-by: Ildar Iskhakov <ildar.iskhakov@grafana.com>
Co-authored-by: Innokentii Konstantinov <innokenty.konstantinov@grafana.com>
Co-authored-by: Julia <ferril.darkdiver@gmail.com>
Co-authored-by: maskin25 <kengurek@gmail.com>
Co-authored-by: Matias Bordese <mbordese@gmail.com>
Co-authored-by: Matvey Kukuy <motakuk@gmail.com>
Co-authored-by: Michael Derynck <michael.derynck@grafana.com>
Co-authored-by: Richard Hartmann <richih@richih.org>
Co-authored-by: Robby Milo <robbymilo@fastmail.com>
Co-authored-by: Timur Olzhabayev <timur.olzhabayev@grafana.com>
Co-authored-by: Vadim Stepanov <vadimkerr@gmail.com>
Co-authored-by: Yulia Shanyrova <yulia.shanyrova@grafana.com>
30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
from django.db import transaction
|
|
|
|
from apps.base.models import FailedToInvokeCeleryTask
|
|
from common.custom_celery_tasks import shared_dedicated_queue_retry_task
|
|
from common.utils import batch_queryset
|
|
|
|
|
|
@shared_dedicated_queue_retry_task
|
|
def process_failed_to_invoke_celery_tasks():
|
|
task_pks = FailedToInvokeCeleryTask.objects.filter(is_sent=False).values_list("pk", flat=True)
|
|
|
|
batches = batch_queryset(task_pks)
|
|
for idx, batch in enumerate(batches):
|
|
countdown = idx * 60
|
|
process_failed_to_invoke_celery_tasks_batch.apply_async((list(batch),), countdown=countdown)
|
|
|
|
|
|
@shared_dedicated_queue_retry_task
|
|
def process_failed_to_invoke_celery_tasks_batch(task_pks):
|
|
sent_task_pks = []
|
|
with transaction.atomic():
|
|
for task in FailedToInvokeCeleryTask.objects.filter(pk__in=task_pks, is_sent=False).select_for_update():
|
|
try:
|
|
task.send()
|
|
except Exception:
|
|
continue
|
|
|
|
sent_task_pks.append(task.pk)
|
|
|
|
FailedToInvokeCeleryTask.objects.filter(pk__in=sent_task_pks).update(is_sent=True)
|