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

28
src/api-service/__app__/onefuzzlib/workers/scalesets.py
View File

@ -13,6 +13,7 @@ from onefuzztypes.events import (
EventScalesetCreated,
EventScalesetDeleted,
EventScalesetFailed,
EventScalesetResizeScheduled,
EventScalesetStateUpdated,
)
from onefuzztypes.models import Error
@ -472,6 +473,14 @@ class Scaleset(BASE_SCALESET, ORMMixin):
)
self.set_state(ScalesetState.resize)
def set_size(self, size: int) -> None:
# ensure we always stay within max_size boundaries
size = min(size, self.max_size())
if self.size != size:
self.size = size
self.set_state(ScalesetState.resize)
self.save()
def resize(self) -> None:
# no longer needing to resize
if self.state != ScalesetState.resize:
@ -780,8 +789,19 @@ class Scaleset(BASE_SCALESET, ORMMixin):
self.state = state
self.save()
send_event(
EventScalesetStateUpdated(
scaleset_id=self.scaleset_id, pool_name=self.pool_name, state=self.state
if self.state == ScalesetState.resize:
send_event(
EventScalesetResizeScheduled(
scaleset_id=self.scaleset_id,
pool_name=self.pool_name,
size=self.size,
)
)
else:
send_event(
EventScalesetStateUpdated(
scaleset_id=self.scaleset_id,
pool_name=self.pool_name,
state=self.state,
)
)
)