RabbitMQ的幾種應用場景

以前的幾篇文章介紹了一下RabbitMQ的概念以及環境的搭建和配置,有了RabbitMQ環境就能夠基於其實現一些特殊的任務場景了。RabbitMQ官方有個很好的Tutorials基本覆蓋了RabbitMQ的各中常見應用場景,現以代碼加註釋的方式以其Python客戶端pika爲例簡單介紹以下。更詳盡的信息可參閱:http://www.rabbitmq.com/getstarted.html 。html

以前的幾篇文章:
RabbitMQ概念及環境搭建(一)單節點安裝與配置
RabbitMQ概念及環境搭建(二)RabbitMQ Broker管理
RabbitMQ概念及環境搭建(三)RabbitMQ cluster
RabbitMQ概念及環境搭建(四)RabbitMQ High Availability
RabbitMQ概念及環境搭建(五)與web的整合

python

RabbitMQ是一個消息代理,從「生產者」接收消息並傳遞消息至「消費者」,期間可根據規則路由、緩存、持久化消息。「生產者」也即message發送者如下簡稱P,相對應的「消費者」乃message接收者如下簡稱C,message經過queue由P到C,queue存在於RabbitMQ,可存儲儘量多的message,多個P可向同一queue發送message,多個C可從同一個queue接收message。

web

應用場景1-「Hello Word」緩存

一個P向queue發送一個message,一個C從該queue接收message並打印。
併發

send.py 
producer,鏈接至RabbitMQ Server,聲明隊列,發送message,關閉鏈接,退出。函數

 

[python]  view plain  copy
 
  1. #!/usr/bin/python27  
  2. #encoding:utf8  
  3. import pika  
  4.   
  5. #與RabbitMQ Server創建鏈接  
  6. #鏈接到的broker在本機-localhost上  
  7. connection = pika.BlockingConnection(pika.ConnectionParameters(  
  8.         host='localhost'))  
  9. channel = connection.channel()  
  10.   
  11. #聲明隊列以向其發送消息消息  
  12. #向不存在的位置發送消息時RabbitMQ將消息丟棄  
  13. #queue='hello'指定隊列名字  
  14. channel.queue_declare(queue='hello', durable=True)  
  15.   
  16. #message不能直接發送給queue,需經exchange到達queue,此處使用以空字符串標識的默認的exchange  
  17. #使用默認exchange時容許經過routing_key明確指定message將被髮送給哪一個queue  
  18. #body參數指定了要發送的message內容  
  19. channel.basic_publish(exchange='',  
  20.                       routing_key='hello',  
  21.                       body='Hello World!')  
  22.   
  23. print " [x] Sent 'Hello World!'"  
  24.   
  25. #關閉與RabbitMq Server間的鏈接  
  26. connection.close()  

receive.py 
consumer,鏈接至RabbitMQ Server,聲明隊列,接收消息並進行處理這裏爲打印出消息,退出。測試

 

[python]  view plain  copy
 
  1. #!/usr/bin/env python  
  2. #encoding:utf8  
  3. import pika  
  4.   
  5. #創建到達RabbitMQ Server的connection  
  6. #此處RabbitMQ Server位於本機-localhost  
  7. connection = pika.BlockingConnection(pika.ConnectionParameters(  
  8.         host='localhost'))  
  9. channel = connection.channel()  
  10.   
  11. #聲明queue,確認要從中接收message的queue  
  12. #queue_declare函數是冪等的,可運行屢次,但只會建立一次  
  13. #若能夠確信queue是已存在的,則此處可省略該聲明,如producer已經生成了該queue  
  14. #但在producer和consumer中重複聲明queue是一個好的習慣  
  15. channel.queue_declare(queue='hello')  
  16.   
  17. print ' [*] Waiting for messages. To exit press CTRL+C'  
  18.   
  19. #定義回調函數  
  20. #一旦從queue中接收到一個message回調函數將被調用  
  21. #ch:channel  
  22. #method:  
  23. #properties:  
  24. #body:message  
  25. def callback(ch, method, properties, body):  
  26.     print " [x] Received %r" % (body,)  
  27.   
  28. #從queue接收message的參數設置  
  29. #包括從哪一個queue接收message,用於處理message的callback,是否要確認message  
  30. #默認狀況下是要對消息進行確認的,以防止消息丟失。  
  31. #此處將no_ack明確指明爲True,不對消息進行確認。  
  32. channel.basic_consume(callback,  
  33.                       queue='hello',  
  34.                       no_ack=True)  
  35.   
  36. #開始循環從queue中接收message並使用callback進行處理  
  37. channel.start_consuming()  

