164 lines
6.3 KiB
Python
164 lines
6.3 KiB
Python
# Copyright 2015 Red Hat, 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_versionedobjects import base as obj_base
|
|
from oslo_versionedobjects import fields as obj_fields
|
|
|
|
from neutron.common import exceptions
|
|
from neutron.db import api as db_api
|
|
from neutron.db.qos import api as qos_db_api
|
|
from neutron.db.qos import models as qos_db_model
|
|
from neutron.objects import base
|
|
from neutron.objects.qos import rule as rule_obj_impl
|
|
|
|
|
|
@obj_base.VersionedObjectRegistry.register
|
|
class QosPolicy(base.NeutronDbObject):
|
|
|
|
db_model = qos_db_model.QosPolicy
|
|
|
|
port_binding_model = qos_db_model.QosPortPolicyBinding
|
|
network_binding_model = qos_db_model.QosNetworkPolicyBinding
|
|
|
|
fields = {
|
|
'id': obj_fields.UUIDField(),
|
|
'tenant_id': obj_fields.UUIDField(),
|
|
'name': obj_fields.StringField(),
|
|
'description': obj_fields.StringField(),
|
|
'shared': obj_fields.BooleanField(default=False),
|
|
'rules': obj_fields.ListOfObjectsField('QosRule', subclasses=True),
|
|
}
|
|
|
|
fields_no_update = ['id', 'tenant_id']
|
|
|
|
synthetic_fields = ['rules']
|
|
|
|
def to_dict(self):
|
|
dict_ = super(QosPolicy, self).to_dict()
|
|
if 'rules' in dict_:
|
|
dict_['rules'] = [rule.to_dict() for rule in dict_['rules']]
|
|
return dict_
|
|
|
|
def obj_load_attr(self, attrname):
|
|
if attrname != 'rules':
|
|
raise exceptions.ObjectActionError(
|
|
action='obj_load_attr', reason='unable to load %s' % attrname)
|
|
|
|
if not hasattr(self, attrname):
|
|
self.reload_rules()
|
|
|
|
def reload_rules(self):
|
|
rules = rule_obj_impl.get_rules(self._context, self.id)
|
|
setattr(self, 'rules', rules)
|
|
self.obj_reset_changes(['rules'])
|
|
|
|
@staticmethod
|
|
def _is_policy_accessible(context, db_obj):
|
|
#TODO(QoS): Look at I3426b13eede8bfa29729cf3efea3419fb91175c4 for
|
|
# other possible solutions to this.
|
|
return (context.is_admin or
|
|
db_obj.shared or
|
|
db_obj.tenant_id == context.tenant_id)
|
|
|
|
@classmethod
|
|
def get_by_id(cls, context, id):
|
|
# We want to get the policy regardless of its tenant id. We'll make
|
|
# sure the tenant has permission to access the policy later on.
|
|
admin_context = context.elevated()
|
|
with db_api.autonested_transaction(admin_context.session):
|
|
policy_obj = super(QosPolicy, cls).get_by_id(admin_context, id)
|
|
if (not policy_obj or
|
|
not cls._is_policy_accessible(context, policy_obj)):
|
|
return
|
|
|
|
policy_obj.reload_rules()
|
|
return policy_obj
|
|
|
|
@classmethod
|
|
def get_objects(cls, context, **kwargs):
|
|
# We want to get the policy regardless of its tenant id. We'll make
|
|
# sure the tenant has permission to access the policy later on.
|
|
admin_context = context.elevated()
|
|
with db_api.autonested_transaction(admin_context.session):
|
|
objs = super(QosPolicy, cls).get_objects(admin_context,
|
|
**kwargs)
|
|
result = []
|
|
for obj in objs:
|
|
if not cls._is_policy_accessible(context, obj):
|
|
continue
|
|
obj.reload_rules()
|
|
result.append(obj)
|
|
return result
|
|
|
|
@classmethod
|
|
def _get_object_policy(cls, context, model, **kwargs):
|
|
with db_api.autonested_transaction(context.session):
|
|
binding_db_obj = db_api.get_object(context, model, **kwargs)
|
|
if binding_db_obj:
|
|
return cls.get_by_id(context, binding_db_obj['policy_id'])
|
|
|
|
@classmethod
|
|
def get_network_policy(cls, context, network_id):
|
|
return cls._get_object_policy(context, cls.network_binding_model,
|
|
network_id=network_id)
|
|
|
|
@classmethod
|
|
def get_port_policy(cls, context, port_id):
|
|
return cls._get_object_policy(context, cls.port_binding_model,
|
|
port_id=port_id)
|
|
|
|
# TODO(QoS): Consider extending base to trigger registered methods for us
|
|
def create(self):
|
|
with db_api.autonested_transaction(self._context.session):
|
|
super(QosPolicy, self).create()
|
|
self.reload_rules()
|
|
|
|
def delete(self):
|
|
models = (
|
|
('network', self.network_binding_model),
|
|
('port', self.port_binding_model)
|
|
)
|
|
with db_api.autonested_transaction(self._context.session):
|
|
for object_type, model in models:
|
|
binding_db_obj = db_api.get_object(self._context, model,
|
|
policy_id=self.id)
|
|
if binding_db_obj:
|
|
raise exceptions.QosPolicyInUse(
|
|
policy_id=self.id,
|
|
object_type=object_type,
|
|
object_id=binding_db_obj['%s_id' % object_type])
|
|
|
|
super(QosPolicy, self).delete()
|
|
|
|
def attach_network(self, network_id):
|
|
qos_db_api.create_policy_network_binding(self._context,
|
|
policy_id=self.id,
|
|
network_id=network_id)
|
|
|
|
def attach_port(self, port_id):
|
|
qos_db_api.create_policy_port_binding(self._context,
|
|
policy_id=self.id,
|
|
port_id=port_id)
|
|
|
|
def detach_network(self, network_id):
|
|
qos_db_api.delete_policy_network_binding(self._context,
|
|
policy_id=self.id,
|
|
network_id=network_id)
|
|
|
|
def detach_port(self, port_id):
|
|
qos_db_api.delete_policy_port_binding(self._context,
|
|
policy_id=self.id,
|
|
port_id=port_id)
|