From 5d00101e5c0e6ba5e4ecba49c132f74c18b5267c Mon Sep 17 00:00:00 2001
From: Sandy Walsh <sandy.walsh@rackspace.com>
Date: Thu, 3 Mar 2011 16:28:04 -0400
Subject: [PATCH] start of fanout

---
 nova/rpc.py                    | 20 ++++++++++++++++++++
 nova/scheduler/zone_manager.py |  4 ++++
 2 files changed, 24 insertions(+)

diff --git a/nova/rpc.py b/nova/rpc.py
index 8fe4565d..a02cdc90 100644
--- a/nova/rpc.py
+++ b/nova/rpc.py
@@ -218,6 +218,16 @@ class TopicPublisher(Publisher):
         super(TopicPublisher, self).__init__(connection=connection)
 
 
+class FanoutPublisher(Publisher):
+    """Publishes messages to a fanout exchange."""
+    exchange_type = "fanout"
+
+    def __init__(self, topic, connection=None):
+        self.exchange = "%s_fanout" % topic
+        self.durable = False
+        super(FanoutPublisher, self).__init__(connection=connection)
+
+
 class DirectConsumer(Consumer):
     """Consumes messages directly on a channel specified by msg_id"""
     exchange_type = "direct"
@@ -360,6 +370,16 @@ def cast(context, topic, msg):
     publisher.close()
 
 
+def fanout_cast(context, topic, msg):
+    """Sends a message on a fanout exchange without waiting for a response"""
+    LOG.debug(_("Making asynchronous fanout cast..."))
+    _pack_context(msg, context)
+    conn = Connection.instance()
+    publisher = FanoutPublisher(topic, connection=conn)
+    publisher.send(msg)
+    publisher.close()
+
+
 def generic_response(message_data, message):
     """Logs a result and exits"""
     LOG.debug(_('response %s'), message_data)
diff --git a/nova/scheduler/zone_manager.py b/nova/scheduler/zone_manager.py
index edf9000c..eedc5c23 100644
--- a/nova/scheduler/zone_manager.py
+++ b/nova/scheduler/zone_manager.py
@@ -105,6 +105,7 @@ class ZoneManager(object):
     def __init__(self):
         self.last_zone_db_check = datetime.min
         self.zone_states = {}
+        self.compute_states = {}
         self.green_pool = greenpool.GreenPool()
 
     def get_zone_list(self):
@@ -141,3 +142,6 @@ class ZoneManager(object):
             self.last_zone_db_check = datetime.now()
             self._refresh_from_db(context)
         self._poll_zones(context)
+
+    def update_compute_capabilities(self):
+        logging.debug(_("****** UPDATE COMPUTE CAPABILITIES *******"))