測試fetch

 

[plain]  view plain  copy
 
  1. python send.py  
  2. python receive.py  


應用場景2-work queuesui

將耗時的消息處理經過隊列分配給多個consumer來處理,咱們稱此處的consumer爲worker,咱們將此處的queue稱爲Task Queue,其目的是爲了不資源密集型的task的同步處理,也即當即處理task並等待完成。相反,調度task使其稍後被處理。也即把task封裝進message併發送到task queue,worker進程在後臺運行,從task queue取出task並執行job,若運行了多個worker,則task可在多個worker間分配。spa


new_task.py
創建鏈接,聲明隊列,發送能夠模擬耗時任務的message,斷開鏈接、退出。

[python]  view plain  copy
 
  1. #!/usr/bin/env python  
  2. #encoding:utf8  
  3. import pika  
  4. import sys  
  5.   
  6. connection = pika.BlockingConnection(pika.ConnectionParameters(  
  7.         host='localhost'))  
  8. channel = connection.channel()  
  9.   
  10. #僅僅對message進行確認不能保證message不丟失,好比RabbitMQ崩潰了queue就會丟失  
  11. #所以還需使用durable=True聲明queue是持久化的,這樣即使Rabb崩潰了重啓後queue仍然存在  
  12. channel.queue_declare(queue='task_queue', durable=True)  
  13.   
  14. #從命令行構造將要發送的message  
  15. message = ' '.join(sys.argv[1:]) or "Hello World!"  
  16.   
  17. #除了要聲明queue是持久化的外,還需聲明message是持久化的  
  18. #basic_publish的properties參數指定message的屬性  
  19. #此處pika.BasicProperties中的delivery_mode=2指明message爲持久的  
  20. #這樣一來RabbitMQ崩潰重啓後queue仍然存在其中的message也仍然存在  
  21. #需注意的是將message標記爲持久的並不能徹底保證message不丟失,由於  
  22. #從RabbitMQ接收到message到將其存儲到disk仍需一段時間,若此時RabbitMQ崩潰則message會丟失  
  23. #何況RabbitMQ不會對每條message作fsync動做  
  24. #可經過publisher confirms實現更強壯的持久性保證  
  25. channel.basic_publish(exchange='',  
  26.                       routing_key='task_queue',  
  27.                       body=message,  
  28.                       properties=pika.BasicProperties(  
  29.                          delivery_mode = 2, # make message persistent  
  30.                       ))  
  31. print " [x] Sent %r" % (message,)  
  32. connection.close()  

worker.py
創建鏈接,聲明隊列,不斷的接收message,處理任務,進行確認。

