龙空技术网

如何用Python脚本统计redis key类型数据大小分布

波波说运维 202

前言:

现在大家对“python统计学校类型”大约比较关怀,朋友们都想要学习一些“python统计学校类型”的相关文章。那么小编在网上收集了一些关于“python统计学校类型””的相关文章,希望咱们能喜欢,咱们一起来了解一下吧!

概述

今天主要介绍怎么统计redis key类型数据大小分布。

原理:使用redis命令: scan、pipline、type 和 debug object 来得到 redis key 信息。

脚本

内容如下:

#!/usr/bin/python# -*- coding: utf-8 -*-import getoptimport timeimport sysimport redisimport osreload(sys)sys.setdefaultencoding('utf8')__Version__ = "1.8.1"''' Function : 1. Show All Key type 2. Show Key size distrute 3. Show Key key counter 4. Show Key with no ttl counter'''#LIST FORMAT#0 1 2 3 4 5 6 7 8 9 10 11 #ALL <=64 <=128 <=256 <=512 <=1024 <=2048 <=3072 <=4096 <=5120 <=6044 >6044string_size_counter_list = [0,0,0,0,0,0,0,0,0,0,0,0]list_size_counter_list = [0,0,0,0,0,0,0,0,0,0,0,0]hash_size_counter_list = [0,0,0,0,0,0,0,0,0,0,0,0]set_size_counter_list = [0,0,0,0,0,0,0,0,0,0,0,0]zset_size_counter_list = [0,0,0,0,0,0,0,0,0,0,0,0]def size_counter_do_count(keySizeCounterList,keySize,keyTtl): keySizeCounterList[0] += 1 #// Add all global counter if keyTtl < 0: keySizeCounterList[1] += 1 if keySize <= 64: keySizeCounterList[2] += 1 elif keySize <= 128: keySizeCounterList[3] += 1 elif keySize <= 512: keySizeCounterList[4] += 1 elif keySize <= 1024: keySizeCounterList[5] += 1 elif keySize <= 2048: keySizeCounterList[6] += 1 elif keySize <= 3072: keySizeCounterList[7] += 1 elif keySize <= 4096: keySizeCounterList[8] += 1 elif keySize <= 5120: keySizeCounterList[9] += 1 elif keySize <= 6044: keySizeCounterList[10] += 1 else: keySizeCounterList[11] += 1 return keySizeCounterListdef key_size_static(keyType,keySize,keyTtl): global string_size_counter_list global list_size_counter_list  global hash_size_counter_list  global set_size_counter_list  global zset_size_counter_list  if keyType == "string": string_size_counter_list = size_counter_do_count(string_size_counter_list,keySize,keyTtl) elif keyType == "hash": hash_size_counter_list = size_counter_do_count(hash_size_counter_list,keySize,keyTtl) elif keyType == "list": list_size_counter_list = size_counter_do_count(list_size_counter_list,keySize,keyTtl) elif keyType == "set": set_size_counter_list = size_counter_do_count(set_size_counter_list,keySize,keyTtl) elif keyType == "zset": zset_size_counter_list = size_counter_do_count(zset_size_counter_list,keySize,keyTtl) else: print "UNKONW KEY TYPE !!",keyType '''Redis SCAN command got keys '''def get_key(rdbConn,start): try: keys_list = rdbConn.scan(start,count=20) return keys_list except Exception,e: print e''' Redis DEBUG OBJECT command got key info '''def get_key_info(rdbConn,keyName): try: rpiple = rdbConn.pipeline() rpiple.type(keyName) rpiple.debug_object(keyName) rpiple.ttl(keyName) key_info_list = rpiple.execute() return key_info_list except Exception,e: print "INFO : ",edef redis_key_static(key_info_list): keyType = key_info_list[0] keySize = key_info_list[1]['serializedlength'] keyTtl = key_info_list[2] key_size_static(keyType,keySize,keyTtl)'''Print Key distrubution '''def show_static_info(host,port,start_time,end_time): print u'''----------------------------------------------------------------------------------------------------------------------------------------------------统计时间:[%s ~ %s]Redis服务器[%s:%s]数据类型和数据大小分布情况如下:---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|KEY TYPE | KEY COUNT | KEY No TTL | KEY 64(byte) | KEY 128 | Key 512 | Key 1024 | Key 2048 | Key 3072 | Key 4096 | Key 5120 | Key 6044 | Key large ||%7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s ||%7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s ||%7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s ||%7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s ||%7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s | %7s |----------------------------------------------------------------------------------------------------------------------------------------------------''' % (start_time,end_time,host,port,'String',string_size_counter_list[0],string_size_counter_list[1],string_size_counter_list[2],string_size_counter_list[3],string_size_counter_list[4],string_size_counter_list[5],string_size_counter_list[6],string_size_counter_list[7],string_size_counter_list[8],string_size_counter_list[9],string_size_counter_list[10],string_size_counter_list[11], 'LIST',list_size_counter_list[0],list_size_counter_list[1],list_size_counter_list[2],list_size_counter_list[3],list_size_counter_list[4],list_size_counter_list[5],list_size_counter_list[6],list_size_counter_list[7],list_size_counter_list[8],list_size_counter_list[9],list_size_counter_list[10],list_size_counter_list[11], 'HASH',hash_size_counter_list[0],hash_size_counter_list[1],hash_size_counter_list[2],hash_size_counter_list[3],hash_size_counter_list[4],hash_size_counter_list[5],hash_size_counter_list[6],hash_size_counter_list[7],hash_size_counter_list[8],hash_size_counter_list[9],hash_size_counter_list[10],hash_size_counter_list[11], 'SET',set_size_counter_list[0],set_size_counter_list[1],set_size_counter_list[2],set_size_counter_list[3],set_size_counter_list[4],set_size_counter_list[5],set_size_counter_list[6],set_size_counter_list[7],set_size_counter_list[8],set_size_counter_list[9],set_size_counter_list[10],set_size_counter_list[11], 'ZSET',zset_size_counter_list[0],zset_size_counter_list[1],zset_size_counter_list[2],zset_size_counter_list[3],zset_size_counter_list[4],zset_size_counter_list[5],zset_size_counter_list[6],zset_size_counter_list[7],zset_size_counter_list[8],zset_size_counter_list[9],zset_size_counter_list[10],zset_size_counter_list[11])def usage(): print '''This Scripts is used for statistic key distribution , Current is Version is Just show key type and key size Distrub!--INPUT : -p,--password= Author pass -P,--port= Redis Port ,Default is 6379 -h,--host= Redis Host ,Default is 127.0.0.1 -H,--help show Scritps Usages !--EXAMPLE: python redis_key_distribution.py -h 服务器IP -P 6379--Sample:--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------统计时间:[Sun Nov 3 17:17:09 2019 ~ Sun Nov 3 17:17:09 2019]Redis服务器[localhost:6379]数据类型和数据大小分布情况如下:---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|KEY TYPE | KEY COUNT | KEY No TTL | KEY 64(byte) | KEY 128 | Key 512 | Key 1024 | Key 2048 | Key 3072 | Key 4096 | Key 5120 | Key 6044 | Key large || String | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 || LIST | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 || HASH | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 || SET | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 || ZSET | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |---------------------------------------------------------------------------------------------------------------------------------------------------- Above shows key type for goods-sort, as you can see most key type is : ZSET; for ZSET: there are 26489 keys size between 512 ~ 1024 byte ! ''' sys.exit()def parse_args(sys_argvs): passwd = '' host = '127.0.0.1' port = 6379 try: opts,args = getopt.getopt(sys_argvs,"HP:h:",["help","password=","port=","host="]) for op,value in opts: if op in ("-p","--password"): password = value elif op in ("-h","--host"): host = value elif op in ("-P","--port"): port = int(value) elif op in ("-H","--help"): usage() except Exception,e: print "Parse Args Error ,%s" % (e) ##//return {'passwd':passwd,'host':host,'port':port,'outputdir':outputdir} return {'passwd':passwd,'host':host,'port':port}''' Main Function '''def main(): if len(sys.argv) < 2: usage() input_args = parse_args(sys.argv[1:]) host = input_args['host'] port = input_args['port'] start_time = time.ctime() rdbConn = redis.Redis(host=host,port=port) init_keys = get_key(rdbConn,0) #keysFile = open(host.replace('.','_')+'_'+str(port)+'.redis','w') keys_without_expire_time_handle = open(host.replace('.','_')+'_'+str(port)+'_without_expiretime.redis','w') while True : for key in init_keys[1] : key_info = get_key_info(rdbConn,key) if key_info: redis_key_static(key_info) #keysFile.write('''"%s"\n''' % (key+','+key_info[0]+','+str(key_info[1]['serializedlength'])+','+str(key_info[2]))) if key_info[2] < 0: keys_without_expire_time_handle.write(key_info[0]+','+key+'\n') init_keys = get_key(rdbConn,init_keys[0]) if init_keys[0] == 0: break end_time = time.ctime() #keysFile.close() keys_without_expire_time_handle.close() show_static_info(host,port,start_time,end_time)if __name__ == "__main__": print ''' Please Wait For Seconds .... ''' main()
实验测试

1、为Python安装Redis库

登陆 后点击Download ZIP下载安装包。

解压并安装:

unzip redis-py-master.zip -d /usr/local cd /usr/local/redis-py-master/python setup.py install 

安装完成后,就可以引用redis库了。

2、执行结果:

python redis_key_distribution.py -h localhost -P 6379

后面会分享更多devops和DBA方面内容,感兴趣的朋友可以关注下~

标签: #python统计学校类型