237 lines
8.7 KiB
Python
237 lines
8.7 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.
|
|
|
|
import json
|
|
import pecan
|
|
from six.moves import http_client
|
|
import wsme
|
|
from wsme import types as wtypes
|
|
|
|
from cyborg.agent.rpcapi import AgentAPI
|
|
from cyborg.api.controllers import base
|
|
from cyborg.api.controllers import link
|
|
from cyborg.api.controllers.v1 import types
|
|
from cyborg.api import expose
|
|
from cyborg.common import policy
|
|
from cyborg import objects
|
|
|
|
|
|
class Deployable(base.APIBase):
|
|
"""API representation of a deployable.
|
|
|
|
This class enforces type checking and value constraints, and converts
|
|
between the internal object model and the API representation of
|
|
a deployable.
|
|
"""
|
|
|
|
uuid = types.uuid
|
|
"""The UUID of the deployable"""
|
|
|
|
parent_id = types.integer
|
|
"""The parent ID of the deployable"""
|
|
|
|
root_id = types.integer
|
|
"""The root ID of the deployable"""
|
|
|
|
name = wtypes.text
|
|
"""The name of the deployable"""
|
|
|
|
num_accelerators = types.integer
|
|
"""The number of accelerators of the deployable"""
|
|
|
|
device_id = types.integer
|
|
"""The device on which the deployable is located"""
|
|
|
|
attributes_list = wtypes.text
|
|
"""The json list of attributes of the deployable"""
|
|
|
|
links = wsme.wsattr([link.Link], readonly=True)
|
|
"""A list containing a self link"""
|
|
|
|
def __init__(self, **kwargs):
|
|
super(Deployable, self).__init__(**kwargs)
|
|
self.fields = []
|
|
for field in objects.Deployable.fields:
|
|
self.fields.append(field)
|
|
setattr(self, field, kwargs.get(field, wtypes.Unset))
|
|
|
|
@classmethod
|
|
def convert_with_links(cls, obj_dep):
|
|
api_dep = cls(**obj_dep.as_dict())
|
|
url = pecan.request.public_url
|
|
api_dep.links = [
|
|
link.Link.make_link('self', url, 'deployables', api_dep.uuid),
|
|
link.Link.make_link('bookmark', url, 'deployables', api_dep.uuid,
|
|
bookmark=True)
|
|
]
|
|
query = {"deployable_id": obj_dep.id}
|
|
attr_get_list = objects.Attribute.get_by_filter(pecan.request.context,
|
|
query)
|
|
attributes_list = []
|
|
for exist_attr in attr_get_list:
|
|
attributes_list.append({exist_attr.key: exist_attr.value})
|
|
api_dep.attributes_list = json.dumps(attributes_list)
|
|
return api_dep
|
|
|
|
|
|
class DeployableCollection(base.APIBase):
|
|
"""API representation of a collection of deployables."""
|
|
|
|
deployables = [Deployable]
|
|
"""A list containing deployable objects"""
|
|
|
|
@classmethod
|
|
def convert_with_links(cls, obj_deps):
|
|
collection = cls()
|
|
collection.deployables = [Deployable.convert_with_links(obj_dep)
|
|
for obj_dep in obj_deps]
|
|
return collection
|
|
|
|
|
|
class DeployablePatchType(types.JsonPatchType):
|
|
|
|
_api_base = Deployable
|
|
|
|
@staticmethod
|
|
def internal_attrs():
|
|
defaults = types.JsonPatchType.internal_attrs()
|
|
return defaults + ['/name', '/num_accelerators']
|
|
|
|
|
|
class DeployablesController(base.CyborgController):
|
|
"""REST controller for Deployables."""
|
|
|
|
_custom_actions = {'program': ['PATCH']}
|
|
|
|
@policy.authorize_wsgi("cyborg:deployable", "program", False)
|
|
@expose.expose(Deployable, types.uuid, body=[DeployablePatchType])
|
|
def program(self, uuid, program_info):
|
|
"""Program a new deployable(FPGA).
|
|
|
|
:param uuid: The uuid of the target deployable.
|
|
:param program_info: JSON string containing what to program.
|
|
"""
|
|
|
|
image_uuid = program_info[0]['value'][0]['image_uuid']
|
|
agent_api = AgentAPI()
|
|
obj_dep = objects.Deployable.get(pecan.request.context, uuid)
|
|
# Set attribute of the new bitstream/image information
|
|
obj_dep.add_attribute(pecan.request.context, 'image_uuid', image_uuid)
|
|
# TODO (Li Liu) Trigger the program api in Agnet.
|
|
agent_api.program_fpga_with_bitstream(pecan.request.context,
|
|
uuid,
|
|
image_uuid)
|
|
return Deployable.convert_with_links(obj_dep)
|
|
|
|
@policy.authorize_wsgi("cyborg:deployable", "create", False)
|
|
@expose.expose(Deployable, body=types.jsontype,
|
|
status_code=http_client.CREATED)
|
|
def post(self, dep):
|
|
"""Create a new deployable.
|
|
|
|
:param dep: a deployable within the request body.
|
|
"""
|
|
context = pecan.request.context
|
|
obj_dep = objects.Deployable(context, **dep)
|
|
new_dep = pecan.request.conductor_api.deployable_create(context,
|
|
obj_dep)
|
|
# Set the HTTP Location Header
|
|
pecan.response.location = link.build_url('deployables', new_dep.uuid)
|
|
return Deployable.convert_with_links(new_dep)
|
|
|
|
@policy.authorize_wsgi("cyborg:deployable", "get_one")
|
|
@expose.expose(Deployable, types.uuid)
|
|
def get_one(self, uuid):
|
|
"""Retrieve information about the given deployable.
|
|
|
|
:param uuid: UUID of a deployable.
|
|
"""
|
|
|
|
obj_dep = objects.Deployable.get(pecan.request.context, uuid)
|
|
return Deployable.convert_with_links(obj_dep)
|
|
|
|
@policy.authorize_wsgi("cyborg:deployable", "get_all")
|
|
@expose.expose(DeployableCollection, int, types.uuid, wtypes.text,
|
|
wtypes.text, wtypes.ArrayType(types.FilterType))
|
|
# TODO(wangzhh): Remove limit, marker, sort_key, sort_dir in next release.
|
|
# They are used to compatible with R release client.
|
|
def get_all(self, limit=None, marker=None, sort_key='id', sort_dir='asc',
|
|
filters=None):
|
|
"""Retrieve a list of deployables."""
|
|
filters_dict = {}
|
|
self._generate_filters(limit, sort_key, sort_dir, filters_dict)
|
|
if filters:
|
|
for filter in filters:
|
|
filters_dict.update(filter.as_dict())
|
|
context = pecan.request.context
|
|
if marker:
|
|
marker_obj = objects.Deployable.get(context, marker)
|
|
filters_dict["marker_obj"] = marker_obj
|
|
obj_deps = objects.Deployable.list(context, filters=filters_dict)
|
|
return DeployableCollection.convert_with_links(obj_deps)
|
|
|
|
def _generate_filters(self, limit, sort_key, sort_dir, filters_dict):
|
|
"""This method are used to compatible with R release client."""
|
|
if limit:
|
|
filters_dict["limit"] = limit
|
|
if sort_key:
|
|
filters_dict["sort_key"] = sort_key
|
|
if sort_dir:
|
|
filters_dict["sort_dir"] = sort_dir
|
|
|
|
@policy.authorize_wsgi("cyborg:deployable", "update")
|
|
@expose.expose(Deployable, types.uuid, body=[DeployablePatchType])
|
|
def patch(self, uuid, patch):
|
|
"""Update a deployable.
|
|
|
|
Usage: curl -X PATCH {ip}:{port}/v1/accelerators/deployables/
|
|
{deployable_uuid} -d '[{"path":"/instance_uuid","value":
|
|
{instance_uuid}, "op":"replace"}]' -H "Content-type:
|
|
application/json"
|
|
|
|
:param uuid: UUID of a deployable.
|
|
:param patch: a json PATCH document to apply to this deployable.
|
|
"""
|
|
context = pecan.request.context
|
|
|
|
obj_dep = objects.Deployable.get(context, uuid)
|
|
|
|
# Update only the fields that have changed
|
|
for field in objects.Deployable.fields:
|
|
try:
|
|
patch_val = getattr(obj_dep, field)
|
|
except AttributeError:
|
|
# Ignore fields that aren't exposed in the API
|
|
continue
|
|
if patch_val == wtypes.Unset:
|
|
patch_val = None
|
|
if obj_dep[field] != patch_val:
|
|
obj_dep[field] = patch_val
|
|
|
|
new_dep = pecan.request.conductor_api.deployable_update(context,
|
|
obj_dep)
|
|
return Deployable.convert_with_links(new_dep)
|
|
|
|
@policy.authorize_wsgi("cyborg:deployable", "delete")
|
|
@expose.expose(None, types.uuid, status_code=http_client.NO_CONTENT)
|
|
def delete(self, uuid):
|
|
"""Delete a deployable.
|
|
|
|
:param uuid: UUID of a deployable.
|
|
"""
|
|
context = pecan.request.context
|
|
obj_dep = objects.Deployable.get(context, uuid)
|
|
pecan.request.conductor_api.deployable_delete(context, obj_dep)
|