¡@
OpenStack Study: pci.py**** CubicPower OpenStack Study ****
# Copyright 2013 Intel Corporation # 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. import webob.exc from nova.api.openstack import extensions from nova.api.openstack import wsgi from nova import compute from nova import exception from nova.objects import pci_device from nova.openstack.common import jsonutils ALIAS = 'os-pci' instance_authorize = extensions.soft_extension_authorizer( 'compute', 'v3:' + ALIAS + ':pci_servers') authorize = extensions.extension_authorizer('compute', 'v3:' + ALIAS) PCI_ADMIN_KEYS = ['id', 'address', 'vendor_id', 'product_id', 'status', 'compute_node_id'] PCI_DETAIL_KEYS = ['dev_type', 'label', 'instance_uuid', 'dev_id', 'extra_info']
**** CubicPower OpenStack Study **** class PciServerController(wsgi.Controller):
**** CubicPower OpenStack Study **** def _extend_server(self, server, instance): dev_id = [] for dev in instance.pci_devices: dev_id.append({'id': dev['id']}) server['%s:pci_devices' % Pci.alias] = dev_id @wsgi.extends
**** CubicPower OpenStack Study **** def show(self, req, resp_obj, id): context = req.environ['nova.context'] if instance_authorize(context): server = resp_obj.obj['server'] instance = req.get_db_instance(server['id']) self._extend_server(server, instance) @wsgi.extends
**** CubicPower OpenStack Study **** def detail(self, req, resp_obj): context = req.environ['nova.context'] if instance_authorize(context): servers = list(resp_obj.obj['servers']) for server in servers: instance = req.get_db_instance(server['id']) self._extend_server(server, instance)
**** CubicPower OpenStack Study **** class PciHypervisorController(wsgi.Controller):
**** CubicPower OpenStack Study **** def _extend_hypervisor(self, hypervisor, compute_node): hypervisor['%s:pci_stats' % Pci.alias] = jsonutils.loads( compute_node['pci_stats']) @wsgi.extends
**** CubicPower OpenStack Study **** def show(self, req, resp_obj, id): hypervisor = resp_obj.obj['hypervisor'] compute_node = req.get_db_compute_node(hypervisor['id']) self._extend_hypervisor(hypervisor, compute_node) @wsgi.extends
**** CubicPower OpenStack Study **** def detail(self, req, resp_obj): hypervisors = list(resp_obj.obj['hypervisors']) for hypervisor in hypervisors: compute_node = req.get_db_compute_node(hypervisor['id']) self._extend_hypervisor(hypervisor, compute_node)
**** CubicPower OpenStack Study **** class PciController(object):
**** CubicPower OpenStack Study **** def __init__(self): self.host_api = compute.HostAPI()
**** CubicPower OpenStack Study **** def _view_pcidevice(self, device, detail=False): dev_dict = {} for key in PCI_ADMIN_KEYS: dev_dict[key] = device[key] if detail: for field in PCI_DETAIL_KEYS: if field == 'instance_uuid': dev_dict['server_uuid'] = device[field] else: dev_dict[field] = device[field] return dev_dict
**** CubicPower OpenStack Study **** def _get_all_nodes_pci_devices(self, req, detail, action): context = req.environ['nova.context'] authorize(context, action=action) compute_nodes = self.host_api.compute_node_get_all(context) results = [] for node in compute_nodes: pci_devs = pci_device.PciDeviceList.get_by_compute_node( context, node['id']) results.extend([self._view_pcidevice(dev, detail) for dev in pci_devs]) return results @extensions.expected_errors(())
**** CubicPower OpenStack Study **** def detail(self, req): results = self._get_all_nodes_pci_devices(req, True, 'detail') return dict(pci_devices=results) @extensions.expected_errors(404)
**** CubicPower OpenStack Study **** def show(self, req, id): context = req.environ['nova.context'] authorize(context, action='show') try: pci_dev = pci_device.PciDevice.get_by_dev_id(context, id) except exception.PciDeviceNotFoundById as e: raise webob.exc.HTTPNotFound(explanation=e.format_message()) result = self._view_pcidevice(pci_dev, True) return dict(pci_device=result) @extensions.expected_errors(())
**** CubicPower OpenStack Study **** def index(self, req): results = self._get_all_nodes_pci_devices(req, False, 'index') return dict(pci_devices=results)
**** CubicPower OpenStack Study **** class Pci(extensions.V3APIExtensionBase): """Pci access support.""" name = "PCIAccess" alias = ALIAS version = 1
**** CubicPower OpenStack Study **** def get_resources(self): resources = [extensions.ResourceExtension(ALIAS, PciController(), collection_actions={'detail': 'GET'})] return resources
**** CubicPower OpenStack Study **** def get_controller_extensions(self): server_extension = extensions.ControllerExtension( self, 'servers', PciServerController()) compute_extension = extensions.ControllerExtension( self, 'os-hypervisors', PciHypervisorController()) return [server_extension, compute_extension]
|