From: Aaron Rosen <arosen@nicira.com>
Date: Mon, 17 Sep 2012 21:44:11 +0000 (-0400)
Subject: NVP plugin missing dhcp rpc callbacks
X-Git-Url: https://review.fuel-infra.org/gitweb?a=commitdiff_plain;h=2e4d541fa82e7c1294059ad7d3db1c044d1789ce;p=openstack-build%2Fneutron-build.git

NVP plugin missing dhcp rpc callbacks

Fixes bug 1052202

Change-Id: I6a906bd77cce85c7488d901bb2461fb970d1d5d0
---

diff --git a/quantum/plugins/nicira/nicira_nvp_plugin/QuantumPlugin.py b/quantum/plugins/nicira/nicira_nvp_plugin/QuantumPlugin.py
index 46dcfc181..a753a479b 100644
--- a/quantum/plugins/nicira/nicira_nvp_plugin/QuantumPlugin.py
+++ b/quantum/plugins/nicira/nicira_nvp_plugin/QuantumPlugin.py
@@ -41,10 +41,15 @@ from nvp_plugin_version import PLUGIN_VERSION
 from quantum.api.v2 import attributes
 from quantum.common import constants
 from quantum.common import exceptions as exception
+from quantum.common import topics
 from quantum.db import api as db
 from quantum.db import db_base_plugin_v2
+from quantum.db import dhcp_rpc_base
 from quantum.db import models_v2
 from quantum.openstack.common import cfg
+from quantum.openstack.common import context
+from quantum.openstack.common import rpc
+from quantum.openstack.common.rpc import dispatcher
 
 
 CONFIG_FILE = "nvp.ini"
@@ -94,6 +99,23 @@ def parse_config():
     return db_options, nvp_options, clusters_options
 
 
+class NVPRpcCallbacks(dhcp_rpc_base.DhcpRpcCallbackMixin):
+
+    # Set RPC API version to 1.0 by default.
+    RPC_API_VERSION = '1.0'
+
+    def __init__(self, rpc_context):
+        self.rpc_context = rpc_context
+
+    def create_rpc_dispatcher(self):
+        '''Get the rpc dispatcher for this manager.
+
+        If a manager would like to set an rpc API version, or support more than
+        one class as the target of rpc messages, override this method.
+        '''
+        return dispatcher.RpcDispatcher([self])
+
+
 class NVPCluster(object):
     """Encapsulates controller connection and api_client for a cluster.
 
@@ -268,6 +290,20 @@ class NvpPluginV2(db_base_plugin_v2.QuantumDbPluginV2):
             'base': models_v2.model_base.BASEV2,
         }
         db.configure_db(options)
+        self.setup_rpc()
+
+    def setup_rpc(self):
+        # RPC support for dhcp
+        self.topic = topics.PLUGIN
+        self.rpc_context = context.RequestContext('quantum', 'quantum',
+                                                  is_admin=False)
+        self.conn = rpc.create_connection(new=True)
+        self.callbacks = NVPRpcCallbacks(self.rpc_context)
+        self.dispatcher = self.callbacks.create_rpc_dispatcher()
+        self.conn.create_consumer(self.topic, self.dispatcher,
+                                  fanout=False)
+        # Consume from all consumers in a thread
+        self.conn.consume_in_thread()
 
     @property
     def cluster(self):