Merge branch 'master' into 4823-python-sdk-writable-collection-api
[arvados.git] / services / nodemanager / arvnodeman / nodelist.py
1 #!/usr/bin/env python
2
3 from __future__ import absolute_import, print_function
4
5 from . import clientactor
6 from . import config
7
8 class ArvadosNodeListMonitorActor(clientactor.RemotePollLoopActor):
9     """Actor to poll the Arvados node list.
10
11     This actor regularly polls the list of Arvados node records, and
12     sends it to subscribers.
13     """
14     LOGGER_NAME = 'arvnodeman.arvados_nodes'
15
16     def is_common_error(self, exception):
17         return isinstance(exception, config.ARVADOS_ERRORS)
18
19     def _item_key(self, node):
20         return node['uuid']
21
22     def _send_request(self):
23         return self._client.nodes().list(limit=10000).execute()['items']
24
25
26 class CloudNodeListMonitorActor(clientactor.RemotePollLoopActor):
27     """Actor to poll the cloud node list.
28
29     This actor regularly polls the cloud to get a list of running compute
30     nodes, and sends it to subscribers.
31     """
32     LOGGER_NAME = 'arvnodeman.cloud_nodes'
33
34     def is_common_error(self, exception):
35         return self._client.is_cloud_exception(exception)
36
37     def _item_key(self, node):
38         return node.id
39
40     def _send_request(self):
41         return self._client.list_nodes()