feat: raw delete expired instead of Queryset.delete

django_celery_results/managers.py

@@ -8,7 +8,7 @@
 from django.conf import settings
 from django.db import connections, models, router, transaction
 
-from .utils import now
+from .utils import now, raw_delete
 
 W_ISOLATION_REP = """
 Polling results with transaction isolation level 'repeatable-read'
@@ -88,7 +88,7 @@ def get_all_expired(self, expires):
     def delete_expired(self, expires):
         """Delete all expired results."""
         with transaction.atomic():
-            self.get_all_expired(expires).delete()
+            raw_delete(queryset=self.get_all_expired(expires))
 
 
 class TaskResultManager(ResultManager):

django_celery_results/utils.py

@@ -15,3 +15,8 @@ def now():
         return now_localtime(timezone.now())
     else:
         return timezone.now()
+
+
+def raw_delete(queryset):
+    """Raw delete given queryset."""
+    return queryset._raw_delete(queryset.db)

t/unit/test_utils.py

@@ -0,0 +1,41 @@
+from celery import uuid
+from django.test import TransactionTestCase
+
+from django_celery_results.models import TaskResult
+from django_celery_results.utils import raw_delete
+
+_TEST_RECORDS_COUNT = 100
+
+
+class RawDeleteTest(TransactionTestCase):
+    """Unit test for `utils.raw_delete` func."""
+
+    def setUp(self):
+        # setup test results to be used against raw_delete
+        TaskResult.objects.bulk_create(
+            [TaskResult(task_id=uuid()) for _ in range(_TEST_RECORDS_COUNT)]
+        )
+        assert TaskResult.objects.count() == _TEST_RECORDS_COUNT
+
+    def _get_sample_results(self, count):
+        return TaskResult.objects.values_list("pk").order_by("?")[:count]
+
+    def test_all_rows_delete(self):
+        raw_delete(
+            queryset=TaskResult.objects.all()
+        )
+        assert TaskResult.objects.count() == 0
+
+    def test_correct_rows_delete(self):
+        # sample random 10 rows
+        sample_size = 10
+        sample_ids = self._get_sample_results(count=sample_size)
+        # update task_name to "test"
+        TaskResult.objects.filter(pk__in=sample_ids).update(task_name="test")
+        # "raw delete" results with name = "test"
+        raw_delete(
+            queryset=TaskResult.objects.filter(task_name="test")
+        )
+        assert TaskResult.objects.filter(task_name="test").count() == 0
+        results_remaining = _TEST_RECORDS_COUNT - sample_size
+        assert TaskResult.objects.count() == results_remaining