1. 程式人生 > >python--基於RabbitMQ rpc實現的主機管理

python--基於RabbitMQ rpc實現的主機管理

input == pub tag 3.4 num {} 配置 local

要求:

可以異步的執行多個命令
對多臺機器
>>:run "df -h" --hosts 192.168.3.55 10.4.3.4
task id: 45334
>>: check_task 45334
>>:

思考:
1、分解其中需要實現的功能
(1)命令是發到遠程主機上執行的,命令放在隊列裏,再發到主機處理,主機執行完結果放在隊列裏,提交命令的人自取。
就需要2個進程,一個client,提交命令,取結果,一個server,處理命令,放結果
(2)發送命令的時候,exchange決定往哪個隊列放消息,每個server取自己的命令,用ip作為篩選的binding_key
(3)取結果的時候,就用默認的exchange,直接往reply_to的隊列裏放

server端
 1 #rabbitMQserver=‘10.21.147.189‘
 2 rabbitMQserver=localhost
 3 
 4 import pika
 5 import os
 6 import socket
 7 
 8 class server(object):
 9     def __init__(self):
10         self.connection=pika.BlockingConnection(pika.ConnectionParameters(rabbitMQserver))
11         self.channel=self.connection.channel()
12 self.channel.exchange_declare(exchange=cmd,exchange_type=topic) 13 self.queue_default = self.channel.queue_declare(exclusive=True) 14 self.quname = self.queue_default.method.queue 15 # 獲取本機ip作為binding_key,binding_key的格式是以點分隔的一系列字符串,client發過來的消息,routing_key滿足server的binding_key匹配原則,就會加入server連的那個queue
16 self.hostip=socket.gethostbyname(socket.gethostname()) 17 self.binding_key=#.+self.hostip+.# 18 print("binding_key: %s" % self.binding_key) 19 self.channel.queue_bind(queue=self.quname, exchange=cmd, routing_key=self.binding_key) 20 self.channel.basic_qos(prefetch_count=1) 21 22 self.channel.basic_consume(self.execcmd,queue=self.quname,no_ack=False) 23 self.channel.start_consuming() 24 return 25 26 def execcmd(self,ch,method,props,body): 27 cmd=bytes.decode(body) 28 print("[*] Received %s" % cmd) 29 result = os.popen(cmd).read() 30 print(result) 31 ch.basic_publish( 32 exchange=‘‘, 33 routing_key=props.reply_to, 34 properties=pika.BasicProperties( 35 correlation_id=props.correlation_id 36 ), 37 body=result 38 ) 39 ch.basic_ack(delivery_tag=method.delivery_tag) 40 return 41 42 se=server()



client端
 1 import pika
 2 import uuid
 3 
 4 #rabbitMQserver=‘10.21.147.189‘
 5 rabbitMQserver=localhost
 6 
 7 class client(object):
 8     def __init__(self):
 9         self.cmdid={}
10         self.connection = pika.BlockingConnection(pika.ConnectionParameters(rabbitMQserver))
11         self.channel = self.connection.channel()
12         self.channel.exchange_declare(exchange=cmd, exchange_type=topic)
13 
14         #回寫result的隊列用默認生成的,需要取得名字,exclusive定為True表示只能有本client消費這個queue
15         self.result=self.channel.queue_declare(exclusive=True)
16         self.resultqueue=self.result.method.queue
17         return
18 
19     def showcmdid(self):
20         print("-----命令標識如下:--------")
21         for id,hosts in self.cmdid.items():
22             print("cmdid:%s run on hosts: %s"%(id,hosts))
23         return
24 
25     def callcmd(self,cmdinput):
26         #如果是call命令,格式是 call+"命令"+ --host + 一串ip地址
27         #首先判斷格式對不對,至少4個參數
28         msglist=cmdinput.split()
29         argnum=len(msglist)
30         if argnum<4:
31             print("Wrong cmd. Input again.")
32             return
33         #其次,命令是第二個參數,命令用雙引號標記,所以要strip
34         msg = msglist[1].strip("\"")
35         #然後,第三個是--host,第四個開始是ip,ip當作routing_key
36         routing_key=msglist[3]
37         i=4
38         while i < argnum:
39             routing_key=routing_key+.+msglist[i]
40             i+=1
41         print("routing_key: %s"%routing_key)
42         #再然後,生成一個隨機數,把他作為消息的屬性參數
43         self.corr_id=str(uuid.uuid4())
44         self.cmdid[self.corr_id]=msglist[3:]
45         print("命令標識:%s"%self.corr_id)
46         #然後,把消息發到exchange,routing_key,corr_id當作參數發布
47         self.channel.basic_publish(
48             exchange=cmd,
49             routing_key=routing_key,
50             body=msg,
51             properties=pika.BasicProperties(
52                 reply_to=self.resultqueue,
53                 correlation_id=self.corr_id
54             )
55         )
56         print("[*] Send message %s" % msg)
57         return
58 
59     def on_response(self,ch,method,props,body):
60         if self.targetcmdid==props.correlation_id:
61             self.response=bytes.decode(body)
62             print(self.response)
63         ch.basic_ack(delivery_tag=method.delivery_tag)
64         return
65 
66     def getres(self,cmdinput):
67         msglist = cmdinput.split()
68         self.targetcmdid=msglist[1]
69         self.response=None
70         self.channel.basic_consume(self.on_response,queue=self.resultqueue)
71         while self.response is None:
72             self.connection.process_data_events()
73         return
74 
75 cl=client()
76 while True:
77     msginput=input(">>: ").strip()
78     if msginput.startswith(call):
79         cl.callcmd(msginput)
80     elif msginput.startswith(get):
81         cl.getres(msginput)
82     elif msginput.startswith(show):
83         cl.showcmdid()
84     elif msginput.startswith(exit):
85         cl.connection.close()
86         exit(0)
87     else:
88         print("Wrong cmd. Input again.")
89         continue

目前沒有驗證遠程登陸rabbitMQ server的情況,應該是需要配置用戶名密碼,不能用默認的guest/guest。不過以上功能是實現了。

python--基於RabbitMQ rpc實現的主機管理