140 lines
5.6 KiB
Python
140 lines
5.6 KiB
Python
# Copyright 2018 Huawei Technologies Co.,LTD.
|
|
# 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 as logging
|
|
from oslo_versionedobjects import base as object_base
|
|
|
|
from cyborg.common import exception
|
|
from cyborg.db import api as dbapi
|
|
from cyborg.objects import base
|
|
from cyborg.objects import fields as object_fields
|
|
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
@base.CyborgObjectRegistry.register
|
|
class Deployable(base.CyborgObject, object_base.VersionedObjectDictCompat):
|
|
# Version 1.0: Initial version
|
|
VERSION = '1.0'
|
|
|
|
dbapi = dbapi.get_instance()
|
|
attributes_list = []
|
|
|
|
fields = {
|
|
'id': object_fields.IntegerField(nullable=False),
|
|
'uuid': object_fields.UUIDField(nullable=False),
|
|
'name': object_fields.StringField(nullable=False),
|
|
'parent_uuid': object_fields.UUIDField(nullable=True),
|
|
# parent_uuid refers to the id of the VF's parent node
|
|
'root_uuid': object_fields.UUIDField(nullable=True),
|
|
# root_uuid refers to the id of the VF's root which has to be a PF
|
|
'pcie_address': object_fields.StringField(nullable=False),
|
|
'host': object_fields.StringField(nullable=False),
|
|
'board': object_fields.StringField(nullable=False),
|
|
# board refers to a specific acc board type, e.g P100 GPU card
|
|
'vendor': object_fields.StringField(nullable=False),
|
|
'version': object_fields.StringField(nullable=False),
|
|
'type': object_fields.StringField(nullable=False),
|
|
# similar to the acc_type in accelerator.py
|
|
'assignable': object_fields.BooleanField(nullable=False),
|
|
# identify if an accelerator is in use
|
|
'instance_uuid': object_fields.UUIDField(nullable=True),
|
|
# The id of the virtualized accelerator instance
|
|
'availability': object_fields.StringField(nullable=False),
|
|
# identify the state of acc, e.g released/claimed/...
|
|
'accelerator_id': object_fields.IntegerField(nullable=False)
|
|
# Foreign key constrain to reference accelerator table
|
|
}
|
|
|
|
def _get_parent_root_uuid(self):
|
|
obj_dep = Deployable.get(None, self.parent_uuid)
|
|
return obj_dep.root_uuid
|
|
|
|
def create(self, context):
|
|
"""Create a Deployable record in the DB."""
|
|
if 'uuid' not in self:
|
|
raise exception.ObjectActionError(action='create',
|
|
reason='uuid is required')
|
|
|
|
if self.parent_uuid is None:
|
|
self.root_uuid = self.uuid
|
|
else:
|
|
self.root_uuid = self._get_parent_root_uuid()
|
|
|
|
values = self.obj_get_changes()
|
|
|
|
db_dep = self.dbapi.deployable_create(context, values)
|
|
self._from_db_object(self, db_dep)
|
|
|
|
@classmethod
|
|
def get(cls, context, uuid):
|
|
"""Find a DB Deployable and return an Obj Deployable."""
|
|
db_dep = cls.dbapi.deployable_get(context, uuid)
|
|
obj_dep = cls._from_db_object(cls(context), db_dep)
|
|
return obj_dep
|
|
|
|
@classmethod
|
|
def get_by_host(cls, context, host):
|
|
"""Get a Deployable by host."""
|
|
db_deps = cls.dbapi.deployable_get_by_host(context, host)
|
|
return cls._from_db_object_list(db_deps, context)
|
|
|
|
@classmethod
|
|
def list(cls, context):
|
|
"""Return a list of Deployable objects."""
|
|
db_deps = cls.dbapi.deployable_list(context)
|
|
return cls._from_db_object_list(db_deps, context)
|
|
|
|
def save(self, context):
|
|
"""Update a Deployable record in the DB."""
|
|
updates = self.obj_get_changes()
|
|
db_dep = self.dbapi.deployable_update(context, self.uuid, updates)
|
|
self._from_db_object(self, db_dep)
|
|
|
|
def destroy(self, context):
|
|
"""Delete a Deployable from the DB."""
|
|
self.dbapi.deployable_delete(context, self.uuid)
|
|
self.obj_reset_changes()
|
|
|
|
def add_attribute(self, attribute):
|
|
"""add a attribute object to the attribute_list.
|
|
If the attribute already exists, it will update the value,
|
|
otherwise, the vf will be appended to the list
|
|
"""
|
|
if not isinstance(vf, VirtualFunction) or vf.type != 'vf':
|
|
raise exception.InvalidDeployType()
|
|
for exist_vf in self.virtual_function_list:
|
|
if base.obj_equal_prims(vf, exist_vf):
|
|
LOG.warning("The vf already exists")
|
|
return None
|
|
|
|
@classmethod
|
|
def get_by_filter(cls, context,
|
|
filters, sort_key='created_at',
|
|
sort_dir='desc', limit=None,
|
|
marker=None, join=None):
|
|
obj_dpl_list = []
|
|
db_dpl_list = cls.dbapi.deployable_get_by_filters(context, filters,
|
|
sort_key=sort_key,
|
|
sort_dir=sort_dir,
|
|
limit=limit,
|
|
marker=marker,
|
|
join_columns=join)
|
|
for db_dpl in db_dpl_list:
|
|
obj_dpl = cls._from_db_object(cls(context), db_dpl)
|
|
obj_dpl_list.append(obj_dpl)
|
|
return obj_dpl_list
|