class RemoteBroker:
def __init__(self, hostname, port, request_queue_name, response_queue_name,
request_timeout_millis):
hosts = [(hostname, port)]
connect_timeout = 10
self.conn = Connection(host_and_ports=hosts, timeout=connect_timeout)
self.conn.start()
self.conn.connect(wait=True)
self.request_queue_name = request_queue_name
self.response_queue_name = response_queue_name
self.request_timeout_millis = request_timeout_millis
self._timer = None
def acknowledge(self, headers):
self.conn.ack(headers['message-id'], headers['subscription'])
def publish(self, response):
self.conn.send(body=json.dumps(response, separators=(',', ':')),
destination=self.response_queue_name)
def subscribe(self, handling_strategy, audit):
listener = Listener(self, handling_strategy, self.start_timer,
self.stop_timer, audit)
self.conn.set_listener('listener', listener)
self.conn.subscribe(destination=self.request_queue_name,
id=1,
ack='client-individual')
self.start_timer()
def respond_to(self, headers, response):
self.acknowledge(headers)
self.publish(
OrderedDict([('result', response.result), ('error', None),
('id', response.id)]))
def stop(self):
self.conn.unsubscribe(1)
self.conn.remove_listener('listener')
def close(self):
self.conn.disconnect()
def is_connected(self):
return self.conn.is_connected()
def stop_timer(self):
if self._timer is not None:
self._timer.cancel()
def start_timer(self):
self._timer = Timer(self.request_timeout_millis / 1000.00, self.close)
self._timer.start()
def send(
body: Any,
garden_headers: dict = None,
conn: stomp.Connection = None,
send_destination: str = None,
request_headers: dict = None,
):
message, model_headers = process(body)
headers = consolidate_headers(request_headers, model_headers,
garden_headers)
if conn.is_connected() and send_destination:
destination = send_destination
if request_headers and "reply-to" in request_headers:
destination = request_headers["reply-to"]
conn.send(body=message, headers=headers, destination=destination)
