openstack
/
oslo.messaging
Code Issues Proposed changes

Merge "Ensures listener queues exist in fake driver"

This commit is contained in:
Jenkins 2014-06-23 16:04:24 +00:00 committed by Gerrit Code Review
parent 889bf73021 e582da68f4
commit 1a0a3dd4b9
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
oslo/messaging/_drivers/impl_fake.py
View File

@ -45,6 +45,12 @@ class FakeListener(base.Listener):
self._exchange_manager = exchange_manager
self._targets = targets
# NOTE(sileht): Ensure that all needed queues exists even the listener
# have not been polled yet
for target in self._targets:
exchange = self._exchange_manager.get_exchange(target.exchange)
exchange.ensure_queue(target)
def poll(self, timeout=None):
if timeout is not None:
deadline = time.time() + timeout
@ -77,6 +83,13 @@ class FakeExchange(object):
self._topic_queues = {}
self._server_queues = {}
def ensure_queue(self, target):
with self._queues_lock:
if target.server:
self._get_server_queue(target.topic, target.server)
else:
self._get_topic_queue(target.topic)
def _get_topic_queue(self, topic):
return self._topic_queues.setdefault(topic, [])