ExternalVendorCode/onefuzz
1
0
Fork 0
mirror of https://github.com/microsoft/onefuzz.git synced 2025-06-17 04:18: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

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

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