76 lines
3.0 KiB
Python
76 lines
3.0 KiB
Python
# 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 neutron import manager
|
|
from neutron.plugins.common import constants as service_constants
|
|
from neutron.plugins.ml2 import plugin as ml2_plugin
|
|
from neutron.services.qos.notification_drivers import qos_base
|
|
|
|
from dragonflow.db.neutron import lockedobjects_db as lock_db
|
|
|
|
|
|
class DFQosServiceNotificationDriver(
|
|
qos_base.QosServiceNotificationDriverBase):
|
|
"""Dragonflow notification driver for QoS."""
|
|
|
|
def __init__(self):
|
|
self._nb_api = None
|
|
|
|
def get_description(self):
|
|
return "Notification driver for Dragonflow"
|
|
|
|
@property
|
|
def _plugin(self):
|
|
return manager.NeutronManager.get_service_plugins().get(
|
|
service_constants.QOS)
|
|
|
|
@property
|
|
def nb_api(self):
|
|
if self._nb_api is None:
|
|
plugin = manager.NeutronManager.get_plugin()
|
|
if isinstance(plugin, ml2_plugin.Ml2Plugin):
|
|
mech_driver = plugin.mechanism_manager.mech_drivers['df'].obj
|
|
self._nb_api = mech_driver.nb_api
|
|
else:
|
|
# DF neutron plugin
|
|
self._nb_api = plugin.nb_api
|
|
return self._nb_api
|
|
|
|
@lock_db.wrap_db_lock(lock_db.RESOURCE_QOS_POLICY_CREATE_OR_UPDATE)
|
|
def create_policy(self, context, policy):
|
|
self.nb_api.create_qos_policy(policy['id'],
|
|
policy['tenant_id'],
|
|
name=policy['name'],
|
|
rules=policy.get('rules', []),
|
|
version=policy['revision_number'])
|
|
|
|
@lock_db.wrap_db_lock(lock_db.RESOURCE_QOS_POLICY_CREATE_OR_UPDATE)
|
|
def update_policy(self, context, policy):
|
|
policy_id = policy['id']
|
|
# NOTE: Neutron will not pass policy with latest revision_number
|
|
# in argument. Get the latest policy from neutron.
|
|
policy_neutron = self._plugin.get_policy(context, policy_id)
|
|
|
|
self.nb_api.update_qos_policy(
|
|
policy_id, policy_neutron['tenant_id'],
|
|
name=policy['name'], rules=policy_neutron['rules'],
|
|
version=policy_neutron['revision_number'])
|
|
|
|
@lock_db.wrap_db_lock(lock_db.RESOURCE_QOS_POLICY_DELETE)
|
|
def delete_policy(self, context, policy):
|
|
policy_id = policy['id']
|
|
# Only id will be in policy in the argument. Get full policy from
|
|
# neutron.
|
|
policy_neutron = self._plugin.get_policy(context, policy_id)
|
|
|
|
self.nb_api.delete_qos_policy(policy_id, policy_neutron['tenant_id'])
|