[python]  view plain  copy
 
  1. #!/usr/bin/env python  
  2. #encoding:utf8  
  3. import pika  
  4. import time  
  5.   
  6. #默認狀況RabbirMQ將message以round-robin方式發送給下一個consumer  
  7. #每一個consumer接收到的平均message量是同樣的  
  8. #能夠同時運行兩個或三個該程序進行測試  
  9.   
  10. connection = pika.BlockingConnection(pika.ConnectionParameters(  
  11.         host='localhost'))  
  12. channel = connection.channel()  
  13.   
  14. #僅僅對message進行確認不能保證message不丟失,好比RabbitMQ崩潰了  
  15. #還需使用durable=True聲明queue是持久化的,這樣即使Rabb崩潰了重啓後queue仍然存在其中的message不會丟失  
  16. #RabbitMQ中不容許使用不一樣的參數定義同名queue  
  17. channel.queue_declare(queue='task_queue', durable=True)  
  18.   
  19. print ' [*] Waiting for messages. To exit press CTRL+C'  
  20.   
  21. #回調函數,函數體模擬耗時的任務處理:以message中'.'的數量表示sleep的秒數  
  22. def callback(ch, method, properties, body):  
  23.     print " [x] Received %r" % (body,)  
  24.     time.sleep( body.count('.') )  
  25.     print " [x] Done"  
  26.     #對message進行確認  
  27.     ch.basic_ack(delivery_tag = method.delivery_tag)  
  28.   
  29. #若存在多個consumer每一個consumer的負載可能不一樣,有些處理的快有些處理的慢  
  30. #RabbitMQ並無論這些,只是簡單的以round-robin的方式分配message  
  31. #這可能形成某些consumer積壓不少任務處理不完而一些consumer長期處於飢餓狀態  
  32. #可使用prefetch_count=1的basic_qos方法可告知RabbitMQ只有在consumer處理並確認了上一個message後才分配新的message給他  
  33. #不然分給另外一個空閒的consumer  
  34. channel.basic_qos(prefetch_count=1)  
  35.   
  36. #這裏移除了no_ack=True這個參數,也即須要對message進行確認(默認行爲)  
  37. #不然consumer在偶然down後其正在處理和分配到該consumer還未處理的message可能發生丟失  
  38. #由於此時RabbitMQ在發送完message後當即從內存刪除該message  
  39. #假如沒有設置no_ack=True則consumer在偶然down掉後其正在處理和分配至該consumer但還將來得及處理的message會從新分配到其餘consumer  
  40. #沒有設置no_ack=True則consumer在收到message後會向RabbitMQ反饋已收到並處理了message告訴RabbitMQ能夠刪除該message  
  41. #RabbitMQ中沒有超時的概念,只有在consumer down掉後從新分發message  
  42. channel.basic_consume(callback,  
  43.                       queue='task_queue')  
  44.   
  45. channel.start_consuming()  

測試

[plain]  view plain  copy
 
  1. python new_task.py "A very hard task which takes two seconds.."  
  2. python worker.py  

 

應用場景3-Publish/Subscribe

在應用場景2中一個message(task)僅被傳遞給了一個comsumer(worker)。如今咱們設法將一個message傳遞給多個consumer。這種模式被稱爲publish/subscribe。此處以一個簡單的日誌系統爲例進行說明。該系統包含一個log發送程序和一個log接收並打印的程序。由log發送者發送到queue的消息能夠被全部運行的log接收者接收。所以,咱們能夠運行一個log接收者直接在屏幕上顯示log,同時運行另外一個log接收者將log寫入磁盤文件。

 


receive_logs.py
日誌消息接收者:創建鏈接,聲明exchange,將exchange與queue進行綁定,開始不停的接收log並打印。

[python]  view plain  copy
 
  1. #!/usr/bin/env python  
  2. #encoding:utf8  
  3. import pika  
  4.   
  5. connection = pika.BlockingConnection(pika.ConnectionParameters(  
  6.         host='localhost'))  
  7. channel = connection.channel()  
  8.   
  9. #做爲好的習慣,在producer和consumer中分別聲明一次以保證所要使用的exchange存在  
  10. channel.exchange_declare(exchange='logs',  
  11.                          type='fanout')  
  12.   
  13. #在不一樣的producer和consumer間共享queue時指明queue的name是重要的  
  14. #但某些時候,好比日誌系統,須要接收全部的log message而非一個子集  
  15. #並且僅對當前的message 流感興趣,對於過期的message不感興趣,那麼  
  16. #能夠申請一個臨時隊列這樣,每次鏈接到RabbitMQ時會以一個隨機的名字生成  
  17. #一個新的空的queue,將exclusive置爲True,這樣在consumer從RabbitMQ斷開後會刪除該queue  
  18. result = channel.queue_declare(exclusive=True)  
  19.   
  20. #用於獲取臨時queue的name  
  21. queue_name = result.method.queue  
  22.   
  23. #exchange與queue之間的關係成爲binding  
  24. #binding告訴exchange將message發送該哪些queue  
  25. channel.queue_bind(exchange='logs',  
  26.                    queue=queue_name)  
  27.   
  28. print ' [*] Waiting for logs. To exit press CTRL+C'  
  29.   
  30. def callback(ch, method, properties, body):  
  31.     print " [x] %r" % (body,)  
  32.   
  33. #從指定地queue中consume message且不確認  
  34. channel.basic_consume(callback,  
  35.                       queue=queue_name,  
  36.                       no_ack=True)  
  37.   
  38. channel.start_consuming()  

