¡@

Home 

OpenStack Study: rpc.py

OpenStack Index

**** CubicPower OpenStack Study ****

# Copyright 2013 Red Hat, Inc.

#

# 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.

__all__ = [

'init',

'cleanup',

'set_defaults',

'add_extra_exmods',

'clear_extra_exmods',

'get_allowed_exmods',

'RequestContextSerializer',

'get_client',

'get_server',

'get_notifier',

'TRANSPORT_ALIASES',

]

from oslo.config import cfg

from oslo import messaging

import nova.context

import nova.exception

from nova.openstack.common import jsonutils

CONF = cfg.CONF

TRANSPORT = None

NOTIFIER = None

ALLOWED_EXMODS = [

nova.exception.__name__,

]

EXTRA_EXMODS = []

# NOTE(markmc): The nova.openstack.common.rpc entries are for backwards compat

# with Havana rpc_backend configuration values. The nova.rpc entries are for

# compat with Essex values.

TRANSPORT_ALIASES = {

'nova.openstack.common.rpc.impl_kombu': 'rabbit',

'nova.openstack.common.rpc.impl_qpid': 'qpid',

'nova.openstack.common.rpc.impl_zmq': 'zmq',

'nova.rpc.impl_kombu': 'rabbit',

'nova.rpc.impl_qpid': 'qpid',

'nova.rpc.impl_zmq': 'zmq',

}

**** CubicPower OpenStack Study ****

def init(conf):

    global TRANSPORT, NOTIFIER

    exmods = get_allowed_exmods()

    TRANSPORT = messaging.get_transport(conf,

                                        allowed_remote_exmods=exmods,

                                        aliases=TRANSPORT_ALIASES)

    serializer = RequestContextSerializer(JsonPayloadSerializer())

    NOTIFIER = messaging.Notifier(TRANSPORT, serializer=serializer)

**** CubicPower OpenStack Study ****

def cleanup():

    global TRANSPORT, NOTIFIER

    assert TRANSPORT is not None

    assert NOTIFIER is not None

    TRANSPORT.cleanup()

    TRANSPORT = NOTIFIER = None

**** CubicPower OpenStack Study ****

def set_defaults(control_exchange):

    messaging.set_transport_defaults(control_exchange)

**** CubicPower OpenStack Study ****

def add_extra_exmods(*args):

    EXTRA_EXMODS.extend(args)

**** CubicPower OpenStack Study ****

def clear_extra_exmods():

    del EXTRA_EXMODS[:]

**** CubicPower OpenStack Study ****

def get_allowed_exmods():

    return ALLOWED_EXMODS + EXTRA_EXMODS

**** CubicPower OpenStack Study ****

class JsonPayloadSerializer(messaging.NoOpSerializer):

@staticmethod

**** CubicPower OpenStack Study ****

    def serialize_entity(context, entity):

        return jsonutils.to_primitive(entity, convert_instances=True)

**** CubicPower OpenStack Study ****

class RequestContextSerializer(messaging.Serializer):

**** CubicPower OpenStack Study ****

    def __init__(self, base):

        self._base = base

**** CubicPower OpenStack Study ****

    def serialize_entity(self, context, entity):

        if not self._base:

            return entity

        return self._base.serialize_entity(context, entity)

**** CubicPower OpenStack Study ****

    def deserialize_entity(self, context, entity):

        if not self._base:

            return entity

        return self._base.deserialize_entity(context, entity)

**** CubicPower OpenStack Study ****

    def serialize_context(self, context):

        return context.to_dict()

**** CubicPower OpenStack Study ****

    def deserialize_context(self, context):

        return nova.context.RequestContext.from_dict(context)

def get_transport_url(url_str=None):

    return messaging.TransportURL.parse(CONF, url_str, TRANSPORT_ALIASES)

def get_client(target, version_cap=None, serializer=None):

    assert TRANSPORT is not None

    serializer = RequestContextSerializer(serializer)

    return messaging.RPCClient(TRANSPORT,

                               target,

                               version_cap=version_cap,

                               serializer=serializer)

def get_server(target, endpoints, serializer=None):

    assert TRANSPORT is not None

    serializer = RequestContextSerializer(serializer)

    return messaging.get_rpc_server(TRANSPORT,

                                    target,

                                    endpoints,

                                    executor='eventlet',

                                    serializer=serializer)

def get_notifier(service=None, host=None, publisher_id=None):

    assert NOTIFIER is not None

    if not publisher_id:

        publisher_id = "%s.%s" % (service, host or CONF.host)

    return NOTIFIER.prepare(publisher_id=publisher_id)

**** CubicPower OpenStack Study ****

def get_transport_url(url_str=None):

    return messaging.TransportURL.parse(CONF, url_str, TRANSPORT_ALIASES)

**** CubicPower OpenStack Study ****

def get_client(target, version_cap=None, serializer=None):

    assert TRANSPORT is not None

    serializer = RequestContextSerializer(serializer)

    return messaging.RPCClient(TRANSPORT,

                               target,

                               version_cap=version_cap,

                               serializer=serializer)

**** CubicPower OpenStack Study ****

def get_server(target, endpoints, serializer=None):

    assert TRANSPORT is not None

    serializer = RequestContextSerializer(serializer)

    return messaging.get_rpc_server(TRANSPORT,

                                    target,

                                    endpoints,

                                    executor='eventlet',

                                    serializer=serializer)

**** CubicPower OpenStack Study ****

def get_notifier(service=None, host=None, publisher_id=None):

    assert NOTIFIER is not None

    if not publisher_id:

        publisher_id = "%s.%s" % (service, host or CONF.host)

    return NOTIFIER.prepare(publisher_id=publisher_id)