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

Check for nodes

Browse Source
This commit is contained in:
Ilan Steemers
2020-06-28 20:23:56 +02:00
parent 5bf2cd49eb
commit a57f7e3f89
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
django_q/brokers/disque.py
View File

@@ -1,8 +1,11 @@
import random
# External
import redis
from redis import Redis
# Django
from django.utils.translation import gettext_lazy as _
from django_q.brokers import Broker
from django_q.conf import Conf
@@ -52,6 +55,8 @@ class Disque(Broker):
@staticmethod
def get_connection(list_key: str = Conf.PREFIX) -> Redis:
if not Conf.DISQUE_NODES:
raise redis.exceptions.ConnectionError(_("No Disque nodes configured"))
# randomize nodes
random.shuffle(Conf.DISQUE_NODES)
# find one that works
@@ -67,4 +72,6 @@ class Disque(Broker):
return redis_client
except redis.exceptions.ConnectionError:
continue
raise redis.exceptions.ConnectionError("Could not connect to any Disque nodes")
raise redis.exceptions.ConnectionError(
_("Could not connect to any Disque nodes")
)