bada59703f
Remove extra parameters of function and add ut for rpcapi Implements: blueprint add-ut-for-rpcapi Change-Id: I812d1ede07bf2ab4f4567121efb2ed3d757fcae7
196 lines
8.8 KiB
Python
196 lines
8.8 KiB
Python
# 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.
|
|
|
|
"""Handles all requests relating to compute resources (e.g. containers,
|
|
networking and storage of containers, and compute hosts on which they run)."""
|
|
|
|
from zun.common import consts
|
|
from zun.common import exception
|
|
from zun.common import profiler
|
|
from zun.compute import container_actions
|
|
from zun.compute import rpcapi
|
|
import zun.conf
|
|
from zun import objects
|
|
from zun.scheduler import client as scheduler_client
|
|
|
|
CONF = zun.conf.CONF
|
|
|
|
|
|
@profiler.trace_cls("rpc")
|
|
class API(object):
|
|
"""API for interacting with the compute manager."""
|
|
|
|
def __init__(self, context):
|
|
self.rpcapi = rpcapi.API(context=context)
|
|
self.scheduler_client = scheduler_client.SchedulerClient()
|
|
super(API, self).__init__()
|
|
|
|
def _record_action_start(self, context, container, action):
|
|
objects.ContainerAction.action_start(context, container.uuid,
|
|
action, want_result=False)
|
|
|
|
def container_create(self, context, new_container, extra_spec,
|
|
requested_networks, requested_volumes, run,
|
|
pci_requests=None):
|
|
try:
|
|
host_state = self._schedule_container(context, new_container,
|
|
extra_spec)
|
|
except Exception as exc:
|
|
new_container.status = consts.ERROR
|
|
new_container.status_reason = str(exc)
|
|
new_container.save(context)
|
|
return
|
|
|
|
# NOTE(mkrai): Intent here is to check the existence of image
|
|
# before proceeding to create container. If image is not found,
|
|
# container create will fail with 400 status.
|
|
if CONF.api.enable_image_validation:
|
|
try:
|
|
images = self.rpcapi.image_search(
|
|
context, new_container.image,
|
|
new_container.image_driver, True, host_state['host'])
|
|
if not images:
|
|
raise exception.ImageNotFound(image=new_container.image)
|
|
except Exception as e:
|
|
new_container.status = consts.ERROR
|
|
new_container.status_reason = str(e)
|
|
new_container.save(context)
|
|
raise
|
|
|
|
self._record_action_start(context, new_container,
|
|
container_actions.CREATE)
|
|
self.rpcapi.container_create(context, host_state['host'],
|
|
new_container, host_state['limits'],
|
|
requested_networks, requested_volumes,
|
|
run, pci_requests)
|
|
|
|
def _schedule_container(self, context, new_container, extra_spec):
|
|
dests = self.scheduler_client.select_destinations(context,
|
|
[new_container],
|
|
extra_spec)
|
|
return dests[0]
|
|
|
|
def container_delete(self, context, container, *args):
|
|
self._record_action_start(context, container, container_actions.DELETE)
|
|
return self.rpcapi.container_delete(context, container, *args)
|
|
|
|
def container_show(self, context, container):
|
|
return self.rpcapi.container_show(context, container)
|
|
|
|
def container_reboot(self, context, container, *args):
|
|
self._record_action_start(context, container, container_actions.REBOOT)
|
|
return self.rpcapi.container_reboot(context, container, *args)
|
|
|
|
def container_stop(self, context, container, *args):
|
|
self._record_action_start(context, container, container_actions.STOP)
|
|
return self.rpcapi.container_stop(context, container, *args)
|
|
|
|
def container_start(self, context, container):
|
|
self._record_action_start(context, container, container_actions.START)
|
|
return self.rpcapi.container_start(context, container)
|
|
|
|
def container_pause(self, context, container):
|
|
self._record_action_start(context, container, container_actions.PAUSE)
|
|
return self.rpcapi.container_pause(context, container)
|
|
|
|
def container_unpause(self, context, container):
|
|
self._record_action_start(context, container,
|
|
container_actions.UNPAUSE)
|
|
return self.rpcapi.container_unpause(context, container)
|
|
|
|
def container_logs(self, context, container, stdout, stderr,
|
|
timestamps, tail, since):
|
|
return self.rpcapi.container_logs(context, container, stdout, stderr,
|
|
timestamps, tail, since)
|
|
|
|
def container_exec(self, context, container, *args):
|
|
return self.rpcapi.container_exec(context, container, *args)
|
|
|
|
def container_exec_resize(self, context, container, *args):
|
|
return self.rpcapi.container_exec_resize(context, container, *args)
|
|
|
|
def container_kill(self, context, container, *args):
|
|
self._record_action_start(context, container, container_actions.KILL)
|
|
return self.rpcapi.container_kill(context, container, *args)
|
|
|
|
def container_update(self, context, container, *args):
|
|
return self.rpcapi.container_update(context, container, *args)
|
|
|
|
def container_attach(self, context, container):
|
|
return self.rpcapi.container_attach(context, container)
|
|
|
|
def container_resize(self, context, container, *args):
|
|
return self.rpcapi.container_resize(context, container, *args)
|
|
|
|
def container_top(self, context, container, *args):
|
|
return self.rpcapi.container_top(context, container, *args)
|
|
|
|
def container_get_archive(self, context, container, *args):
|
|
return self.rpcapi.container_get_archive(context, container, *args)
|
|
|
|
def add_security_group(self, context, container, *args):
|
|
self._record_action_start(context, container,
|
|
container_actions.ADD_SECURITY_GROUP)
|
|
return self.rpcapi.add_security_group(context, container, *args)
|
|
|
|
def remove_security_group(self, context, container, *args):
|
|
self._record_action_start(context, container,
|
|
container_actions.REMOVE_SECURITY_GROUP)
|
|
return self.rpcapi.remove_security_group(context, container, *args)
|
|
|
|
def container_put_archive(self, context, container, *args):
|
|
return self.rpcapi.container_put_archive(context, container, *args)
|
|
|
|
def container_stats(self, context, container):
|
|
return self.rpcapi.container_stats(context, container)
|
|
|
|
def container_commit(self, context, container, *args):
|
|
self._record_action_start(context, container, container_actions.COMMIT)
|
|
return self.rpcapi.container_commit(context, container, *args)
|
|
|
|
def image_pull(self, context, image):
|
|
return self.rpcapi.image_pull(context, image)
|
|
|
|
def image_search(self, context, image, image_driver, exact_match, *args):
|
|
return self.rpcapi.image_search(context, image, image_driver,
|
|
exact_match, *args)
|
|
|
|
def capsule_create(self, context, new_capsule, requested_networks=None,
|
|
requested_volumes=None, extra_spec=None):
|
|
try:
|
|
host_state = self._schedule_container(context, new_capsule,
|
|
extra_spec)
|
|
except Exception as exc:
|
|
new_capsule.status = consts.FAILED
|
|
new_capsule.status_reason = str(exc)
|
|
new_capsule.save(context)
|
|
return
|
|
for container in new_capsule.containers:
|
|
self._record_action_start(context, container,
|
|
container_actions.CREATE)
|
|
self.rpcapi.capsule_create(context, host_state['host'], new_capsule,
|
|
requested_networks, requested_volumes,
|
|
host_state['limits'])
|
|
|
|
def capsule_delete(self, context, capsule):
|
|
return self.rpcapi.capsule_delete(context, capsule)
|
|
|
|
def network_detach(self, context, container, *args):
|
|
self._record_action_start(context, container,
|
|
container_actions.NETWORK_DETACH)
|
|
return self.rpcapi.network_detach(context, container, *args)
|
|
|
|
def network_attach(self, context, container, *args):
|
|
self._record_action_start(context, container,
|
|
container_actions.NETWORK_ATTACH)
|
|
return self.rpcapi.network_attach(context, container, *args)
|