fork/django-q2
1
0
Fork 0
mirror of https://github.com/django-q2/django-q2.git synced 2026-09-28 14:48:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

allow atomic on external db (#14)

Browse Source 
Co-authored-by: Rajeev KL <rajeev@labmoneta.io>
This commit is contained in:
Stan Triepels
2022-10-15 23:24:30 +02:00
committed by GitHub
co-authored by Rajeev KL
parent 00fe611b86
commit ce98aaa15c
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
django_q/cluster.py
+4 -2
View File
@@ -496,7 +496,9 @@ def save_task(task, broker: Broker):
if Conf.SAVE_LIMIT_PER == "func": if Conf.SAVE_LIMIT_PER == "func":
value = get_func_repr(value) value = get_func_repr(value)
filters[Conf.SAVE_LIMIT_PER] = value filters[Conf.SAVE_LIMIT_PER] = value
with db.transaction.atomic():
database_to_use = {"using": Conf.ORM if Conf.ORM else Schedule.objects.db} if not Conf.HAS_REPLICA else {}
with db.transaction.atomic(**database_to_use):
last = Success.objects.filter(**filters).select_for_update().last() last = Success.objects.filter(**filters).select_for_update().last()
if task["success"] and 0 < Conf.SAVE_LIMIT <= Success.objects.filter(**filters).count(): if task["success"] and 0 < Conf.SAVE_LIMIT <= Success.objects.filter(**filters).count():
last.delete() last.delete()
@@ -599,7 +601,7 @@ def scheduler(broker: Broker = None):
broker = get_broker() broker = get_broker()
close_old_django_connections() close_old_django_connections()
try: try:
database_to_use = {"using": Conf.ORM} if not Conf.HAS_REPLICA else {} database_to_use = {"using": Conf.ORM if Conf.ORM else Schedule.objects.db} if not Conf.HAS_REPLICA else {}
with db.transaction.atomic(**database_to_use): with db.transaction.atomic(**database_to_use):
for s in ( for s in (
Schedule.objects.select_for_update() Schedule.objects.select_for_update()