From: Guido Trotter <ultrotter@google.com>
Date: Mon, 31 Aug 2009 14:54:20 +0000 (+0100)
Subject: confd: add cluster master query
X-Git-Tag: v2.1.0beta0~255
X-Git-Url: https://code.grnet.gr/git/ganeti-local/commitdiff_plain/481665515e0dbe450d5c7014a01750485eb10ba7

confd: add cluster master query

Signed-off-by: Guido Trotter <ultrotter@google.com>
Reviewed-by: Michael Hanselmann <hansmi@google.com>
---

diff --git a/lib/confd/querylib.py b/lib/confd/querylib.py
index af0c32d..b299034 100644
--- a/lib/confd/querylib.py
+++ b/lib/confd/querylib.py
@@ -82,6 +82,25 @@ class PingQuery(ConfdQuery):
 
     return status, answer
 
+class ClusterMasterQuery(ConfdQuery):
+  """Cluster master query.
+
+  It accepts no arguments, and returns the current cluster master.
+
+  """
+  def Exec(self, query):
+    """ClusterMasterQuery main execution
+
+    """
+    if query is None:
+      status = constants.CONFD_REPL_STATUS_OK
+      answer = self.reader.GetMasterNode()
+    else:
+      status = constants.CONFD_REPL_STATUS_ERROR
+      answer = 'master query accepts no query argument'
+
+    return status, answer
+
 
 class NodeRoleQuery(ConfdQuery):
   """A query for the role of a node.
diff --git a/lib/confd/server.py b/lib/confd/server.py
index 601b957..f27b225 100644
--- a/lib/confd/server.py
+++ b/lib/confd/server.py
@@ -51,6 +51,7 @@ class ConfdProcessor(object):
       constants.CONFD_REQ_NODE_ROLE_BYNAME: querylib.NodeRoleQuery,
       constants.CONFD_REQ_NODE_PIP_BY_INSTANCE_IP:
         querylib.InstanceIpToNodePrimaryIpQuery,
+      constants.CONFD_REQ_CLUSTER_MASTER: querylib.ClusterMasterQuery,
   }
 
   def __init__(self):
diff --git a/lib/constants.py b/lib/constants.py
index 5384765..80a663a 100644
--- a/lib/constants.py
+++ b/lib/constants.py
@@ -638,11 +638,13 @@ CONFD_PROTOCOL_VERSION = 1
 CONFD_REQ_PING = 0
 CONFD_REQ_NODE_ROLE_BYNAME = 1
 CONFD_REQ_NODE_PIP_BY_INSTANCE_IP = 2
+CONFD_REQ_CLUSTER_MASTER = 3
 
 CONFD_REQS = frozenset([
   CONFD_REQ_PING,
   CONFD_REQ_NODE_ROLE_BYNAME,
   CONFD_REQ_NODE_PIP_BY_INSTANCE_IP,
+  CONFD_REQ_CLUSTER_MASTER,
   ])
 
 CONFD_REPL_STATUS_OK = 0