mirror of
https://github.com/django-q2/django-q2.git
synced 2026-10-07 13:08:11 +08:00
Moves custom broker class to top
This commit is contained in:
@@ -158,28 +158,32 @@ def get_broker(list_key=Conf.PREFIX):
|
|||||||
:type list_key: str
|
:type list_key: str
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
# disque
|
# custom
|
||||||
if Conf.DISQUE_NODES:
|
if Conf.BROKER:
|
||||||
from brokers import disque
|
|
||||||
return disque.Disque(list_key=list_key)
|
|
||||||
elif Conf.IRON_MQ:
|
|
||||||
from brokers import ironmq
|
|
||||||
return ironmq.IronMQBroker(list_key=list_key)
|
|
||||||
elif Conf.SQS:
|
|
||||||
from brokers import aws_sqs
|
|
||||||
return aws_sqs.Sqs(list_key=list_key)
|
|
||||||
elif Conf.ORM:
|
|
||||||
from brokers import orm
|
|
||||||
return orm.ORM(list_key=list_key)
|
|
||||||
elif Conf.MONGO:
|
|
||||||
from brokers import mongo
|
|
||||||
return mongo.Mongo(list_key=list_key)
|
|
||||||
elif Conf.BROKER:
|
|
||||||
module, func = Conf.BROKER.rsplit('.', 1)
|
module, func = Conf.BROKER.rsplit('.', 1)
|
||||||
m = importlib.import_module(module)
|
m = importlib.import_module(module)
|
||||||
broker = getattr(m, func)
|
broker = getattr(m, func)
|
||||||
return broker(list_key=list)
|
return broker(list_key=list)
|
||||||
|
# disque
|
||||||
|
elif Conf.DISQUE_NODES:
|
||||||
|
from brokers import disque
|
||||||
|
return disque.Disque(list_key=list_key)
|
||||||
|
# Iron MQ
|
||||||
|
elif Conf.IRON_MQ:
|
||||||
|
from brokers import ironmq
|
||||||
|
return ironmq.IronMQBroker(list_key=list_key)
|
||||||
|
# SQS
|
||||||
|
elif Conf.SQS:
|
||||||
|
from brokers import aws_sqs
|
||||||
|
return aws_sqs.Sqs(list_key=list_key)
|
||||||
|
# ORM
|
||||||
|
elif Conf.ORM:
|
||||||
|
from brokers import orm
|
||||||
|
return orm.ORM(list_key=list_key)
|
||||||
|
# Mongo
|
||||||
|
elif Conf.MONGO:
|
||||||
|
from brokers import mongo
|
||||||
|
return mongo.Mongo(list_key=list_key)
|
||||||
# default to redis
|
# default to redis
|
||||||
else:
|
else:
|
||||||
from brokers import redis_broker
|
from brokers import redis_broker
|
||||||
|
|||||||
Reference in New Issue
Block a user