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

Adding auto scale via cli (#1717)

Browse Source 
* Initial implementation for adding auto scale via cli

* Remove unused argument

* Remove unused import

* I had a 👻extra line👻
This commit is contained in:
Teo Voinea
2022-03-29 09:50:07 -04:00
committed by GitHub
parent 5e31ba5b18
commit 424ffdb4b5
6 changed files with 142 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/cli/onefuzz/api.py
View File

@ -1357,6 +1357,11 @@ class Scaleset(Endpoint):
spot_instances: bool = False,
ephemeral_os_disks: bool = False,
tags: Optional[Dict[str, str]] = None,
min_instances: Optional[int] = 1,
scale_out_amount: Optional[int] = 1,
scale_out_cooldown: Optional[int] = 10,
scale_in_amount: Optional[int] = 1,
scale_in_cooldown: Optional[int] = 15,
) -> models.Scaleset:
self.logger.debug("create scaleset")
@ -1372,6 +1377,16 @@ class Scaleset(Endpoint):
else:
raise NotImplementedError
auto_scale = requests.AutoScaleOptions(
min=min_instances,
max=size,
default=size,
scale_out_amount=scale_out_amount,
scale_out_cooldown=scale_out_cooldown,
scale_in_amount=scale_in_amount,
scale_in_cooldown=scale_in_cooldown,
)
return self._req_model(
"POST",
models.Scaleset,
@ -1384,6 +1399,7 @@ class Scaleset(Endpoint):
spot_instances=spot_instances,
ephemeral_os_disks=ephemeral_os_disks,
tags=tags,
auto_scale=auto_scale,
),
)