X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/2c693ee50677969a3aa81fe1ba8fd0c702acc20c..ca136c579fc014b6428c18cc3a74a45550539543:/services/nodemanager/arvnodeman/daemon.py diff --git a/services/nodemanager/arvnodeman/daemon.py b/services/nodemanager/arvnodeman/daemon.py index 83e3ec9fc4..eaf10be324 100644 --- a/services/nodemanager/arvnodeman/daemon.py +++ b/services/nodemanager/arvnodeman/daemon.py @@ -94,7 +94,7 @@ class NodeManagerDaemonActor(actor_class): def __init__(self, server_wishlist_actor, arvados_nodes_actor, cloud_nodes_actor, cloud_update_actor, timer_actor, arvados_factory, cloud_factory, - shutdown_windows, max_nodes, + shutdown_windows, min_nodes, max_nodes, poll_stale_after=600, node_stale_after=7200, node_setup_class=cnode.ComputeNodeSetupActor, node_shutdown_class=cnode.ComputeNodeShutdownActor, @@ -111,6 +111,7 @@ class NodeManagerDaemonActor(actor_class): self._logger = logging.getLogger('arvnodeman.daemon') self._later = self.actor_ref.proxy() self.shutdown_windows = shutdown_windows + self.min_nodes = min_nodes self.max_nodes = max_nodes self.poll_stale_after = poll_stale_after self.node_stale_after = node_stale_after @@ -203,7 +204,8 @@ class NodeManagerDaemonActor(actor_class): self.max_nodes) - self._nodes_up() def _nodes_excess(self): - return self._nodes_up() - self._nodes_busy() - len(self.last_wishlist) + needed_nodes = self._nodes_busy() + len(self.last_wishlist) + return (self._nodes_up() - max(self.min_nodes, needed_nodes)) def update_server_wishlist(self, wishlist): self._update_poll_time('server_wishlist')