ExternalVendorCode/onefuzz
1
0
Fork 0
mirror of https://github.com/microsoft/onefuzz.git synced 2025-06-16 03:48:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove use of batch in NodeMessages (#60)

Browse Source
This commit is contained in:
bmc-msft
2020-10-01 11:14:39 -04:00
committed by GitHub
parent ab09cb8ae9
commit eae0f42938
3 changed files with 8 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/api-service/__app__/onefuzzlib/pools.py
View File

@ -40,7 +40,6 @@ from .azure.creds import get_fuzz_storage
from .azure.image import get_os
from .azure.network import Network
from .azure.queue import create_queue, delete_queue, peek_queue, queue_object
from .azure.table import get_client
from .azure.vmss import (
UnableToUpdate,
create_vmss,
@ -231,21 +230,11 @@ class NodeMessage(ORMMixin):
)
return entries
@classmethod
def delete_messages(cls, agent_id: UUID, message_ids: List[str]) -> None:
client = get_client(table=cls.table_name())
batch = client.batch(table_name=cls.table_name())
for message_id in message_ids:
batch.delete_entity(agent_id, message_id)
client.commit_batch(cls.table_name(), batch)
@classmethod
def clear_messages(cls, agent_id: UUID) -> None:
messages = cls.get_messages(agent_id)
message_ids = [m.message_id for m in messages]
cls.delete_messages(agent_id, message_ids)
for message in messages:
message.delete()
class Pool(BASE_POOL, ORMMixin):