如何利用Redis Lua脚本封装复杂的HyperLogLog合并_实现原子性的去重统计

秋萱吖_2835

秋萱吖_2835

2026-05-28

266人浏览

原创

直接用pfcount+pfmerge无法满足原子性去重统计需求,因两步执行间可能被并发修改导致结果不一致;redis事务不能保证中间状态隔离,而lua脚本可原子执行合并与统计。

如何利用redis lua脚本封装复杂的hyperloglog合并_实现原子性的去重统计

为什么直接用 PFCOUNT + PFMERGE 无法满足原子性去重统计需求

当你需要对多个 key 的 HyperLogLog 数据做「先合并、再统计」,又要求整个过程不可被并发打断时,PFMERGE 和 PFCOUNT 分两步执行是危险的。中间可能有其他客户端插入新数据、甚至覆盖目标 key,导致结果不一致。Redis 的事务(MULTI/EXEC)也不能保证 PFMERGE 后立刻 PFCOUNT 的原子性——因为 EXEC 只保证命令排队执行,不保证中间状态对外不可见;而 Lua 脚本在 Redis 单线程中运行,天然具备原子性。

用 Lua 脚本封装 PFMERGE + PFCOUNT 的正确写法

核心是把合并与计数压进一个脚本,且避免副作用(比如意外修改源 key)。注意:Lua 中不能直接调用 PFMERGE 的返回值,必须显式用 redis.call("PFMERGE", ...) 执行合并,再单独调用 redis.call("PFCOUNT", ...) 获取结果。

  • 脚本接收至少两个参数:KEYS[1] 作为临时合并目标(建议用带前缀的随机名或 ARGV[1] 指定),KEYS[2..n] 是待合并的原始 HLL key
  • 务必检查 KEYS 数量 ≥ 2,否则 PFMERGE 会报错 ERR wrong number of arguments
  • 不要复用业务 key 做临时目标(如直接用 KEYS[1] = "user:active:hll"),否则会污染原始数据;应生成唯一临时 key,用 redis.call("DEL", tmp_key) 清理(或依赖 TTL)
  • 示例脚本:
