From 4ef489b3974f261535d7b8ae921daa28bb37d37f Mon Sep 17 00:00:00 2001 From: Anshuman Goel Date: Tue, 3 Nov 2020 08:39:51 -0800 Subject: [PATCH] adding node shutdown (#252) --- src/api-service/__app__/onefuzzlib/pools.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/api-service/__app__/onefuzzlib/pools.py b/src/api-service/__app__/onefuzzlib/pools.py index 7e43ea473..3b1d99d10 100644 --- a/src/api-service/__app__/onefuzzlib/pools.py +++ b/src/api-service/__app__/onefuzzlib/pools.py @@ -898,6 +898,9 @@ class Scaleset(BASE_SCALESET, ORMMixin): def shutdown(self) -> None: size = get_vmss_size(self.scaleset_id) 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: self.halt()