e19cb128b0
This adds a method to get a count of fixed ips in the database for checking quota. Note that fixed ip quota is only relevant to nova-network and will be obsolete when nova-network is removed. Co-Authored-By: melanie witt <melwittt@gmail.com> Part of blueprint cells-count-resources-to-check-quota-in-api Change-Id: Ia781d9785165e71d7b90657b1931f29d988d4956
316 lines
14 KiB
Python
316 lines
14 KiB
Python
# Copyright 2014 Red Hat, Inc.
|
|
#
|
|
# 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_utils import timeutils
|
|
from oslo_utils import versionutils
|
|
|
|
from nova import db
|
|
from nova.db.sqlalchemy import api as db_api
|
|
from nova.db.sqlalchemy import models
|
|
from nova import exception
|
|
from nova import objects
|
|
from nova.objects import base as obj_base
|
|
from nova.objects import fields
|
|
from nova import utils
|
|
|
|
|
|
FIXED_IP_OPTIONAL_ATTRS = ['instance', 'network', 'virtual_interface',
|
|
'floating_ips']
|
|
|
|
|
|
# TODO(berrange): Remove NovaObjectDictCompat
|
|
@obj_base.NovaObjectRegistry.register
|
|
class FixedIP(obj_base.NovaPersistentObject, obj_base.NovaObject,
|
|
obj_base.NovaObjectDictCompat):
|
|
# Version 1.0: Initial version
|
|
# Version 1.1: Added virtual_interface field
|
|
# Version 1.2: Instance version 1.14
|
|
# Version 1.3: Instance 1.15
|
|
# Version 1.4: Added default_route field
|
|
# Version 1.5: Added floating_ips field
|
|
# Version 1.6: Instance 1.16
|
|
# Version 1.7: Instance 1.17
|
|
# Version 1.8: Instance 1.18
|
|
# Version 1.9: Instance 1.19
|
|
# Version 1.10: Instance 1.20
|
|
# Version 1.11: Instance 1.21
|
|
# Version 1.12: Instance 1.22, FloatingIPList 1.9
|
|
# Version 1.13: Instance 1.23, FloatingIPList 1.10
|
|
# Version 1.14: Added vif_id kwarg to associate(_pool), FloatingIPList 1.11
|
|
VERSION = '1.14'
|
|
|
|
fields = {
|
|
'id': fields.IntegerField(),
|
|
'address': fields.IPV4AndV6AddressField(),
|
|
'network_id': fields.IntegerField(nullable=True),
|
|
'virtual_interface_id': fields.IntegerField(nullable=True),
|
|
'instance_uuid': fields.UUIDField(nullable=True),
|
|
'allocated': fields.BooleanField(),
|
|
'leased': fields.BooleanField(),
|
|
'reserved': fields.BooleanField(),
|
|
'host': fields.StringField(nullable=True),
|
|
'default_route': fields.BooleanField(),
|
|
'instance': fields.ObjectField('Instance', nullable=True),
|
|
'network': fields.ObjectField('Network', nullable=True),
|
|
'virtual_interface': fields.ObjectField('VirtualInterface',
|
|
nullable=True),
|
|
# NOTE(danms): This should not ever be made lazy-loadable
|
|
# because it would create a bit of a loop between FixedIP
|
|
# and FloatingIP
|
|
'floating_ips': fields.ObjectField('FloatingIPList'),
|
|
}
|
|
|
|
def obj_make_compatible(self, primitive, target_version):
|
|
super(FixedIP, self).obj_make_compatible(primitive, target_version)
|
|
target_version = versionutils.convert_version_to_tuple(target_version)
|
|
if target_version < (1, 4) and 'default_route' in primitive:
|
|
del primitive['default_route']
|
|
|
|
@staticmethod
|
|
def _from_db_object(context, fixedip, db_fixedip, expected_attrs=None):
|
|
if expected_attrs is None:
|
|
expected_attrs = []
|
|
for field in fixedip.fields:
|
|
if field == 'default_route':
|
|
# NOTE(danms): This field is only set when doing a
|
|
# FixedIPList.get_by_network() because it's a relatively
|
|
# special-case thing, so skip it here
|
|
continue
|
|
if field not in FIXED_IP_OPTIONAL_ATTRS:
|
|
fixedip[field] = db_fixedip[field]
|
|
# NOTE(danms): Instance could be deleted, and thus None
|
|
if 'instance' in expected_attrs:
|
|
fixedip.instance = objects.Instance._from_db_object(
|
|
context,
|
|
objects.Instance(context),
|
|
db_fixedip['instance']) if db_fixedip['instance'] else None
|
|
if 'network' in expected_attrs:
|
|
fixedip.network = objects.Network._from_db_object(
|
|
context,
|
|
objects.Network(context),
|
|
db_fixedip['network']) if db_fixedip['network'] else None
|
|
if 'virtual_interface' in expected_attrs:
|
|
db_vif = db_fixedip['virtual_interface']
|
|
vif = objects.VirtualInterface._from_db_object(
|
|
context,
|
|
objects.VirtualInterface(context),
|
|
db_fixedip['virtual_interface']) if db_vif else None
|
|
fixedip.virtual_interface = vif
|
|
if 'floating_ips' in expected_attrs:
|
|
fixedip.floating_ips = obj_base.obj_make_list(
|
|
context, objects.FloatingIPList(context),
|
|
objects.FloatingIP, db_fixedip['floating_ips'])
|
|
fixedip._context = context
|
|
fixedip.obj_reset_changes()
|
|
return fixedip
|
|
|
|
@obj_base.remotable_classmethod
|
|
def get_by_id(cls, context, id, expected_attrs=None):
|
|
if expected_attrs is None:
|
|
expected_attrs = []
|
|
get_network = 'network' in expected_attrs
|
|
db_fixedip = db.fixed_ip_get(context, id, get_network=get_network)
|
|
return cls._from_db_object(context, cls(context), db_fixedip,
|
|
expected_attrs)
|
|
|
|
@obj_base.remotable_classmethod
|
|
def get_by_address(cls, context, address, expected_attrs=None):
|
|
if expected_attrs is None:
|
|
expected_attrs = []
|
|
db_fixedip = db.fixed_ip_get_by_address(context, str(address),
|
|
columns_to_join=expected_attrs)
|
|
return cls._from_db_object(context, cls(context), db_fixedip,
|
|
expected_attrs)
|
|
|
|
@obj_base.remotable_classmethod
|
|
def get_by_floating_address(cls, context, address):
|
|
db_fixedip = db.fixed_ip_get_by_floating_address(context, str(address))
|
|
if db_fixedip is not None:
|
|
return cls._from_db_object(context, cls(context), db_fixedip)
|
|
|
|
@obj_base.remotable_classmethod
|
|
def get_by_network_and_host(cls, context, network_id, host):
|
|
db_fixedip = db.fixed_ip_get_by_network_host(context, network_id, host)
|
|
return cls._from_db_object(context, cls(context), db_fixedip)
|
|
|
|
@obj_base.remotable_classmethod
|
|
def associate(cls, context, address, instance_uuid, network_id=None,
|
|
reserved=False, vif_id=None):
|
|
db_fixedip = db.fixed_ip_associate(context, address, instance_uuid,
|
|
network_id=network_id,
|
|
reserved=reserved,
|
|
virtual_interface_id=vif_id)
|
|
return cls._from_db_object(context, cls(context), db_fixedip)
|
|
|
|
@obj_base.remotable_classmethod
|
|
def associate_pool(cls, context, network_id, instance_uuid=None,
|
|
host=None, vif_id=None):
|
|
db_fixedip = db.fixed_ip_associate_pool(context, network_id,
|
|
instance_uuid=instance_uuid,
|
|
host=host,
|
|
virtual_interface_id=vif_id)
|
|
return cls._from_db_object(context, cls(context), db_fixedip)
|
|
|
|
@obj_base.remotable_classmethod
|
|
def disassociate_by_address(cls, context, address):
|
|
db.fixed_ip_disassociate(context, address)
|
|
|
|
@obj_base.remotable_classmethod
|
|
def _disassociate_all_by_timeout(cls, context, host, time_str):
|
|
time = timeutils.parse_isotime(time_str)
|
|
return db.fixed_ip_disassociate_all_by_timeout(context, host, time)
|
|
|
|
@classmethod
|
|
def disassociate_all_by_timeout(cls, context, host, time):
|
|
return cls._disassociate_all_by_timeout(context, host,
|
|
utils.isotime(time))
|
|
|
|
@obj_base.remotable
|
|
def create(self):
|
|
updates = self.obj_get_changes()
|
|
if 'id' in updates:
|
|
raise exception.ObjectActionError(action='create',
|
|
reason='already created')
|
|
if 'address' in updates:
|
|
updates['address'] = str(updates['address'])
|
|
db_fixedip = db.fixed_ip_create(self._context, updates)
|
|
self._from_db_object(self._context, self, db_fixedip)
|
|
|
|
@obj_base.remotable
|
|
def save(self):
|
|
updates = self.obj_get_changes()
|
|
if 'address' in updates:
|
|
raise exception.ObjectActionError(action='save',
|
|
reason='address is not mutable')
|
|
db.fixed_ip_update(self._context, str(self.address), updates)
|
|
self.obj_reset_changes()
|
|
|
|
@obj_base.remotable
|
|
def disassociate(self):
|
|
db.fixed_ip_disassociate(self._context, str(self.address))
|
|
self.instance_uuid = None
|
|
self.instance = None
|
|
self.obj_reset_changes(['instance_uuid', 'instance'])
|
|
|
|
|
|
@obj_base.NovaObjectRegistry.register
|
|
class FixedIPList(obj_base.ObjectListBase, obj_base.NovaObject):
|
|
# Version 1.0: Initial version
|
|
# Version 1.1: Added get_by_network()
|
|
# Version 1.2: FixedIP <= version 1.2
|
|
# Version 1.3: FixedIP <= version 1.3
|
|
# Version 1.4: FixedIP <= version 1.4
|
|
# Version 1.5: FixedIP <= version 1.5, added expected attrs to gets
|
|
# Version 1.6: FixedIP <= version 1.6
|
|
# Version 1.7: FixedIP <= version 1.7
|
|
# Version 1.8: FixedIP <= version 1.8
|
|
# Version 1.9: FixedIP <= version 1.9
|
|
# Version 1.10: FixedIP <= version 1.10
|
|
# Version 1.11: FixedIP <= version 1.11
|
|
# Version 1.12: FixedIP <= version 1.12
|
|
# Version 1.13: FixedIP <= version 1.13
|
|
# Version 1.14: FixedIP <= version 1.14
|
|
# Version 1.15: Added get_count_by_project() for quotas
|
|
VERSION = '1.15'
|
|
|
|
fields = {
|
|
'objects': fields.ListOfObjectsField('FixedIP'),
|
|
}
|
|
|
|
@staticmethod
|
|
@db_api.pick_context_manager_reader
|
|
def _get_count_by_project_from_db(context, project_id):
|
|
return context.session.query(models.FixedIp.id).\
|
|
filter_by(deleted=0).\
|
|
join(models.Instance,
|
|
models.Instance.uuid == models.FixedIp.instance_uuid).\
|
|
filter(models.Instance.project_id == project_id).\
|
|
count()
|
|
|
|
@obj_base.remotable_classmethod
|
|
def get_all(cls, context):
|
|
db_fixedips = db.fixed_ip_get_all(context)
|
|
return obj_base.obj_make_list(context, cls(context),
|
|
objects.FixedIP, db_fixedips)
|
|
|
|
@obj_base.remotable_classmethod
|
|
def get_by_instance_uuid(cls, context, instance_uuid):
|
|
expected_attrs = ['network', 'virtual_interface', 'floating_ips']
|
|
db_fixedips = db.fixed_ip_get_by_instance(context, instance_uuid)
|
|
return obj_base.obj_make_list(context, cls(context),
|
|
objects.FixedIP, db_fixedips,
|
|
expected_attrs=expected_attrs)
|
|
|
|
@obj_base.remotable_classmethod
|
|
def get_by_host(cls, context, host):
|
|
db_fixedips = db.fixed_ip_get_by_host(context, host)
|
|
return obj_base.obj_make_list(context, cls(context),
|
|
objects.FixedIP, db_fixedips)
|
|
|
|
@obj_base.remotable_classmethod
|
|
def get_by_virtual_interface_id(cls, context, vif_id):
|
|
expected_attrs = ['network', 'floating_ips']
|
|
db_fixedips = db.fixed_ips_by_virtual_interface(context, vif_id)
|
|
return obj_base.obj_make_list(context, cls(context),
|
|
objects.FixedIP, db_fixedips,
|
|
expected_attrs=expected_attrs)
|
|
|
|
@obj_base.remotable_classmethod
|
|
def get_by_network(cls, context, network, host=None):
|
|
ipinfo = db.network_get_associated_fixed_ips(context,
|
|
network['id'],
|
|
host=host)
|
|
if not ipinfo:
|
|
return []
|
|
|
|
fips = cls(context=context, objects=[])
|
|
|
|
for info in ipinfo:
|
|
inst = objects.Instance(context=context,
|
|
uuid=info['instance_uuid'],
|
|
hostname=info['instance_hostname'],
|
|
created_at=info['instance_created'],
|
|
updated_at=info['instance_updated'])
|
|
vif = objects.VirtualInterface(context=context,
|
|
id=info['vif_id'],
|
|
address=info['vif_address'])
|
|
fip = objects.FixedIP(context=context,
|
|
address=info['address'],
|
|
instance_uuid=info['instance_uuid'],
|
|
network_id=info['network_id'],
|
|
virtual_interface_id=info['vif_id'],
|
|
allocated=info['allocated'],
|
|
leased=info['leased'],
|
|
default_route=info['default_route'],
|
|
instance=inst,
|
|
virtual_interface=vif)
|
|
fips.objects.append(fip)
|
|
fips.obj_reset_changes()
|
|
return fips
|
|
|
|
@obj_base.remotable_classmethod
|
|
def bulk_create(self, context, fixed_ips):
|
|
ips = []
|
|
for fixedip in fixed_ips:
|
|
ip = obj_base.obj_to_primitive(fixedip)
|
|
if 'id' in ip:
|
|
raise exception.ObjectActionError(action='create',
|
|
reason='already created')
|
|
ips.append(ip)
|
|
db.fixed_ip_bulk_create(context, ips)
|
|
|
|
@obj_base.remotable_classmethod
|
|
def get_count_by_project(cls, context, project_id):
|
|
return cls._get_count_by_project_from_db(context, project_id)
|