fork/django-q2
1
0
Fork 0
mirror of https://github.com/django-q2/django-q2.git synced 2026-10-04 17:58:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Move timeout function from ORM broker into class to allow easy customization (#274)

Browse Source 
* move timeout function to class to allow customization

* format

* fix format
This commit is contained in:
Stan Triepels
2025-04-23 15:44:37 +02:00
committed by GitHub
parent 0090a6f411
commit cbe51f995c
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
django_q/brokers/orm.py
+4 -5
View File
@@ -10,10 +10,6 @@ from django_q.conf import Conf, logger
from django_q.models import OrmQ from django_q.models import OrmQ
def _timeout():
return timezone.now() + timedelta(seconds=Conf.RETRY)
class ORM(Broker): class ORM(Broker):
@staticmethod @staticmethod
def get_connection(list_key: str = None): def get_connection(list_key: str = None):
@@ -28,6 +24,9 @@ class ORM(Broker):
logger.debug("Broker in an atomic transaction") logger.debug("Broker in an atomic transaction")
return OrmQ.objects.using(Conf.ORM) return OrmQ.objects.using(Conf.ORM)
def timeout(self, task):
return timezone.now() + timedelta(seconds=Conf.RETRY)
def queue_size(self) -> int: def queue_size(self) -> int:
return ( return (
self.get_connection() self.get_connection()
@@ -75,7 +74,7 @@ class ORM(Broker):
if ( if (
self.get_connection() self.get_connection()
.filter(id=task.id, lock=task.lock) .filter(id=task.id, lock=task.lock)
.update(lock=_timeout()) .update(lock=self.timeout(task))
): ):
task_list.append((task.pk, task.payload)) task_list.append((task.pk, task.payload))
# else don't process, as another cluster has been faster than us on # else don't process, as another cluster has been faster than us on