|
- # Copyright 2017 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_concurrency import processutils
- from cyborg import context
- from oslo_log import log
- import oslo_messaging as messaging
- from oslo_service import service
- from oslo_service import wsgi
- from oslo_utils import importutils
-
- from cyborg.api import app
- from cyborg.common import config
- from cyborg.common import exception
- from cyborg.common.i18n import _
- from cyborg.common import rpc
- from cyborg.conf import CONF
- from cyborg import objects
- from cyborg.objects import base as objects_base
-
-
- LOG = log.getLogger(__name__)
-
-
- class RPCService(service.Service):
- def __init__(self, manager_module, manager_class, topic, host=None):
- super(RPCService, self).__init__()
- self.topic = topic
- self.host = host or CONF.host
- manager_module = importutils.try_import(manager_module)
- manager_class = getattr(manager_module, manager_class)
- self.manager = manager_class(self.topic, self.host)
- self.rpcserver = None
-
- def start(self):
- super(RPCService, self).start()
-
- target = messaging.Target(topic=self.topic, server=self.host)
- endpoints = [self.manager]
- serializer = objects_base.CyborgObjectSerializer()
- self.rpcserver = rpc.get_server(target, endpoints, serializer)
- self.rpcserver.start()
-
- admin_context = context.get_admin_context()
- self.tg.add_dynamic_timer(
- self.manager.periodic_tasks,
- periodic_interval_max=CONF.periodic_interval,
- context=admin_context)
-
- LOG.info('Created RPC server for service %(service)s on host '
- '%(host)s.',
- {'service': self.topic, 'host': self.host})
-
- def stop(self, graceful=True):
- try:
- self.rpcserver.stop()
- self.rpcserver.wait()
- except Exception as e:
- LOG.exception('Service error occurred when stopping the '
- 'RPC server. Error: %s', e)
-
- super(RPCService, self).stop(graceful=graceful)
- LOG.info('Stopped RPC server for service %(service)s on host '
- '%(host)s.',
- {'service': self.topic, 'host': self.host})
-
-
- def prepare_service(argv=None):
- log.register_options(CONF)
- log.set_defaults(default_log_levels=CONF.default_log_levels)
-
- argv = argv or []
- config.parse_args(argv)
-
- log.setup(CONF, 'cyborg')
- objects.register_all()
-
-
- def process_launcher():
- return service.ProcessLauncher(CONF, restart_method='mutate')
-
-
- class WSGIService(service.ServiceBase):
- """Provides ability to launch cyborg API from wsgi app."""
-
- def __init__(self, name, use_ssl=False):
- """Initialize, but do not start the WSGI server.
-
- :param name: The name of the WSGI server given to the loader.
- :param use_ssl: Wraps the socket in an SSL context if True.
- :returns: None
- """
- self.name = name
- self.app = app.load_app()
- self.workers = (CONF.api.api_workers or
- processutils.get_worker_count())
- if self.workers and self.workers < 1:
- raise exception.ConfigInvalid(
- _("api_workers value of %d is invalid, "
- "must be greater than 0.") % self.workers)
-
- self.server = wsgi.Server(CONF, self.name, self.app,
- host=CONF.api.host_ip,
- port=CONF.api.port,
- use_ssl=use_ssl)
-
- def start(self):
- """Start serving this service using loaded configuration.
-
- :returns: None
- """
- self.server.start()
-
- def stop(self):
- """Stop serving this API.
-
- :returns: None
- """
- self.server.stop()
-
- def wait(self):
- """Wait for the service to stop serving this API.
-
- :returns: None
- """
- self.server.wait()
-
- def reset(self):
- """Reset server greenpool size to default.
-
- :returns: None
- """
- self.server.reset()
|