Redis获取某个前缀的key脚本实例

1.背景

在平时的维护中，经常会遇到要统计某个前缀的key有多少，在请求比较多的redis中，keys * 会直接导致阻塞。
可以采用scan的方式进行增量迭代，查询使用pipeline减少交互，提高效率。

2.scan命令的优缺点

SCAN命令的有SCAN,SSCAN,HSCAN,ZSCAN。

SCAN的话就是遍历所有的keys

其他的SCAN命令的话是SCAN选中的集合。

SCAN命令是增量的循环，每次调用只会返回一小部分的元素。所以不会有KEYS命令的坑。

SCAN命令返回的是一个游标，从0开始遍历，到0结束遍历。

scan 0
1) "655"
2) 1) "test1"
  2) "test2"

返回值一个array，一个是下次循环的cursorId,一个是元素数组。SCAN命令不能保证每次返回的值都是有序的，另外同一个key有可能返回多次，不做区分，需要应用程序去处理。

另外SCAN命令可以指定COUNT,默认是10。但是这个并不是指定多少，就能返回多少，这只是一个提示，并不能保证一定返回这么多条。

优点：

• 提供键空间的遍历操作，支持游标，复杂度O(1), 整体遍历一遍只需要O(N)；
• 提供结果模式匹配；
• 支持一次返回的数据条数设置，但仅仅是个hints，有时候返回的会多；
• 弱状态，所有状态只需要客户端需要维护一个游标；

缺点：

• 无法提供完整的快照遍历，也就是中间如果有数据修改，可能有些涉及改动的数据遍历不到；
• 每次返回的数据条数不一定，极度依赖内部实现；
• 返回的数据可能有重复，应用层必须能够处理重入逻辑；

3. python脚本的实现

python中有一个封装的函数scan_iter--查看所有元素--迭代器

脚本内容：

#!/usr/bin/env python
# -*- coding: UTF-8 -*- 
#作用：统计某个前缀key的个数，并将其输入到文件
#使用方法：python scan_redis.py apus* 100
__author__ = "lcl" 
import sys
import redis 
import os 
pool=redis.ConnectionPool(host='192.168.225.128',port=6379,db=0) 
r = redis.StrictRedis(connection_pool=pool) 
#扫描匹配值，通过sys.argv传参
match = sys.argv[1]
#每次匹配数量
count = sys.argv[2]
#print match
#print count
#总数量
total = 0
#扫描到的key输出到文件
path = os.getcwd()
#扫描到的key输出的文件
txt = path+"/keys.txt"
f = open(txt,"w")
for key in r.scan_iter(match = match,count = count):
# f.write("%s %s" % (key,"\n"))
 f.write(key+"\n")
 total = total+1
f.close
print "匹配: %s 的数量为:%d " % (match,total)

总结