local tmp_key = KEYS[1]
redis.call("DEL", tmp_key)
redis.call("PFMERGE", tmp_key, unpack(KEYS, 2, #KEYS))
local count = redis.call("PFCOUNT", tmp_key)
redis.call("DEL", tmp_key)  -- 立即清理,避免残留
return count

调用方式:redis-cli --eval hll_merge_count.lua tmp:hll , user:hll:202401 user:hll:202402 user:hll:202403

Redis Skill - 高性能缓存管理
Redis Skill - 高性能缓存管理

Redis 缓存和数据结构管理技能。通过自然语言操作 Redis,支持 String、Hash、List、Set、ZSet、Stream 等数据结构操作。当用户提到 Redis、缓存、消息队列、会话存储时使用此技能。

下载

PFMERGE 在 Lua 中的坑:unpack 与空 key 处理

如果传入的某个 KEYS[i] 实际不存在(即对应 key 在 Redis 中未创建),PFMERGE 会静默忽略它——这本身是正常行为,但容易误以为数据丢了。更危险的是,当 KEYS 只有一个(即只有目标 key,无源 key),unpack(KEYS, 2, #KEYS) 返回空,PFMERGE 就变成 PFMERGE tmp_key,触发错误。

  • 安全做法:在脚本开头加校验 if #KEYS
  • 若需容忍空源 key,可先用 redis.call("EXISTS", key) 过滤出真实存在的 key,再构造新数组传给 PFMERGE
  • Redis 7.0+ 支持 PFDEBUG,但 Lua 中不可用;调试时可用 redis.log(redis.LOG_WARNING, "merged keys: ", #KEYS) 输出日志(仅开发环境)

性能和内存要注意的边界情况

HyperLogLog 单个 key 最大内存约 12KB,但 PFMERGE 是浅合并(只合并寄存器),不产生新内存分配;真正耗资源的是后续 PFCOUNT 的估算计算——它本身是 O(1),但若合并后基数极大(>1e9),误差率仍可控(标准误差 0.81%),无需担心精度崩坏。

  • 高频调用该脚本时,避免每次生成新临时 key —— 可复用固定 key(如 hll:merge:tmp),但必须加锁或确保单线程调用,否则并发写会冲突
  • 如果原始 HLL key 数量动态变化(比如从 ZSET 或 SCAN 结果中获取),应在客户端拼好 KEYS 列表再传入,不要在 Lua 里做 SCAN(Lua 不支持 SCAN)
  • 脚本总执行时间应 BUSY 错误,此时需改用异步方案(如先 PFMERGE 到持久化 key,再另起任务 PFCOUNT)

临时 key 的生命周期管理最容易被跳过——不是所有团队都记得加 DEL,也不是所有场景都适合设 TTL;漏掉这步,积累的临时 key 会悄悄吃掉内存。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

redis

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
常用的数据库软件
常用的数据库软件

常用的数据库软件有MySQL、Oracle、SQL Server、PostgreSQL、MongoDB、Redis、Cassandra、Hadoop、Spark和Amazon DynamoDB。更多关于数据库软件的内容详情请看本专题下面的文章。php中文网欢迎大家前来学习。

2023.11.02

4389

19

内存数据库有哪些
内存数据库有哪些

内存数据库有Redis、Memcached、Apache Ignite、VoltDB、TimesTen、H2 Database、Aerospike、Oracle TimesTen In-Memory Database、SAP HANA和ache Cassandra。更多关于内存数据库相关问题,详情请看本专题下面的文章。php中文网欢迎大家前来学习。

2023.11.14

3915

11

mongodb和redis哪个读取速度快
mongodb和redis哪个读取速度快

redis 的读取速度比 mongodb 更快。原因包括:1. redis 使用简单的键值存储,而 mongodb 存储 json 格式的数据,需要解析和反序列化。2. redis 使用哈希表快速查找数据,而 mongodb 使用 b-tree 索引。因此,redis 在需要高性能读取操作的应用程序中是一个更好的选择。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.04.02

7132

6

redis怎么做缓存服务器
redis怎么做缓存服务器

redis 作为缓存服务器的答案:redis 是一款开源、高性能、分布式的键值存储,可作为缓存服务器使用。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.04.07

643

6

redis怎么解决数据一致性
redis怎么解决数据一致性

redis 提供了两种一致性模型,以维护副本数据一致性:强一致性 (sync) 确保写操作仅在复制到所有从节点后才完成;最终一致性 (async) 则在主节点上写操作后认为已完成,牺牲一致性换取性能。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.04.07

756

6

mysql和redis怎么保证双写一致性
mysql和redis怎么保证双写一致性

确保 mysql 和 redis 双写一致性的技术包括:1、事务性更新:同时更新 mysql 和 redis,保证一致性;2、主从复制:mysql 主服务器更改同步到 redis 从服务器;3、基于事件的更新:mysql 记录更改并发送到 redis等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.04.07

6862

6

redis缓存一般存些什么数据
redis缓存一般存些什么数据

redis缓存中存储的数据类型包括:字符串、哈希、列表、集合、有序集合、位图、地理空间数据和hyperloglog。这些数据类型适用于存储各种数据,从简单信息到复杂对象和地理位置。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.04.07

1160

6

redis的8种数据类型有哪些
redis的8种数据类型有哪些

redis 提供 8 种数据类型:字符串(文本、数字、二进制)、哈希(键值对)、列表(有序集合)、集合(无序唯一元素)、有序集合(按分数排序)、地理空间(地理位置)、hyperloglog(估计大数据基数)和位图(位序列存储)。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.04.07

1056

6

redis主要作用有哪些
redis主要作用有哪些

redis 的主要作用包括:1. 缓存数据,提高访问速度;2. 充当消息队列,实现消息传递;3. 存储各种数据类型,如字符串、散列和集合;4. 管理会话信息,确保可靠性和可用性;5. 限制请求速率,防止服务器超载等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.04.07

6058

6

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
phpEnv手册
phpEnv手册

共0课时 | 0人学习