From: Peter Amstutz <pamstutz@veritasgenetics.com>
Date: Tue, 17 Jul 2018 14:26:29 +0000 (-0400)
Subject: Merge branch 'master' into 13822-nm-delayed-daemon
X-Git-Tag: 1.2.0~75^2
X-Git-Url: https://git.arvados.org/arvados.git/commitdiff_plain/6fe8e52020d421797306e5c6536afbcee761510a

Merge branch 'master' into 13822-nm-delayed-daemon

Arvados-DCO-1.1-Signed-off-by: Peter Amstutz <pamstutz@veritasgenetics.com>
---

6fe8e52020d421797306e5c6536afbcee761510a
diff --cc services/nodemanager/arvnodeman/computenode/driver/azure.py
index aa8f3c7695,ae554327ca..719124d400
--- a/services/nodemanager/arvnodeman/computenode/driver/azure.py
+++ b/services/nodemanager/arvnodeman/computenode/driver/azure.py
@@@ -87,7 -89,8 +89,8 @@@ echo %s > /var/tmp/arv-node-data/meta-d
          for n in nodes:
              # Need to populate Node.size
              if not n.size:
 -                n.size = self.sizes[n.extra["properties"]["hardwareProfile"]["vmSize"]]
 +                n.size = self.sizes()[n.extra["properties"]["hardwareProfile"]["vmSize"]]
+             n.extra['arvados_node_size'] = n.extra.get('tags', {}).get('arvados_node_size')
          return nodes
  
      def broken(self, cloud_node):
diff --cc services/nodemanager/arvnodeman/computenode/driver/ec2.py
index 07ed90dfa5,2b15642797..56812d258a
--- a/services/nodemanager/arvnodeman/computenode/driver/ec2.py
+++ b/services/nodemanager/arvnodeman/computenode/driver/ec2.py
@@@ -102,7 -110,8 +110,8 @@@ class ComputeNodeDriver(BaseComputeNode
          nodes = super(ComputeNodeDriver, self).list_nodes()
          for n in nodes:
              if not n.size:
 -                n.size = self.sizes[n.extra["instance_type"]]
 +                n.size = self.sizes()[n.extra["instance_type"]]
+             n.extra['arvados_node_size'] = n.extra.get('tags', {}).get('arvados_node_size')
          return nodes
  
      @classmethod
diff --cc services/nodemanager/arvnodeman/computenode/driver/gce.py
index f1238db403,be39ecba6b..11025f7840
--- a/services/nodemanager/arvnodeman/computenode/driver/gce.py
+++ b/services/nodemanager/arvnodeman/computenode/driver/gce.py
@@@ -114,12 -115,14 +114,14 @@@ class ComputeNodeDriver(BaseComputeNode
          nodelist = [node for node in
                      super(ComputeNodeDriver, self).list_nodes()
                      if self.node_tags.issubset(node.extra.get('tags', []))]
-         # As of 0.18, the libcloud GCE driver sets node.size to the size's name.
-         # It's supposed to be the actual size object.  Check that it's not,
-         # and monkeypatch the results when that's the case.
-         if nodelist and not hasattr(nodelist[0].size, 'id'):
-             for node in nodelist:
+         for node in nodelist:
+             # As of 0.18, the libcloud GCE driver sets node.size to the size's name.
+             # It's supposed to be the actual size object.  Check that it's not,
+             # and monkeypatch the results when that's the case.
+             if not hasattr(node.size, 'id'):
 -                node.size = self._sizes_by_id[node.size]
 +                node.size = self.sizes()[node.size]
+             # Get arvados-assigned cloud size id
+             node.extra['arvados_node_size'] = node.extra.get('metadata', {}).get('arvados_node_size')
          return nodelist
  
      @classmethod