70 lines
No EOL
1.7 KiB
Python
70 lines
No EOL
1.7 KiB
Python
#!/usr/bin/env python
|
|
import uuid
|
|
import pika
|
|
from requests import Request
|
|
|
|
from utils.broker import encode_request
|
|
|
|
|
|
class RpcClient(object):
|
|
def __init__(self):
|
|
self.connection = pika.BlockingConnection(
|
|
pika.ConnectionParameters(host="localhost"),
|
|
)
|
|
|
|
self.channel = self.connection.channel()
|
|
|
|
result = self.channel.queue_declare(queue="", exclusive=True)
|
|
self.callback_queue = result.method.queue
|
|
|
|
self.channel.basic_consume(
|
|
queue=self.callback_queue,
|
|
on_message_callback=self.on_response,
|
|
auto_ack=True,
|
|
)
|
|
|
|
self.response = None
|
|
self.corr_id = None
|
|
|
|
def on_response(self, ch, method, props, body):
|
|
if self.corr_id == props.correlation_id:
|
|
self.response = body
|
|
|
|
def call(self, request, correlation_id=None):
|
|
data = encode_request(request)
|
|
self.response = None
|
|
if not correlation_id:
|
|
self.corr_id = str(uuid.uuid4())
|
|
|
|
|
|
|
|
self.channel.basic_publish(
|
|
exchange="",
|
|
routing_key="rpc_queue",
|
|
properties=pika.BasicProperties(
|
|
reply_to=self.callback_queue,
|
|
correlation_id=self.corr_id,
|
|
),
|
|
body=data,
|
|
)
|
|
self.connection.process_data_events(time_limit=5)
|
|
return self.response
|
|
|
|
|
|
|
|
#
|
|
method = 'GET'
|
|
url = 'http://accounts/users/api/account/'
|
|
|
|
request = Request(method=method, url=url, json={'key_1': 'requests_lib'})
|
|
|
|
|
|
rpc = RpcClient()
|
|
|
|
print(" [x] Requesting ...")
|
|
# for i in range(1000):
|
|
response = rpc.call(request)
|
|
print(response.decode())
|
|
# print(type(response.decode()))
|
|
|
|
# print(i) |