RabbitMQ遠端執行任務RPC。
阿新 • • 發佈:2018-11-12
如果想發一條命令給遠端機器,再把結果返回
這種模式叫RPC:遠端過程呼叫
傳送方將傳送的訊息放在一個queue裡,由接收方取。
接收方再把執行結果放在另外一個queue裡,由傳送方取
實際上,傳送方把1,隨機生成的接收queue名 2,UUID值發過去了(用於確認)
客戶端:作用是傳送指令
''' RPC客戶端程式碼 ''' import pika import uuid import time class FibonacciRpcClient(object): def __init__(self): # 建連線 self.connection = pika.BlockingConnection(pika.ConnectionParameters('localhost')) # 建頻道 self.channel = self.connection.channel() # 宣告隨機queue,不用了自動刪除 result = self.channel.queue_declare(exclusive=True) # 獲取隨機queue名稱 self.callback_queue = result.method.queue # 準備消費者,從隨機queue中消費資訊交給on_response處理 # 聲明瞭我要收了 self.channel.basic_consume(self.on_response, no_ack=True, queue=self.callback_queue) def on_response(self, ch, method, props, body): # 自己發的ID和客戶端返回的ID一樣才執行 # 只有這個相等,才說明這個結果是我發過去的命令的結果。Perfect! if self.corr_id == props.correlation_id: self.response = body def call(self, n): self.response= None # 生成的是唯一字串,每次都不一樣,將這個發給伺服器端 self.corr_id = str(uuid.uuid4()) # 發訊息到rpc_queue裡,可以看出,發的時候引數裡把回撥queue名給伺服器了 # 這樣伺服器就知道向哪個queue裡發,帶的引數是靈活的queue,伺服器計算結果放到該queue self.channel.basic_publish(exchange='', routing_key='rpc_queue', properties=pika.BasicProperties( reply_to=self.callback_queue, correlation_id=self.corr_id ), body=str(n)) # 發了訊息就該收結果了,但沒有寫start_consuming方法(該方法會阻塞), # 這裡self.connection.process_data_events()相當於非阻塞型的start_consuming # self.response初始化為None while self.response is None: self.connection.process_data_events() # 收到訊息就會觸發on_response函式 print("no message...") time.sleep(0.5) # 這裡其實可以不sleep,可以再發個命令。後發的這個可能先執行,不用UUID就不知道哪個是哪個 return int(self.response) fibonacci_rpc = FibonacciRpcClient() print("Requesting fib(30)") response = fibonacci_rpc.call(30) print("response:", response)
伺服器端:接收指令並把處理結果返回
''' RPC伺服器端程式碼 ''' import pika # 建連線 connection = pika.BlockingConnection(pika.ConnectionParameters('localhost')) # 建頻道 channel = connection.channel() # 宣告queue,先啟動伺服器,客戶端就不用宣告rpc_queue了 channel.queue_declare(queue='rpc_queue') def fib(n): if n == 0: return 0 elif n == 1: return 1 else: return fib(n-1) + fib(n-2) def on_request(ch, method, props, body): n = int(body) print("fib({0}):".format(n)) response = fib(n) ch.basic_publish(exchange='', routing_key=props.reply_to, properties=pika.BasicProperties( # 這個是客戶端發過來的correlation_id,再發回去 correlation_id=props.correlation_id ), body=str(response)) ch.basic_ack(delivery_tag=method.delivery_tag) channel.basic_qos(prefetch_count=1) channel.basic_consume(on_request, queue='rpc_queue') print("Awaiting RPC requests") channel.start_consuming()
執行結果:
''' 先啟動服務端: Awaiting RPC requests 再啟動客戶端: Requesting fib(30) no message... no message... no message... no message... response: 832040 再看伺服器端: Awaiting RPC requests fib(30): '''