openstack/deb-python-oslo.messaging
Code Issues Proposed changes

Merge "Ensures listener queues exist in fake driver"

Browse Source
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
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._exchange_manager = exchange_manager
self._targets = targets 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): def poll(self, timeout=None):
if timeout is not None: if timeout is not None:
deadline = time.time() + timeout deadline = time.time() + timeout
@@ -77,6 +83,13 @@ class FakeExchange(object):
self._topic_queues = {} self._topic_queues = {}
self._server_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): def _get_topic_queue(self, topic):
return self._topic_queues.setdefault(topic, []) return self._topic_queues.setdefault(topic, [])