Redis数据结构之Set结构详解
作者:難釋懷
一、前言:无序、唯一、高效的集合
在 Redis 的五大数据类型中,Set(集合) 是一个非常独特且强大的存在。它天生具备两个核心特性:
- 元素唯一性:自动去重,确保集合内不存在重复元素。
- 无序性:元素没有固定的顺序(尽管
SMEMBERS返回的顺序是稳定的)。
基于这两个特性,Set 被广泛应用于:
- 标签系统(用户兴趣标签、文章分类)
- 共同好友/关注(
SINTER交集运算) - 抽奖池(
SRANDMEMBER随机抽取) - 全局去重(如爬虫 URL 去重)
但你是否想过,Redis 是如何在底层高效地实现“唯一性”和“快速查找”的?答案就藏在它的两种精巧数据结构中。
核心价值:
Redis Set 的底层会根据数据特征,在 IntSet(整数集合)和 Dict(字典)之间智能切换,以实现内存效率与操作性能的最佳平衡!
本文将带你:
- 拆解 IntSet 的紧凑内存布局
- 揭秘 Dict 如何实现 O(1) 的唯一性校验
- 理解编码转换背后的阈值逻辑
二、Set 的双重身份:IntSet 与 Dict
Redis Set 并非只有一种底层实现,而是拥有两种编码(encoding),由 redisObject 的 encoding 字段决定:
| 编码 (encoding) | 底层数据结构 | 适用场景 |
|---|---|---|
| OBJ_ENCODING_INTSET | IntSet (整数集合) | 所有元素都是整数,且数量较少 |
| OBJ_ENCODING_HT | Dict (字典/哈希表) | 元素包含非整数,或整数数量过多 |
这种设计体现了 Redis “因地制宜” 的优化哲学:对简单、规则的数据用最省空间的结构;对复杂、庞大的数据用最高效的结构。
三、编码一:IntSet - 整数的极致压缩
3.1 诞生背景
当一个 Set 中的所有元素都是整数时,使用通用的哈希表(Dict)来存储显得有些“大材小用”。因为哈希表需要为每个元素存储一个完整的 dictEntry 结构(包含 key, value, next 指针等),内存开销较大。
为了极致节省内存,Redis 引入了 IntSet。
3.2 源码结构
typedef struct intset {
uint32_t encoding; // 编码方式:INTSET_ENC_INT16, INTSET_ENC_INT32, INTSET_ENC_INT64
uint32_t length; // 元素个数
int8_t contents[]; // 柔性数组,存储实际的整数数据
} intset;关键特性:
- 内存连续:所有整数紧密排列在
contents数组中,无任何指针开销。 - 有序存储:内部元素按从小到大排序,为二分查找提供可能。
- 类型升级:
encoding字段决定了每个整数占用的字节数(2/4/8字节)。
3.3 类型升级机制(核心!)
IntSet 最精妙的设计在于其动态类型升级能力。
- 初始状态:插入第一个整数
5,encoding = INTSET_ENC_INT16,每个元素占 2 字节。 - 插入更大整数:当插入一个超出当前
encoding范围的整数(如70000,超过了int16的最大值32767)时,IntSet 会自动将整个集合升级到INTSET_ENC_INT32。
过程:
- 申请一块新的、更大的内存空间。
- 将原有所有元素按新类型(如
int32)重新写入新空间。 - 更新
encoding和length字段。 - 释放旧内存。
注意:这个过程需要 O(N) 的时间复杂度和额外的内存,但只会在必要时发生一次,之后的插入操作又恢复 O(log N)(二分查找+插入)。
3.4 内存优势
假设一个 Set 包含 1000 个 int16 范围内的整数:
- IntSet:
8 (header) + 1000 * 2 = 2008 bytes - Dict:每个
dictEntry至少需要8(key)+8(value)+8(next)= 24字节,加上哈希表本身的桶数组,总内存轻松超过24000+ bytes。
内存节省高达 90% 以上!
四、编码二:Dict - 通用的高性能解决方案
一旦 Set 不再满足 IntSet 的苛刻条件(出现非整数,或整数太多),Redis 会立即将其转换为 Dict(字典)。
4.1 为什么是 Dict?
Dict 是 Redis 的基石数据结构之一,它是一个哈希表,天然具备以下特性:
- O(1) 平均时间复杂度:用于添加 (
SADD)、删除 (SREM)、查找 (SISMEMBER) 操作。 - 天然去重:哈希表的 key 本身就是唯一的,完美契合 Set 的“元素唯一”要求。
4.2 Dict 在 Set 中的特殊用法
在 Set 的场景下,Dict 的使用非常巧妙:
- Key:存储 Set 的元素(字符串或序列化后的整数)。
- Value:统一设置为 NULL 指针。
// 伪代码示意 dict *d = dictCreate(&setDictType, NULL); dictAdd(d, "element1", NULL); dictAdd(d, "element2", NULL);
✅ 优势:这样既利用了 Dict 的高效哈希和唯一性保证,又省去了 Value 的内存开销。
4.3 渐进式 Rehash
Dict 本身也有一套精妙的扩容/缩容机制(渐进式 rehash),确保在数据量巨大时,单次操作的延迟依然很低。这部分内容在此不展开,但它保证了即使 Set 包含百万级元素,性能依然稳定。
五、编码转换:阈值与触发条件
Redis 通过两个配置项来控制 Set 何时从 intset 转换为 hashtable:
| 配置项 | 默认值 | 说明 |
|---|---|---|
| set-max-intset-entries | 512 | 当 Set 中的整数元素数量超过此值时,即使全是整数,也会转换为 Dict。 |
| 隐式条件 | - | 当尝试向一个 intset 编码的 Set 中插入一个非整数值(如字符串)时,会立即触发转换。 |
设计考量:
- 512 这个阈值:是内存效率和操作性能的平衡点。超过 512 个元素后,IntSet 的 O(log N) 查找和 O(N) 的插入(因需移动内存)开销开始显现,而 Dict 的 O(1) 优势则愈发明显。
- 即时转换:保证了数据模型的一致性。一旦数据不再是“纯整数”,就必须切换到通用模型。
六、动手实验:观察 Set 的编码变化
6.1 验证 IntSet
# 添加纯整数 > SADD myset 1 2 3 100 200 (integer) 5 # 查看编码 > OBJECT ENCODING myset "intset"
6.2 触发转换:插入非整数
# 插入一个字符串 > SADD myset "hello" (integer) 1 # 编码已变为 hashtable > OBJECT ENCODING myset "hashtable"
6.3 触发转换:超过阈值
# 创建一个脚本,添加513个整数
> for i in {1..513}; do redis-cli SADD big_intset $i; done
# 查看编码(应为 hashtable)
> OBJECT ENCODING big_intset
"hashtable"七、总结
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。
