openstack/oslo.messaging
Code Issues Proposed changes

Merge "Use StopWatch timer when waiting for message"

Browse Source
This commit is contained in:
Zuul 2024-04-09 13:47:33 +00:00 committed by Gerrit Code Review
commit 41fc2a2d35
1 changed files with 12 additions and 6 deletions

18
oslo_messaging/_drivers/amqpdriver.py

@ -496,12 +496,18 @@ class ReplyWaiters(object):
self._wrn_threshold = 10
def get(self, msg_id, timeout):
try:
return self._queues[msg_id].get(block=True, timeout=timeout)
except queue.Empty:
raise oslo_messaging.MessagingTimeout(
'Timed out waiting for a reply '
'to message ID %s' % msg_id)
watch = timeutils.StopWatch(duration=timeout)
watch.start()
while not watch.expired():
try:
# NOTE(amorin) we can't use block=True
# See lp-2035113
return self._queues[msg_id].get(block=False)
except queue.Empty:
time.sleep(0.5)
raise oslo_messaging.MessagingTimeout(
'Timed out waiting for a reply '
'to message ID %s' % msg_id)
def put(self, msg_id, message_data):
LOG.info('Received RPC response for msg %s', msg_id)