emit_log.py
日誌消息發送者:創建鏈接,聲明fanout類型的exchange,經過exchage向queue發送日誌消息,消息被廣播給全部接收者,關閉鏈接,退出。

 

[python]  view plain  copy
 
  1. #!/usr/bin/env python  
  2. #encoding:utf8  
  3.   
  4. import pika  
  5. import sys  
  6.   
  7. connection = pika.BlockingConnection(pika.ConnectionParameters(  
  8.         host='localhost'))  
  9. channel = connection.channel()  
  10.   
  11. #producer只能經過exchange將message發給queue  
  12. #exchange的類型決定將message路由至哪些queue  
  13. #可用的exchange類型:direct\topic\headers\fanout  
  14. #此處定義一個名稱爲'logs'的'fanout'類型的exchange,'fanout'類型的exchange簡單的將message廣播到它所知道的全部queue  
  15. channel.exchange_declare(exchange='logs',  
  16.                          type='fanout')  
  17.   
  18. message = ' '.join(sys.argv[1:]) or "info: Hello World!"  
  19.   
  20. #將message publish到名爲log的exchange中  
  21. #由於是fanout類型的exchange,這裏無需指定routing_key  
  22. channel.basic_publish(exchange='logs',  
  23.                       routing_key='',  
  24.                       body=message)  
  25.   
  26. print " [x] Sent %r" % (message,)  
  27.   
  28. connection.close()  

測試

 

 

[plain]  view plain  copy
 
  1. python receive_logs.py  
  2. python emit_log.py "info: This is the log message"  

 

應用場景4-Routing

應用場景3中構建了簡單的log系統,能夠將log message廣播至多個receiver。如今咱們將考慮只把指定的message類型發送給其subscriber,好比,只把error message寫到log file而將全部log message顯示在控制檯。


receive_logs_direct.py
log message接收者:創建鏈接,聲明direct類型的exchange,聲明queue,使用提供的參數做爲routing_key將queue綁定到exchange,開始循環接收log message並打印。

 

[python]  view plain  copy
 
  1. #!/usr/bin/env python  
  2. import pika  
  3. import sys  
  4.   
  5. connection = pika.BlockingConnection(pika.ConnectionParameters(  
  6.         host='localhost'))  
  7. channel = connection.channel()  
  8.   
  9. #聲明一個名爲direct_logs類型爲direct的exchange  
  10. #同時在producer和consumer中聲明exchage或queue是個好習慣,以保證其存在  
  11. channel.exchange_declare(exchange='direct_logs',  
  12.                          type='direct')  
  13.   
  14. result = channel.queue_declare(exclusive=True)  
  15. queue_name = result.method.queue  
  16.   
  17. #從命令行獲取參數:routing_key  
  18. severities = sys.argv[1:]  
  19. if not severities:  
  20.     print >> sys.stderr, "Usage: %s [info] [warning] [error]" % (sys.argv[0],)  
  21.     sys.exit(1)  
  22.   
  23. for severity in severities:  
  24.     #exchange和queue之間的binding可接受routing_key參數  
  25.     #該參數的意義依賴於exchange的類型  
  26.     #fanout類型的exchange直接忽略該參數  
  27.     #direct類型的exchange精確匹配該關鍵字進行message路由  
  28.     #對多個queue使用相同的binding_key是合法的  
  29.     channel.queue_bind(exchange='direct_logs',  
  30.                        queue=queue_name,  
  31.                        routing_key=severity)  
  32.   
  33. print ' [*] Waiting for logs. To exit press CTRL+C'  
  34.   
  35. def callback(ch, method, properties, body):  
  36.     print " [x] %r:%r" % (method.routing_key, body,)  
  37.   
  38. channel.basic_consume(callback,  
  39.                       queue=queue_name,  
  40.                       no_ack=True)  
  41.   
  42. channel.start_consuming()  

