75 lines
2.7 KiB
Python
75 lines
2.7 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 logging
|
|
|
|
from oslo_messaging._drivers.zmq_driver import zmq_async
|
|
from oslo_messaging._drivers.zmq_driver import zmq_names
|
|
from oslo_messaging._drivers.zmq_driver import zmq_socket
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
zmq = zmq_async.import_zmq()
|
|
|
|
|
|
class PublisherProxy(object):
|
|
"""PUB/SUB based request publisher
|
|
|
|
The publisher intended to be used for Fanout and Notify
|
|
multi-sending patterns.
|
|
|
|
It differs from direct publishers like DEALER or PUSH based
|
|
in a way it treats matchmaker. Here all publishers register
|
|
in the matchmaker. Subscribers (server-side) take the list
|
|
of publishers and connect to all of them but subscribe
|
|
only to a specific topic-filtering tag generated from the
|
|
Target object.
|
|
"""
|
|
|
|
def __init__(self, conf, matchmaker):
|
|
super(PublisherProxy, self).__init__()
|
|
self.conf = conf
|
|
self.zmq_context = zmq.Context()
|
|
self.matchmaker = matchmaker
|
|
|
|
port = conf.zmq_proxy_opts.publisher_port
|
|
|
|
self.socket = zmq_socket.ZmqFixedPortSocket(
|
|
self.conf, self.zmq_context, zmq.PUB, conf.zmq_proxy_opts.host,
|
|
port) if port != 0 else \
|
|
zmq_socket.ZmqRandomPortSocket(
|
|
self.conf, self.zmq_context, zmq.PUB, conf.zmq_proxy_opts.host)
|
|
|
|
self.host = self.socket.connect_address
|
|
|
|
def send_request(self, multipart_message):
|
|
message_type = multipart_message.pop(0)
|
|
assert message_type in (zmq_names.CAST_FANOUT_TYPE,
|
|
zmq_names.NOTIFY_TYPE), "Fanout expected!"
|
|
topic_filter = multipart_message.pop(0)
|
|
reply_id = multipart_message.pop(0)
|
|
message_id = multipart_message.pop(0)
|
|
assert reply_id is not None, "Reply id expected!"
|
|
|
|
self.socket.send(topic_filter, zmq.SNDMORE)
|
|
self.socket.send(message_id, zmq.SNDMORE)
|
|
self.socket.send_multipart(multipart_message)
|
|
|
|
LOG.debug("Publishing message %(message_id)s on [%(topic)s]",
|
|
{"topic": topic_filter,
|
|
"message_id": message_id})
|
|
|
|
def cleanup(self):
|
|
self.socket.close()
|