X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/51e79a3be745cfd37efc49471ae21a1d54662e7f..ca136c579fc014b6428c18cc3a74a45550539543:/services/nodemanager/arvnodeman/jobqueue.py

diff --git a/services/nodemanager/arvnodeman/jobqueue.py b/services/nodemanager/arvnodeman/jobqueue.py
index 59659fec0e..239934f529 100644
--- a/services/nodemanager/arvnodeman/jobqueue.py
+++ b/services/nodemanager/arvnodeman/jobqueue.py
@@ -38,10 +38,11 @@ class ServerCalculator(object):
             return True
 
 
-    def __init__(self, server_list, max_nodes=None):
+    def __init__(self, server_list, min_nodes=0, max_nodes=None):
         self.cloud_sizes = [self.CloudSizeWrapper(s, **kws)
                             for s, kws in server_list]
         self.cloud_sizes.sort(key=lambda s: s.price)
+        self.min_nodes = min_nodes
         self.max_nodes = max_nodes or float('inf')
         self.logger = logging.getLogger('arvnodeman.jobqueue')
         self.logged_jobs = set()
@@ -78,6 +79,13 @@ class ServerCalculator(object):
             elif (want_count <= self.max_nodes):
                 servers.extend([cloud_size.real] * max(1, want_count))
         self.logged_jobs.intersection_update(seen_jobs)
+
+        # Make sure the server queue has at least enough entries to
+        # satisfy min_nodes.
+        node_shortfall = self.min_nodes - len(servers)
+        if node_shortfall > 0:
+            basic_node = self.cloud_size_for_constraints({})
+            servers.extend([basic_node.real] * node_shortfall)
         return servers