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

add EventScalesetResizeScheduled (#1047)

Browse Source
This commit is contained in:
bmc-msft
2021-07-07 14:15:26 -04:00
committed by GitHub
parent 7b2679a1ce
commit 52f83b5b26
7 changed files with 121 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/api-service/__app__/onefuzzlib/autoscale.py
View File

@ -35,12 +35,11 @@ def scale_up(pool: Pool, scalesets: List[Scaleset], nodes_needed: int) -> None:
if scaleset.size < max_size:
current_size = scaleset.size
if nodes_needed <= max_size - current_size:
scaleset.size = current_size + nodes_needed
scaleset.set_size(current_size + nodes_needed)
nodes_needed = 0
else:
scaleset.size = max_size
scaleset.set_size(max_size)
nodes_needed = nodes_needed - (max_size - current_size)
scaleset.set_state(ScalesetState.resize)
else:
continue
@ -112,7 +111,7 @@ def scale_down(scalesets: List[Scaleset], nodes_to_remove: int) -> None:
continue
# Resize of VMSS needed
scaleset.size = scaleset.size - max_nodes_remove
scaleset.set_size(scaleset.size - max_nodes_remove)
nodes_to_remove = nodes_to_remove - max_nodes_remove
scaleset.set_state(ScalesetState.resize)