python--基於RabbitMQ rpc實現的主機管理
阿新 • • 發佈:2018-03-01
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連的那個queue16 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實現的主機管理