215 lines
8.1 KiB
Python
215 lines
8.1 KiB
Python
# Copyright 2013 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 nova import block_device
|
|
from nova.cells import rpcapi as cells_rpcapi
|
|
from nova import db
|
|
from nova import exception
|
|
from nova.objects import base
|
|
from nova.objects import fields
|
|
from nova.objects import instance
|
|
from nova.openstack.common.gettextutils import _
|
|
from nova.openstack.common import log as logging
|
|
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
_BLOCK_DEVICE_OPTIONAL_JOINED_FIELD = ['instance']
|
|
BLOCK_DEVICE_OPTIONAL_ATTRS = _BLOCK_DEVICE_OPTIONAL_JOINED_FIELD
|
|
|
|
|
|
def _expected_cols(expected_attrs):
|
|
return [attr for attr in expected_attrs
|
|
if attr in _BLOCK_DEVICE_OPTIONAL_JOINED_FIELD]
|
|
|
|
|
|
class BlockDeviceMapping(base.NovaPersistentObject, base.NovaObject):
|
|
# Version 1.0: Initial version
|
|
VERSION = '1.0'
|
|
|
|
fields = {
|
|
'id': fields.IntegerField(),
|
|
'instance_uuid': fields.UUIDField(),
|
|
'instance': fields.ObjectField('Instance', nullable=True),
|
|
'source_type': fields.StringField(nullable=True),
|
|
'destination_type': fields.StringField(nullable=True),
|
|
'guest_format': fields.StringField(nullable=True),
|
|
'device_type': fields.StringField(nullable=True),
|
|
'disk_bus': fields.StringField(nullable=True),
|
|
'boot_index': fields.IntegerField(nullable=True),
|
|
'device_name': fields.StringField(nullable=True),
|
|
'delete_on_termination': fields.BooleanField(default=False),
|
|
'snapshot_id': fields.StringField(nullable=True),
|
|
'volume_id': fields.StringField(nullable=True),
|
|
'volume_size': fields.IntegerField(nullable=True),
|
|
'image_id': fields.StringField(nullable=True),
|
|
'no_device': fields.BooleanField(default=False),
|
|
'connection_info': fields.StringField(nullable=True),
|
|
}
|
|
|
|
@staticmethod
|
|
def _from_db_object(context, block_device_obj,
|
|
db_block_device, expected_attrs=None):
|
|
if expected_attrs is None:
|
|
expected_attrs = []
|
|
for key in block_device_obj.fields:
|
|
if key in BLOCK_DEVICE_OPTIONAL_ATTRS:
|
|
continue
|
|
block_device_obj[key] = db_block_device[key]
|
|
if 'instance' in expected_attrs:
|
|
my_inst = instance.Instance()
|
|
instance.Instance._from_db_object(
|
|
context, my_inst, db_block_device['instance'])
|
|
block_device_obj.instance = my_inst
|
|
|
|
block_device_obj._context = context
|
|
block_device_obj.obj_reset_changes()
|
|
return block_device_obj
|
|
|
|
@base.remotable
|
|
def create(self, context):
|
|
if self.obj_attr_is_set('id'):
|
|
raise exception.ObjectActionError(action='create',
|
|
reason='already created')
|
|
updates = self.obj_get_changes()
|
|
if 'instance' in updates:
|
|
raise exception.ObjectActionError(action='create',
|
|
reason='instance assigned')
|
|
updates.pop('id', None)
|
|
|
|
db_bdm = db.block_device_mapping_create(context, updates, legacy=False)
|
|
cells_api = cells_rpcapi.CellsAPI()
|
|
cells_api.bdm_update_or_create_at_top(context, db_bdm, create=True)
|
|
self._from_db_object(context, self, db_bdm)
|
|
|
|
@base.remotable
|
|
def destroy(self, context):
|
|
if not self.obj_attr_is_set('id'):
|
|
raise exception.ObjectActionError(action='destroy',
|
|
reason='already destroyed')
|
|
db.block_device_mapping_destroy(context, self.id)
|
|
delattr(self, base.get_attrname('id'))
|
|
cells_api = cells_rpcapi.CellsAPI()
|
|
cells_api.bdm_destroy_at_top(context, self.instance_uuid,
|
|
device_name=self.device_name,
|
|
volume_id=self.volume_id)
|
|
|
|
@base.remotable
|
|
def save(self, context):
|
|
updates = self.obj_get_changes()
|
|
if 'instance' in updates:
|
|
raise exception.ObjectActionError(action='save',
|
|
reason='instance changed')
|
|
updates.pop('id', None)
|
|
updated = db.block_device_mapping_update(self._context, self.id,
|
|
updates, legacy=False)
|
|
cells_api = cells_rpcapi.CellsAPI()
|
|
cells_api.bdm_update_or_create_at_top(context, updated)
|
|
self._from_db_object(context, self, updated)
|
|
|
|
@base.remotable_classmethod
|
|
def get_by_volume_id(cls, context, volume_id, expected_attrs=None):
|
|
if expected_attrs is None:
|
|
expected_attrs = []
|
|
db_bdm = db.block_device_mapping_get_by_volume_id(
|
|
context, volume_id, _expected_cols(expected_attrs))
|
|
if not db_bdm:
|
|
raise exception.VolumeBDMNotFound(volume_id=volume_id)
|
|
return cls._from_db_object(context, cls(), db_bdm,
|
|
expected_attrs=expected_attrs)
|
|
|
|
@property
|
|
def is_root(self):
|
|
return self.boot_index == 0
|
|
|
|
@property
|
|
def is_volume(self):
|
|
return self.destination_type == 'volume'
|
|
|
|
@property
|
|
def is_image(self):
|
|
return self.source_type == 'image'
|
|
|
|
def get_image_mapping(self):
|
|
return block_device.BlockDeviceDict(self).get_image_mapping()
|
|
|
|
def obj_load_attr(self, attrname):
|
|
if attrname not in BLOCK_DEVICE_OPTIONAL_ATTRS:
|
|
raise exception.ObjectActionError(
|
|
action='obj_load_attr',
|
|
reason='attribute %s not lazy-loadable' % attrname)
|
|
if not self._context:
|
|
raise exception.OrphanedObjectError(method='obj_load_attr',
|
|
objtype=self.obj_name())
|
|
|
|
LOG.debug(_("Lazy-loading `%(attr)s' on %(name)s uuid %(uuid)s"),
|
|
{'attr': attrname,
|
|
'name': self.obj_name(),
|
|
'uuid': self.uuid,
|
|
})
|
|
self.instance = instance.Instance.get_by_uuid(self._context,
|
|
self.instance_uuid)
|
|
self.obj_reset_changes(fields=['instance'])
|
|
|
|
|
|
class BlockDeviceMappingList(base.ObjectListBase, base.NovaObject):
|
|
# Version 1.0: Initial version
|
|
VERSION = '1.0'
|
|
|
|
fields = {
|
|
'objects': fields.ListOfObjectsField('BlockDeviceMapping'),
|
|
}
|
|
child_versions = {
|
|
'1.0': '1.0',
|
|
}
|
|
|
|
@base.remotable_classmethod
|
|
def get_by_instance_uuid(cls, context, instance_uuid):
|
|
db_bdms = db.block_device_mapping_get_all_by_instance(
|
|
context, instance_uuid)
|
|
return base.obj_make_list(
|
|
context, cls(), BlockDeviceMapping, db_bdms or [])
|
|
|
|
def root_bdm(self):
|
|
try:
|
|
return (bdm_obj for bdm_obj in self if bdm_obj.is_root).next()
|
|
except StopIteration:
|
|
return
|
|
|
|
def root_metadata(self, context, image_api, volume_api):
|
|
root_bdm = self.root_bdm()
|
|
if not root_bdm:
|
|
return {}
|
|
|
|
if root_bdm.is_volume:
|
|
try:
|
|
volume = volume_api.get(context, root_bdm.volume_id)
|
|
return volume.get('volume_image_metadata', {})
|
|
except Exception:
|
|
raise exception.InvalidBDMVolume(id=root_bdm.id)
|
|
elif root_bdm.is_image:
|
|
try:
|
|
image_meta = image_api.show(context, root_bdm.image_id)
|
|
return image_meta.get('properties', {})
|
|
except Exception:
|
|
raise exception.InvalidBDMImage(id=root_bdm.id)
|
|
else:
|
|
return {}
|
|
|
|
|
|
def block_device_make_list(context, db_list, **extra_args):
|
|
return base.obj_make_list(context, BlockDeviceMappingList(),
|
|
BlockDeviceMapping, db_list, **extra_args)
|