7fdb98cf17
[1] and [2] were added to use the registry callback notifier for dhcp and nova notifications for the legacy wsgi layer. This adds the same functionality to the pecan wsgi layer. It just so happens to clean the code up nicely and also fixes a bug that was introduced by [3] that caused an ever increasing number of subscriptions to the registry callback notifier. [1] I7440becb6d30af7159ecaeba09d7a28eceb71bea [2] I1d7d4b80ee77deefce18df22f76cab81750c0397 [3] I607635601caff0322fd0c80c9023f5c4f663ca25 Change-Id: I0a8b64e7742283a9d6c6b42ebc27887836df69ec Closes-Bug: #1633296
106 lines
4.4 KiB
Python
106 lines
4.4 KiB
Python
# Copyright (c) 2015 Mirantis, Inc.
|
|
# All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
from oslo_log import log
|
|
from pecan import hooks
|
|
|
|
from neutron.callbacks import events
|
|
from neutron.callbacks import registry
|
|
from neutron.common import rpc as n_rpc
|
|
from neutron.pecan_wsgi import constants as pecan_constants
|
|
from neutron.pecan_wsgi.hooks import utils
|
|
|
|
LOG = log.getLogger(__name__)
|
|
|
|
|
|
class NotifierHook(hooks.PecanHook):
|
|
priority = 135
|
|
|
|
@property
|
|
def _notifier(self):
|
|
if not hasattr(self, '_notifier_inst'):
|
|
self._notifier_inst = n_rpc.get_notifier('network')
|
|
return self._notifier_inst
|
|
|
|
def before(self, state):
|
|
if state.request.method not in ('POST', 'PUT', 'DELETE'):
|
|
return
|
|
resource = state.request.context.get('resource')
|
|
if not resource:
|
|
return
|
|
if utils.is_member_action(utils.get_controller(state)):
|
|
return
|
|
action = pecan_constants.ACTION_MAP.get(state.request.method)
|
|
event = '%s.%s.start' % (resource, action)
|
|
if action in ('create', 'update'):
|
|
# notifier just gets plain old body without any treatment other
|
|
# than the population of the object ID being operated on
|
|
payload = state.request.json.copy()
|
|
if action == 'update':
|
|
payload['id'] = state.request.context.get('resource_id')
|
|
elif action == 'delete':
|
|
resource_id = state.request.context.get('resource_id')
|
|
payload = {resource + '_id': resource_id}
|
|
self._notifier.info(state.request.context.get('neutron_context'),
|
|
event, payload)
|
|
|
|
def after(self, state):
|
|
resource_name = state.request.context.get('resource')
|
|
collection_name = state.request.context.get('collection')
|
|
neutron_context = state.request.context.get('neutron_context')
|
|
if not resource_name:
|
|
LOG.debug("Skipping NotifierHook processing as there was no "
|
|
"resource associated with the request")
|
|
return
|
|
action = pecan_constants.ACTION_MAP.get(state.request.method)
|
|
if not action or action not in ('create', 'update', 'delete'):
|
|
LOG.debug("No notification will be sent for action: %s", action)
|
|
return
|
|
if utils.is_member_action(utils.get_controller(state)):
|
|
return
|
|
if state.response.status_int > 300:
|
|
LOG.debug("No notification will be sent due to unsuccessful "
|
|
"status code: %s", state.response.status_int)
|
|
return
|
|
|
|
original = {}
|
|
if (action in ('delete', 'update') and
|
|
state.request.context.get('original_resources', [])):
|
|
# We only need the original resource for updates and deletes
|
|
original = state.request.context.get('original_resources')[0]
|
|
if action == 'delete':
|
|
# The object has been deleted, so we must notify the agent with the
|
|
# data of the original object as the payload, but we do not need
|
|
# to pass it in as the original
|
|
result = {resource_name: original}
|
|
original = {}
|
|
else:
|
|
if not state.response.body:
|
|
result = {}
|
|
else:
|
|
result = state.response.json
|
|
|
|
notifier_method = '%s.%s.end' % (resource_name, action)
|
|
notifier_action = utils.get_controller(state).plugin_handlers[action]
|
|
registry.notify(resource_name, events.BEFORE_RESPONSE, self,
|
|
context=neutron_context, data=result,
|
|
method_name=notifier_method, action=notifier_action,
|
|
collection=collection_name, original=original)
|
|
|
|
if action == 'delete':
|
|
resource_id = state.request.context.get('resource_id')
|
|
result = {resource_name + '_id': resource_id}
|
|
self._notifier.info(neutron_context, notifier_method, result)
|