python批量删除redis key

生产环境中要禁止使用keys *查询key, 因为redis低版本是单线程,如果key非常多的话,直接使用keys *会导致阻塞,所以应当使用scan命令,scan命令介绍请参考其他文档。

python 复制代码
# -*- coding: utf-8 -*-
# @Time    : 2025/01/09
# @Author  : 养乐多
# @Mail    : oct_hao@163.com
# @File    : redis_del_keys.py
# @Use     : 利用scan命令批量删除redis key


import redis


class Redis:
    def __init__(self, host, port, db=0):
        '''
        初始化redis连接
        :param host: redis host
        :param port: redis port
        :param db: redis db
        '''
        self.redis_host = host
        self.redis_port = port
        self.redis_db = db
        try:
            # 连接redis
            self.redis_client = self._connect()
        except Exception as e:
            print('redis连接失败', e)

    def _connect(self):
        '''
        连接redis
        '''
        return redis.StrictRedis(host=self.redis_host, port=self.redis_port, db=self.redis_db)

    def _close(self):
        '''
        关闭redis连接
        '''
        self.redis_client.close()

    def batch_delete_keys(self, pattern, batch_size=1000):
        """
        使用 SCAN 命令批量删除匹配给定模式的 Redis 键
        :param pattern: 要删除的键的模式
        :param batch_size: 每次扫描时的键的批量大小
        """
        # 初始化游标
        cursor = 0 

        while True:
            # 使用 SCAN 命令遍历键
            cursor, keys = self.redis_client.scan(cursor=cursor, match=pattern, count=batch_size)
            # 删除匹配的键
            if keys:
                self.redis_client.delete(*keys)
                print('删除了%s个key' % len(keys))
            else:
                break

            if cursor == 0:
                break

        self._close()
    
    def get_all_key_count(self, pattern, batch_size=1000):
        '''
        使用 SCAN 命令批量删除匹配给定模式的 Redis 键
        :param pattern: 要删除的键的模式
        :param batch_size: 每次扫描时的键的批量大小
        '''
        # 初始化游标
        cursor = 0
        # 初始化计数器
        count = 0  

        while True:
            # 使用 SCAN 命令遍历键
            cursor, keys = self.redis_client.scan(cursor=cursor, match=pattern, count=batch_size)
            # 统计匹配的键的数量
            if keys:
                count += len(keys)
            else:
                break

            if cursor == 0:
                break

        self._close()
        return count

if __name__ == "__main__":
    redis_host = '172.30.11.132'
    redis_port = 6379
    redis_db = 0
    pattern = '*123'
    batch_size = 1000

    # 连接到 Redis
    r = Redis(redis_host, redis_port, redis_db)
    # 查询key总数
    print(r.get_all_key_count(pattern=pattern, batch_size=batch_size))
    # 删除key
    # r.batch_delete_keys(pattern=pattern, batch_size=batch_size)
相关推荐
Blossom.11820 分钟前
把AI“撒”进农田:基于极值量化与状态机的1KB边缘灌溉决策树
人工智能·python·深度学习·算法·目标检测·决策树·机器学习
Red Car20 分钟前
如何向文件夹内所有PDF增加水印
python·pdf
Q_Q51100828530 分钟前
python+uniapp基于微信小程序团购系统
spring boot·python·微信小程序·django·uni-app·node.js·php
java1234_小锋1 小时前
TensorFlow2 Python深度学习 - 循环神经网络(LSTM)示例
python·rnn·深度学习·tensorflow2
爬山算法1 小时前
Redis(69)Redis分布式锁的优点和缺点是什么?
数据库·redis·分布式
测试老哥1 小时前
Postman环境变量设置全攻略
自动化测试·软件测试·python·测试工具·职场和发展·接口测试·postman
惜月_treasure2 小时前
Text2SQL与工作流实现:让数据库查询变得轻松又高效
数据库·人工智能·python
码猩3 小时前
获取dm音视频文案
python
2401_837088503 小时前
Redis通用命令
数据库·redis·缓存
给我起把狙3 小时前
Django与Tornado框架深度对比:从MVCMTV到高并发架构设计
python·django·tornado