Correctly capture Queue.Empty exception

This commit is contained in:
Erik Johnston 2015-05-21 15:37:43 +01:00
parent 4fb858d90a
commit 115ef3ddac

View file

@ -19,7 +19,7 @@ from ._base import SQLBaseStore, cached
from syutil.base64util import encode_base64 from syutil.base64util import encode_base64
import logging import logging
from Queue import PriorityQueue from Queue import PriorityQueue, Empty
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -398,7 +398,10 @@ class EventFederationStore(SQLBaseStore):
queue.put(row) queue.put(row)
while not queue.empty() and len(event_results) < limit: while not queue.empty() and len(event_results) < limit:
try:
_, event_id = queue.get_nowait() _, event_id = queue.get_nowait()
except Empty:
break
event_results.add(event_id) event_results.add(event_id)