Luocs玩Python之：批量杀掉Redis客户端连接

准备一个hosts.txt配置，里面按如下格式填写Redis集群信息
# cat hosts.txt
[test]
h1 = 1.1.1.1
h2 = 1.1.1.2
h3 = 1.1.1.3
[group1]
h1 = 2.1.1.1
h2 = 2.1.1.2
h3 = 2.1.1.3
# cat killclient.py 
#!/usr/bin/env python 
# -*- coding: UTF-8 -*-
###################################################### 
# @Author:       Luocs(XU)  
# @Create Date:  2014-12-18
######################################################
import os, sys, time, ConfigParser
def killclient(clicmd, host, port, s1, s2, myip):
    os.popen("cat /dev/null > /tmp/client.tmp")
    os.popen("%s -h %s -p %s client list >> /tmp/client.tmp" % (clicmd, host, port))
    f = open("/tmp/client.tmp")
    while 1:
        line = f.readline().strip()
        if not line:
            break
        myclient = os.popen("echo %s | grep -v -E '%s|%s|%s' | awk '{print $2}' | awk -F'=' '{print $2}'" % (line, s1, s2, myip)).read().strip()
        if myclient:
            print("Client %s killed!" % myclient)
            os.popen("%s -h %s -p %s client kill %s" % (clicmd, host, port, myclient))
def getMyip():
    myip = os.popen("/sbin/ifconfig | grep 'inet addr' | awk '{print $2}'").read()
    myip = myip[myip.find(':')+1:myip.find('\n')]
    return myip
def gethost(name):
    fp = ConfigParser.ConfigParser()
    fp.readfp(open('hosts.txt'))
    h1 = fp.get(name,"h1") # 主库的IP
    h2 = fp.get(name,"h2") # 从库IP
    h3 = fp.get(name,"h3") # 从库IP，有多少个可以继续写下去
    return h1, h2, h3
def main():
    name = sys.argv[1]
    port = sys.argv[2]
    #port = 7001 # 也可以写死
    clicmd = "/home/luocs/redis-cli" # redis-cli
    h1, h2, h3 = gethost(name)
    host = h1 # 主库ip传给host
    s1 = h2 
    s2 = h3
    myip = getMyip()
    killclient(clicmd, host, port, s1, s2, myip)
if __name__ == '__main__':
    main()
执行结果：
# python killclient.py test 7001
Client 127.0.0.1:58263 killed!
Client 127.0.0.1:58264 killed!
Client 127.0.0.1:58265 killed!
Client 127.0.0.1:58266 killed!
Client 127.0.0.1:58267 killed!
Client 127.0.0.1:58268 killed!
Client 127.0.0.1:58269 killed!