ExternalVendorCode/onefuzz
1
0
Fork 0
mirror of https://github.com/microsoft/onefuzz.git synced 2025-06-17 12:28:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

handle queues vanishing during peek (#832)

Browse Source 
Handle queues getting deleted during peek_queue.  This can happen when
polling the pool for work while the pool is getting shut down.
This commit is contained in:
bmc-msft
2021-04-26 15:42:40 -04:00
committed by GitHub
parent f4b5c1ae73
commit 541e745199
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/api-service/__app__/onefuzzlib/azure/queue.py
View File

@ -158,14 +158,17 @@ def peek_queue(
if max_messages < MIN_PEEK_SIZE or max_messages > MAX_PEEK_SIZE:
raise ValueError("invalid max messages: %s" % max_messages)
queue = get_queue(name, storage_type)
if not queue:
return result
try:
queue = get_queue(name, storage_type)
if not queue:
return result
for message in queue.peek_messages(max_messages=max_messages):
decoded = base64.b64decode(message.content)
raw = json.loads(decoded)
result.append(object_type.parse_obj(raw))
for message in queue.peek_messages(max_messages=max_messages):
decoded = base64.b64decode(message.content)
raw = json.loads(decoded)
result.append(object_type.parse_obj(raw))
except ResourceNotFoundError:
return result
return result