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

handle scaleset resize exceptions (#967)

Browse Source
This commit is contained in:
bmc-msft
2021-06-08 09:30:36 -04:00
committed by GitHub
parent 2d377d1423
commit ed289c9a3c
1 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/api-service/__app__/onefuzzlib/azure/vmss.py
View File

@ -81,9 +81,17 @@ def resize_vmss(name: UUID, capacity: int) -> None:
resource_group = get_base_resource_group() resource_group = get_base_resource_group()
logging.info("updating VM count - name: %s vm_count: %d", name, capacity) logging.info("updating VM count - name: %s vm_count: %d", name, capacity)
compute_client = get_compute_client() compute_client = get_compute_client()
compute_client.virtual_machine_scale_sets.begin_update( try:
resource_group, str(name), {"sku": {"capacity": capacity}} compute_client.virtual_machine_scale_sets.begin_update(
) resource_group, str(name), {"sku": {"capacity": capacity}}
)
except ResourceExistsError as err:
logging.error(
"unable to resize scaleset. name:%s vm_count:%d - err:%s",
name,
capacity,
err,
)
def get_vmss_size(name: UUID) -> Optional[int]: def get_vmss_size(name: UUID) -> Optional[int]: