python使用redis的scan命令实现模糊查询

一、需求

在使用 redis 查询某个 key 的值时，我们可能不记得完整的 key 名称，或者说想要模糊查询满足某些条件的 key 值集合，可以使用scan 命令完成（"keys *"这种方式不被推荐且有性能隐患，生产环境应该禁用）。

二、代码示例

# -*- coding: UTF-8 -*-
"""
# 使用scan命令模糊查询对应的key
"""
import redis

# redis环境地址：IP、端口、密码
REDIS_ADDRESS_QA = "10.xx.xx.xx"
REDIS_PORT_QA = "xxxx"
REDIS_PASSWD_QA = "xxxx"

# redis连接
r = redis.StrictRedis(host=REDIS_ADDRESS_QA, port=REDIS_PORT_QA, password=REDIS_PASSWD_QA)

cursor = 0

# 模糊查询的关键词
pattern = "*user*"

while True:
    # scan会返回两个结果，一个是用于下次遍历的游标，一个是结果集
    cursor, keys = r.scan(cursor, match=pattern, count=10000)
    
    # 只打印非空的key
    if keys:
        print("keys: {}".format(keys))

    # 如果游标返回0, 代表遍历结束
    if cursor == 0:
        break