You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

102 lines
3.1 KiB

# Copyright 2015 - Alcatel-Lucent
# Copyright 2016 - Nokia
# 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
# 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_config import cfg
from oslo_log import log
import oslo_messaging as messaging
from oslo_messaging.rpc import dispatcher
from osprofiler import profiler
OPTS = [
help='The topic vitrage listens on'),
LOG = log.getLogger(__name__)
class ProfilerContextSerializer(messaging.Serializer):
def __init__(self, serializer):
self._serializer = serializer
def serialize_entity(self, context, entity):
if not self._serializer:
return entity
return self._serializer.serialize_entity(context, entity)
def deserialize_entity(self, context, entity):
if not self._serializer:
return entity
return self._serializer.deserialize_entity(context, entity)
def serialize_context(self, context):
ctx = self._serializer.serialize_context(context) \
if self._serializer else context
pfr = profiler.get()
if pfr:
ctx['trace_info'] = {
"hmac_key": pfr.hmac_key,
"base_id": pfr.get_base_id(),
"parent_id": pfr.get_id()
return ctx
def deserialize_context(self, context):
trace_info = context.pop('trace_info', None)
if trace_info:
return self._serializer.deserialize_context(context)\
if self._serializer else context
def set_defaults(control_exchange):
def get_client(transport, target, version_cap=None, serializer=None):
assert transport is not None
if profiler:'profiler enabled for RPC client')
serializer = ProfilerContextSerializer(serializer=serializer)
return messaging.RPCClient(transport,
def get_server(target, endpoints, transport, serializer=None):
assert transport is not None
if profiler:
LOG.debug('profiler enabled for RPC server')
serializer = ProfilerContextSerializer(serializer=serializer)
access_policy = dispatcher.DefaultRPCAccessPolicy
return messaging.get_rpc_server(transport,