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

adding node shutdown (#252)

Browse Source
This commit is contained in:
Anshuman Goel
2020-11-03 08:39:51 -08:00
committed by GitHub
parent 0e04309e33
commit 4ef489b397
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -898,6 +898,9 @@ class Scaleset(BASE_SCALESET, ORMMixin):
def shutdown(self) -> None: def shutdown(self) -> None:
size = get_vmss_size(self.scaleset_id) size = get_vmss_size(self.scaleset_id)
logging.info("scaleset shutdown: %s (current size: %s)", self.scaleset_id, size) logging.info("scaleset shutdown: %s (current size: %s)", self.scaleset_id, size)
nodes = Node.search_states(scaleset_id=self.scaleset_id)
for node in nodes:
node.set_shutdown()
if size is None or size == 0: if size is None or size == 0:
self.halt() self.halt()