services/nodemanager/arvnodeman/computenode/driver/dummy.py

   1 #!/usr/bin/env python
   2
   3 from __future__ import absolute_import, print_function
   4
   5 import time
   6
   7 import libcloud.compute.providers as cloud_provider
   8 import libcloud.compute.types as cloud_types
   9
  10 from . import BaseComputeNodeDriver
  11 from .. import arvados_node_fqdn
  12
  13 class ComputeNodeDriver(BaseComputeNodeDriver):
  14     """Compute node driver wrapper for libcloud's dummy driver.
  15
  16     This class provides the glue necessary to run the node manager with a
  17     dummy cloud.  It's useful for testing.
  18     """
  19     DEFAULT_DRIVER = cloud_provider.get_driver(cloud_types.Provider.DUMMY)
  20     DEFAULT_REAL = DEFAULT_DRIVER('ComputeNodeDriver')
  21     DUMMY_START_TIME = time.time()
  22
  23     def __init__(self, auth_kwargs, list_kwargs, create_kwargs,
  24                  driver_class=DEFAULT_DRIVER):
  25         super(ComputeNodeDriver, self).__init__(
  26             auth_kwargs, list_kwargs, create_kwargs, driver_class)
  27         if driver_class is self.DEFAULT_DRIVER:
  28             self.real = self.DEFAULT_REAL
  29
  30     def _ensure_private_ip(self, node):
  31         if not node.private_ips:
  32             node.private_ips = ['10.10.0.{}'.format(node.id)]
  33
  34     def arvados_create_kwargs(self, size, arvados_node):
  35         return {}
  36
  37     def list_nodes(self):
  38         nodelist = super(ComputeNodeDriver, self).list_nodes()
  39         for node in nodelist:
  40             self._ensure_private_ip(node)
  41             node.size = self.sizes["1"]
  42         return nodelist
  43
  44     def create_node(self, size, arvados_node):
  45         node = super(ComputeNodeDriver, self).create_node(size, arvados_node)
  46         self._ensure_private_ip(node)
  47         return node
  48
  49     def sync_node(self, cloud_node, arvados_node):
  50         cloud_node.name = arvados_node_fqdn(arvados_node)
  51
  52     @classmethod
  53     def node_fqdn(cls, node):
  54         return node.name
  55
  56     @classmethod
  57     def node_start_time(cls, node):
  58         return cls.DUMMY_START_TIME