redis查找大key

redis中查找出比较大的key 

下面直接上代码 （请在测试机上测试）


#!/usr/bin/env python
import sys
import redis


def check_big_key(r, k):
    bigKey = False
    length = 0
    try:
        type = r.type(k)
        if type == "string":
            length = r.strlen(k)
        elif type == "hash":
            length = r.hlen(k)
        elif type == "list":
            length = r.llen(k)
        elif type == "set":
            length = r.scard(k)
        elif type == "zset":
            length = r.zcard(k)
    except:
        return
    if length > 10240:
        bigKey = True
    if bigKey:
        print db, k, type, length


def find_big_key_normal(db_host, db_port, db_num):
    db_password = "123456"
    r = redis.StrictRedis(host=db_host, port=db_port, db=db_num,password=db_password)
    for k in r.scan_iter(count=1000):
        check_big_key(r, k)


def find_big_key_sharding(db_host, db_port, db_num, nodecount):
    r = redis.StrictRedis(host=db_host, port=db_port, db=db_num)
    cursor = 0
    for node in range(0, nodecount):
        while True:
            iscan = r.execute_command("iscan", str(node), str(cursor), "count", "1000")
            for k in iscan[1]:
                check_big_key(r, k)
            cursor = iscan[0]
            print cursor, db, node, len(iscan[1])
            if cursor == "0":
                break;


if __name__ == '__main__':
    if len(sys.argv) != 3:
        print 'Usage: python ', sys.argv[0], ' host port password '
        exit(1)
    db_host = sys.argv[1]
    db_port = sys.argv[2]
    db_password = "123456"
    r = redis.StrictRedis(host=db_host, port=int(db_port),password=db_password)
    nodecount = 1
    #nodecount = r.info()['nodecount']
    keyspace_info = r.info("keyspace")
    for db in keyspace_info:
        print 'check ', db, ' ', keyspace_info[db]
        if nodecount > 1:
            find_big_key_sharding(db_host, db_port, db.replace("db", ""), nodecount)
        else:
            find_big_key_normal(db_host, db_port, db.replace("db", ""))