emit_log_direct.py
log message發送者:創建鏈接,聲明direct類型的exchange,生成併發送log message到exchange,關閉鏈接,退出。

[python]  view plain  copy
 
  1. #!/usr/bin/env python  
  2. #encoding:utf8  
  3.   
  4. import pika  
  5. import sys  
  6.   
  7. connection = pika.BlockingConnection(pika.ConnectionParameters(  
  8.         host='localhost'))  
  9. channel = connection.channel()  
  10.   
  11. #聲明一個名爲direct_logs的direct類型的exchange  
  12. #direct類型的exchange  
  13. channel.exchange_declare(exchange='direct_logs',  
  14.                          type='direct')  
  15.   
  16. #從命令行獲取basic_publish的配置參數  
  17. severity = sys.argv[1] if len(sys.argv) > else 'info'  
  18. message = ' '.join(sys.argv[2:]) or 'Hello World!'  
  19.   
  20. #向名爲direct_logs的exchage按照設置的routing_key發送message  
  21. channel.basic_publish(exchange='direct_logs',  
  22.                       routing_key=severity,  
  23.                       body=message)  
  24.   
  25. print " [x] Sent %r:%r" % (severity, message)  
  26. connection.close()  

測試:

python receive_logs_direct.py info
python emit_log_direct.py info "The message"

應用場景5-topic

應用場景4中改進的log系統中用direct類型的exchange替換應用場景3中的fanout類型exchange實現將不一樣的log message發送給不一樣的subscriber(也即分別經過不一樣的routing_key將queue綁定到exchange,這樣exchange即可將不一樣的message根據message內容路由至不一樣的queue)。但仍然存在限制,不能根據多個規則路由消息,好比接收者要麼只能收error類型的log message要麼只能收info類型的message。若是咱們不只想根據log的重要級別如info、warning、error等來進行log message路由還想同時根據log message的來源如auth、cron、kern來進行路由。爲了達到此目的,須要topic類型的exchange。topic類型的exchange中routing_key中能夠包含兩個特殊字符:「*」用於替代一個詞,「#」用於0個或多個詞。

receive_logs_topic.py
log message接收者:創建鏈接,聲明topic類型的exchange,聲明queue,根據程序參數構造routing_key,根據routing_key將queue綁定到exchange,循環接收並處理message。

[python]  view plain  copy
 
  1. #!/usr/bin/env python  
  2. import pika  
  3. import sys  
  4.   
  5. connection = pika.BlockingConnection(pika.ConnectionParameters(  
  6.         host='localhost'))  
  7. channel = connection.channel()  
  8.   
  9. #聲明一個名爲direct_logs類型爲direct的exchange  
  10. #同時在producer和consumer中聲明exchage或queue是個好習慣,以保證其存在  
  11. channel.exchange_declare(exchange='direct_logs',  
  12.                          type='direct')  
  13.   
  14. result = channel.queue_declare(exclusive=True)  
  15. queue_name = result.method.queue  
  16.   
  17. #從命令行獲取參數:routing_key  
  18. severities = sys.argv[1:]  
  19. if not severities:  
  20.     print >> sys.stderr, "Usage: %s [info] [warning] [error]" % (sys.argv[0],)  
  21.     sys.exit(1)  
  22.   
  23. for severity in severities:  
  24.     #exchange和queue之間的binding可接受routing_key參數  
  25.     #該參數的意義依賴於exchange的類型  
  26.     #fanout類型的exchange直接忽略該參數  
  27.     #direct類型的exchange精確匹配該關鍵字進行message路由  
  28.     #對多個queue使用相同的binding_key是合法的  
  29.     channel.queue_bind(exchange='direct_logs',  
  30.                        queue=queue_name,  
  31.                        routing_key=severity)  
  32.   
  33. print ' [*] Waiting for logs. To exit press CTRL+C'  
  34.   
  35. def callback(ch, method, properties, body):  
  36.     print " [x] %r:%r" % (method.routing_key, body,)  
  37.   
  38. channel.basic_consume(callback,  
  39.                       queue=queue_name,  
  40.                       no_ack=True)  
  41.   
  42. channel.start_consuming()  

