x/compass-core
Code Issues Proposed changes

Update machine and switch db api to send task to specific queues.

Browse Source 
Also update celeryconfig to subscribe to the default queue.

Change-Id: I82d2fd923863e7f226042c078e4286f86acb3d07
This commit is contained in:
Xicheng Chang 2016-06-29 13:44:31 -04:00
parent 70425d96c2
commit 64daaeb3a7
3 changed files with 19 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
compass/db/api/machine.py
View File

@ -228,7 +228,10 @@ def poweron_machine(
)
celery_client.celery.send_task(
'compass.tasks.poweron_machine',
(machine_id,)
(machine_id,),
queue=user.email,
exchange=user.email,
routing_key=user.email
)
return {
'status': 'poweron %s action sent' % machine.mac,
@ -255,7 +258,10 @@ def poweroff_machine(
)
celery_client.celery.send_task(
'compass.tasks.poweroff_machine',
(machine_id,)
(machine_id,),
queue=user.email,
exchange=user.email,
routing_key=user.email
)
return {
'status': 'poweroff %s action sent' % machine.mac,
@ -282,7 +288,10 @@ def reset_machine(
)
celery_client.celery.send_task(
'compass.tasks.reset_machine',
(machine_id,)
(machine_id,),
queue=user.email,
exchange=user.email,
routing_key=user.email
)
return {
'status': 'reset %s action sent' % machine.mac,

5
compass/db/api/switch.py
View File

@ -838,7 +838,10 @@ def poll_switch(switch_id, user=None, session=None, **kwargs):
switch = _get_switch(switch_id, session=session)
celery_client.celery.send_task(
'compass.tasks.pollswitch',
(user.email, switch.ip, switch.credentials)
(user.email, switch.ip, switch.credentials),
queue=user.email,
exchange=user.email,
routing_key=user.email
)
return {
'status': 'action %s sent' % kwargs,

3
conf/celeryconfig
View File

@ -7,3 +7,6 @@ BROKER_URL = "amqp://guest:guest@localhost:5672//"
CELERY_IMPORTS=("compass.tasks.tasks",)
CELERY_ACCEPT_CONTENT = ['pickle', 'json', 'msgpack', 'yaml']
C_FORCE_ROOT = 1
CELERY_DEFAULT_QUEUE = 'admin@huawei.com'
CELERY_DEFAULT_EXCHANGE = 'admin@huawei.com'
CELERY_DEFAULT_ROUTING_KEY = 'admin@huawei.com'