08dd23d1d4
Since the change I643a111fca8bac32f41ced232d54ff2a2ebcbf77 we don't need proxy for direct types because any message woud be sent before server listening to the target appears on name server registry, so DEALER wouldn't block. Change-Id: I3c0f3e6930a4092cac5a6e18529d98e6d6e65f32
136 lines
3.9 KiB
Python
136 lines
3.9 KiB
Python
# Copyright 2015 Mirantis, 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.
|
|
|
|
import abc
|
|
import logging
|
|
import uuid
|
|
|
|
import six
|
|
|
|
from oslo_messaging._drivers.zmq_driver import zmq_async
|
|
from oslo_messaging._drivers.zmq_driver import zmq_names
|
|
from oslo_messaging._i18n import _LE
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
zmq = zmq_async.import_zmq()
|
|
|
|
|
|
@six.add_metaclass(abc.ABCMeta)
|
|
class Request(object):
|
|
|
|
"""Zmq request abstract class
|
|
|
|
Represents socket (publisher) independent data object to publish.
|
|
Request object should contain all needed information for a publisher
|
|
to publish it, for instance: message payload, target, timeout
|
|
and retries etc.
|
|
"""
|
|
|
|
def __init__(self, target, context=None, message=None, retry=None):
|
|
|
|
"""Construct request object
|
|
|
|
:param target: Message destination target
|
|
:type target: oslo_messaging.Target
|
|
:param context: Message context
|
|
:type context: dict
|
|
:param message: Message payload to pass
|
|
:type message: dict
|
|
:param retry: an optional default connection retries configuration
|
|
None or -1 means to retry forever
|
|
0 means no retry
|
|
N means N retries
|
|
:type retry: int
|
|
"""
|
|
|
|
if self.msg_type not in zmq_names.MESSAGE_TYPES:
|
|
raise RuntimeError("Unknown message type!")
|
|
|
|
self.target = target
|
|
self.context = context
|
|
self.message = message
|
|
|
|
self.retry = retry
|
|
if not isinstance(retry, int) and retry is not None:
|
|
raise ValueError(
|
|
"retry must be an integer, not {0}".format(type(retry)))
|
|
|
|
self.message_id = str(uuid.uuid1())
|
|
|
|
def create_envelope(self):
|
|
return {'msg_type': self.msg_type,
|
|
'message_id': self.message_id,
|
|
'target': self.target}
|
|
|
|
@abc.abstractproperty
|
|
def msg_type(self):
|
|
"""ZMQ message type"""
|
|
|
|
def close(self):
|
|
"""Nothing to close in base request"""
|
|
|
|
|
|
class RpcRequest(Request):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
message = kwargs.get("message")
|
|
if message['method'] is None:
|
|
errmsg = _LE("No method specified for RPC call")
|
|
LOG.error(_LE("No method specified for RPC call"))
|
|
raise KeyError(errmsg)
|
|
|
|
self.timeout = kwargs.pop("timeout")
|
|
assert self.timeout is not None, "Timeout should be specified!"
|
|
|
|
if not isinstance(self.timeout, int) and self.timeout is not None:
|
|
raise ValueError(
|
|
"timeout must be an integer, not {0}"
|
|
.format(type(self.timeout)))
|
|
|
|
super(RpcRequest, self).__init__(*args, **kwargs)
|
|
|
|
def create_envelope(self):
|
|
envelope = super(RpcRequest, self).create_envelope()
|
|
envelope['timeout'] = self.timeout
|
|
return envelope
|
|
|
|
|
|
class CallRequest(RpcRequest):
|
|
|
|
msg_type = zmq_names.CALL_TYPE
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
self.allowed_remote_exmods = kwargs.pop("allowed_remote_exmods")
|
|
super(CallRequest, self).__init__(*args, **kwargs)
|
|
|
|
|
|
class CastRequest(RpcRequest):
|
|
|
|
msg_type = zmq_names.CAST_TYPE
|
|
|
|
|
|
class FanoutRequest(RpcRequest):
|
|
|
|
msg_type = zmq_names.CAST_FANOUT_TYPE
|
|
|
|
|
|
class NotificationRequest(Request):
|
|
|
|
msg_type = zmq_names.NOTIFY_TYPE
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
self.version = kwargs.pop("version")
|
|
super(NotificationRequest, self).__init__(*args, **kwargs)
|