emit_log_topic.py
log message發送者:創建鏈接、聲明topic類型的exchange、根據程序參數構建routing_key和要發送的message,以構建的routing_key將message發送給topic類型的exchange,關閉鏈接,退出。

[python]  view plain  copy
 
  1. #!/usr/bin/env python  
  2. #encoding:utf8  
  3. import pika  
  4. import sys  
  5.   
  6. connection = pika.BlockingConnection(pika.ConnectionParameters(  
  7.         host='localhost'))  
  8. channel = connection.channel()  
  9.   
  10. #聲明一個名爲topic_logs的topic類型的exchange  
  11. #topic類型的exchange可經過通配符對message進行匹配從而路由至不一樣queue  
  12. channel.exchange_declare(exchange='topic_logs',  
  13.                          type='topic')  
  14.   
  15. routing_key = sys.argv[1] if len(sys.argv) > else 'anonymous.info'  
  16. message = ' '.join(sys.argv[2:]) or 'Hello World!'  
  17.   
  18. channel.basic_publish(exchange='topic_logs',  
  19.                       routing_key=routing_key,  
  20.                       body=message)  
  21.   
  22. print " [x] Sent %r:%r" % (routing_key, message)  
  23. connection.close()  

 

測試:

 

[plain]  view plain  copy
 
  1. python receive_logs_topic.py "*.rabbit"  
  2. python emit_log_topic.py red.rabbit Hello  

 

應用場景6-PRC

在應用場景2中描述瞭如何使用work queue將耗時的task分配到不一樣的worker中。可是,若是咱們task是想在遠程的計算機上運行一個函數並等待返回結果呢。這根場景2中的描述是一個徹底不一樣的故事。這一模式被稱爲遠程過程調用。如今,咱們將構建一個RPC系統,包含一個client和可擴展的RPC server,經過返回斐波那契數來模擬RPC service。

rpc_server.py
RPC server:創建鏈接,聲明queue,定義了一個返回指定數字的斐波那契數的函數,定義了一個回調函數在接收到包含參數的調用請求後調用本身的返回斐波那契數的函數並將結果發送到與接收到message的queue相關聯的queue,並進行確認。開始接收調用請求並用回調函數進行請求處理。

 

[python]  view plain  copy
 
  1. #!/usr/bin/env python  
  2. #encoding:utf8  
  3. import pika  
  4.   
  5. #創建到達RabbitMQ Server的connection  
  6. connection = pika.BlockingConnection(pika.ConnectionParameters(  
  7.         host='localhost'))  
  8. channel = connection.channel()  
  9.   
  10. #聲明一個名爲rpc_queue的queue  
  11. channel.queue_declare(queue='rpc_queue')  
  12.   
  13. #計算指定數字的斐波那契數  
  14. def fib(n):  
  15.     if n == 0:  
  16.         return 0  
  17.     elif n == 1:  
  18.         return 1  
  19.     else:  
  20.         return fib(n-1) + fib(n-2)  
  21.   
  22. #回調函數,從queue接收到message後調用該函數進行處理  
  23. def on_request(ch, method, props, body):  
  24.     #由message獲取要計算斐波那契數的數字  
  25.     n = int(body)  
  26.   
  27.     print " [.] fib(%s)"  % (n,)  
  28.     #調用fib函數得到計算結果  
  29.     response = fib(n)  
  30.       
  31.     #exchage爲空字符串則將message發送個到routing_key指定的queue  
  32.     #這裏queue爲回調函數參數props中reply_ro指定的queue  
  33.     #要發送的message爲計算所得的斐波那契數  
  34.     #properties中correlation_id指定爲回調函數參數props中co的rrelation_id  
  35.     #最後對消息進行確認  
  36.     ch.basic_publish(exchange='',  
  37.                      routing_key=props.reply_to,  
  38.                      properties=pika.BasicProperties(correlation_id = \  
  39.                                                          props.correlation_id),  
  40.                      body=str(response))  
  41.     ch.basic_ack(delivery_tag = method.delivery_tag)  
  42.   
  43. #只有consumer已經處理並確認了上一條message時queue才分派新的message給它  
  44. channel.basic_qos(prefetch_count=1)  
  45.   
  46. #設置consumeer參數,即從哪一個queue獲取消息使用哪一個函數進行處理,是否對消息進行確認  
  47. channel.basic_consume(on_request, queue='rpc_queue')  
  48.   
  49. print " [x] Awaiting RPC requests"  
  50.   
  51. #開始接收並處理消息  
  52. channel.start_consuming()  

