Added support for node metadata in API calls

Author: Chris Barbour

consulate/api/catalog.py

@@ -19,7 +19,8 @@ def __init__(self, uri, adapter, dc=None, token=None):
     def register(self, node, address,
                  datacenter=None,
                  service=None,
-                 check=None):
+                 check=None,
+                 node_meta=None):
         """A a low level mechanism for directly registering or updating
         entries in the catalog. It is usually recommended to use the agent
         local endpoints, as they are simpler and perform anti-entropy.
@@ -76,11 +77,20 @@ def register(self, node, address,
                 'ServiceID': 'redis1'
             }
 
+        Example node_meta dict:
+
+        .. code:: python
+
+            'NodeMeta': {
+                'somekey': 'somevalue'
+            }
+
         :param str node: The node name
         :param str address: The node address
         :param str datacenter: The optional node datacenter
         :param dict service: An optional node service
         :param dict check: An optional node check
+        :param dict node_meta: Optional node metadata
         :rtype: bool
 
         """
@@ -91,6 +101,8 @@ def register(self, node, address,
             payload['Service'] = service
         if check:
             payload['Check'] = check
+        if node_meta:
+            payload['NodeMeta'] = node_meta
 
         return self._put_response_body(['register'], None, payload)
 
@@ -141,13 +153,15 @@ def node(self, node_id):
         """
         return self._get(['node', node_id])
 
-    def nodes(self):
+    def nodes(self, node_meta=None):
         """Return all of the nodes for the current datacenter.
 
+        :param str node_meta: Desired node metadata
         :rtype: list
 
         """
-        return self._get_list(['nodes'])
+        query_params = {'node-meta': node_meta} if node_meta else {}
+        return self._get_list(['nodes'], query_params)
 
     def service(self, service_id):
         """Return the service details for the given service

consulate/api/health.py

@@ -14,14 +14,16 @@ class Health(base.Endpoint):
 
     """
 
-    def checks(self, service_id):
+    def checks(self, service_id, node_meta=None):
         """Return checks for the given service.
 
         :param str service_id: The service ID
+        :param str node_meta: Filter checks using node metadata
         :rtype: list
 
         """
-        return self._get_list(['checks', service_id])
+        query_params = {'node-meta': node_meta} if node_meta else {}
+        return self._get_list(['checks', service_id], query_params)
 
     def node(self, node_id):
         """Return the health info for a given node.
@@ -32,10 +34,11 @@ def node(self, node_id):
         """
         return self._get_list(['node', node_id])
 
-    def service(self, service_id, tag=None, passing=None):
+    def service(self, service_id, tag=None, passing=None, node_meta=None):
         """Returns the nodes and health info of a service
 
         :param str service_id: The service ID
+        :param str node_meta: Filter services using node metadata
         :rtype: list
 
         """
@@ -45,6 +48,8 @@ def service(self, service_id, tag=None, passing=None):
             query_params['tag'] = tag
         if passing:
             query_params['passing'] = ''
+        if node_meta:
+            query_params['node-meta'] = node_meta
 
         return self._get_list(['service', service_id],
                               query_params=query_params)