您现在的位置是:首页 > 文章详情

使用Python实现清除RabbitMQ里面1小时没有数据的连接

日期:2017-07-25点击:406

  公司RabbitMQ消费者重新连接后旧连接不会断开,会一直存在于服务端,至于是什么原因目前还没有找到。这样导致连接数每过一段时间就增加很多,所以写了一个Python脚本来删除无效的连接,根据每个连接里面的数据传输来定义是否是有效,如果1小时都没有数据传输那么就是无效,具体实现代码如下:

#!/usr/bin/python #coding:utf8 """ 删除RabbitMQ所有在1小时内没有数据的连接 """ from optparse import OptionParser import sys import json import urllib2 import requests #使用选项帮助信息可以使用中文 reload(sys) sys.setdefaultencoding("utf-8") usage = sys.argv[0] + " <Options>" parser = OptionParser(usage) parser.add_option("-v",                   dest="vhost",                   action="store",                   default=False,                   help="Select vhost, Defaulte All") options, args = parser.parse_args() class RabbitMQ:     def __init__(self, user='guest', passwd='guest', server_ip='192.168.1.31'):         self.user = user         self.password = passwd         self.server_ip = server_ip     def getAllConnections(self):         """         获取所有连接返回Json         """         if not options.vhost:             connections = requests.get("http://{0}:15672/api/connections".format(self.server_ip), auth=(self.user, self.password))         else:             connections = requests.get("http://{0}:15672/api/vhosts/{1}/connections".format(self.server_ip, options.vhost),                                        auth=(self.user, self.password))         connections = connections.json()         return connections     def connectionsNumber(self):         """         对上面返回的Json进行处理,获取关键数据,返回列表         """         list1 = []         data = self.getAllConnections()         for i in data:             list1.append(i['name'])          return list1     def url(self):         """         生成最终访问API,返回列表         """         url_list = []         data = self.connectionsNumber()         for i in data:             send_ip = i.split("->")[0].split(":")[0]             send_port = i.split("->")[0].split(":")[1].strip()             Receive_ip = i.split("->")[1].split(":")[0].strip()             url = 'http://{0}:15672/api/connections/{1}%3A{2}%20-%3E%20{3}%3A5672?data_rates_age=3600&data_rates_incr=60'.format(self.server_ip, send_ip, send_port, Receive_ip)             url_list.append(url)         return url_list     def getAllData(self):         """         获取API的所有信息,返回列表         """         data = self.url()         if self.server_ip == "10.8.5.3":             authorization = "Basic Z3Vlc3Q6UmRuN3lsV2FmZWs2Sjk4aA=="         else:             authorization = "Basic Z3Vlc3Q6Z3Vlc3Q="         user_agent = "Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/59.0.3071.115 Safari/537.36"         list1 = []         for i in data:             url = i             request = urllib2.Request(i, headers={"User-Agent":user_agent, "Authorization":authorization})             req = urllib2.urlopen(request)             list1.append(json.loads(req.read()))         return list1     def getAvgRate(self):         """         获取每个连接的recv_oct_details和send_oct_details,根据1小时平均,如果无数据传输则为无效连接,生成删除API并删除         """         data = self.getAllData()         for i in data:             n = 2             recv_avg_rate = i.get("recv_oct_details").get("avg_rate")             n += recv_avg_rate             send_avg_rate = i.get("send_oct_details").get("avg_rate")             n += send_avg_rate             if n == 2:                 name = i.get("name")                 send_ip = name.split("->")[0].split(":")[0]                 send_port = name.split("->")[0].split(":")[1].strip()                 Receive_ip = name.split("->")[1].split(":")[0].strip()                 api = 'http://{0}:15672/api/connections/{1}:{2}%20-%3E%20{3}:5672'.format(self.server_ip, send_ip, send_port, Receive_ip)                 code = requests.delete(i, auth=(self.user, self.passwd))                 if code.status_code == 204:                     print("{0}删除成功".format(send_port))                 else:                     print("{0}删除失败".format(send_port))                     sys.exit(1)         print("done...") if __name__ == '__main__':     mq = RabbitMQ()     mq.getAvgRate()


原文链接:https://blog.51cto.com/270142877/1950642
关注公众号

低调大师中文资讯倾力打造互联网数据资讯、行业资源、电子商务、移动互联网、网络营销平台。

持续更新报道IT业界、互联网、市场资讯、驱动更新,是最及时权威的产业资讯及硬件资讯报道平台。

转载内容版权归作者及来源网站所有,本站原创内容转载请注明来源。

文章评论

共有0条评论来说两句吧...

文章二维码

扫描即可查看该文章

点击排行

推荐阅读

最新文章