rpc_client.py
RPC client:遠程過程調用發起者:定義了一個類,類中初始化到RabbitMQ Server的鏈接、聲明回調queue、開始在回調queue上等待接收響應、定義了在回調queue上接收到響應後的處理函數on_response根據響應關聯的correlation_id屬性做出響應、定義了調用函數並在其中向調用queue發送包含correlation_id等屬性的調用請求、初始化一個client實例,以30爲參數發起遠程過程調用。

 

[python]  view plain  copy
 
  1. #!/usr/bin/env python  
  2. #encoding:utf8  
  3. import pika  
  4. import uuid  
  5.   
  6. #在一個類中封裝了connection創建、queue聲明、consumer配置、回調函數等  
  7. class FibonacciRpcClient(object):  
  8.     def __init__(self):  
  9.         #創建到RabbitMQ Server的connection  
  10.         self.connection = pika.BlockingConnection(pika.ConnectionParameters(  
  11.                 host='localhost'))  
  12.   
  13.         self.channel = self.connection.channel()  
  14.           
  15.         #聲明一個臨時的回調隊列  
  16.         result = self.channel.queue_declare(exclusive=True)  
  17.         self.callback_queue = result.method.queue  
  18.   
  19.         #此處client既是producer又是consumer,所以要配置consume參數  
  20.         #這裏的指明從client本身建立的臨時隊列中接收消息  
  21.         #並使用on_response函數處理消息  
  22.         #不對消息進行確認  
  23.         self.channel.basic_consume(self.on_response, no_ack=True,  
  24.                                    queue=self.callback_queue)  
  25.       
  26.     #定義回調函數  
  27.     #比較類的corr_id屬性與props中corr_id屬性的值  
  28.     #若相同則response屬性爲接收到的message  
  29.     def on_response(self, ch, method, props, body):  
  30.         if self.corr_id == props.correlation_id:  
  31.             self.response = body  
  32.    
  33.     def call(self, n):  
  34.         #初始化response和corr_id屬性  
  35.         self.response = None  
  36.         self.corr_id = str(uuid.uuid4())  
  37.          
  38.         #使用默認exchange向server中定義的rpc_queue發送消息  
  39.         #在properties中指定replay_to屬性和correlation_id屬性用於告知遠程server  
  40.         #correlation_id屬性用於匹配request和response  
  41.         self.channel.basic_publish(exchange='',  
  42.                                    routing_key='rpc_queue',  
  43.                                    properties=pika.BasicProperties(  
  44.                                          reply_to = self.callback_queue,  
  45.                                          correlation_id = self.corr_id,  
  46.                                          ),  
  47.                                    #message需爲字符串  
  48.                                    body=str(n))  
  49.   
  50.         while self.response is None:  
  51.             self.connection.process_data_events()  
  52.           
  53.         return int(self.response)  
  54.   
  55. #生成類的實例  
  56. fibonacci_rpc = FibonacciRpcClient()  
  57.   
  58. print " [x] Requesting fib(30)"  
  59. #調用實例的call方法  
  60. response = fibonacci_rpc.call(30)  
  61. print " [.] Got %r" % (response,)  

測試:

[python]  view plain  copy
 
    1. python rpc_server.py  
    2. python rpc_client.py  
相關文章
相關標籤/搜索