Redis6种底层数据结构之整数集合

原创已于 2024-10-30 21:26:36 修改 · 2.1k 阅读

24 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构 #算法 #redis

于 2024-10-30 21:23:47 首次发布

Redis底层数据结构专栏收录该内容

5 篇文章

订阅专栏

1.1 什么是整数集合

整数集合是集合键（也就是SET）的底层实现之一，当一个集合只包含整数值元素，且集合的元素数量不多时，Redis就会使用整数集合作为集合键的底层实现

1.2 结构

typedef struct intset {
    //编码方式
    uint32_t encoding;

    //集合包含的元素数量
    uint32_t length;

    //保存元素的数组
    int8_t contents[];
} intset;

虽然intset结构将contents属性声明为int8_t类型的数组，但实际上contents数组并不保存任何int8_t类型的值，保存的类型取决于encoding属性的值
length属性记录了整数集合包含的元素数量，也即是contents数组的长度
数组是整数集合的底层实现：整数集合中的每个元素都是数组中的一个数组项，各个项在数组中的按值的大小从小到大排列，且没有任何重复项。
如果encoding属性的值为INTSET_ENC_INT16，那么contents就是一个int16_t类型的数组，数组里的每个项都是一个int16_t类型的整数值（最小值为-32768，最大值为32767）
如果encoding属性的值为INTSET_ENC_INT32，那么contents就是一个int32_t类型的数组，数组里的每个项都是一个int32_t类型的整数值（最小值为-2147483648，最大值为2147483647）
如果encoding属性的值为INTSET_ENC_INT64，那么contents就是一个int64_t类型的数组，数组里的每个项都是一个int64_t类型的整数值（最小值为-9223372036854775808，最大值为9223372036854775807）

1.3 升级

每当将一个新元素添加到整数集合里，并且新元素的类型比整数集合现有所有元素的类型都要长时，整数集合要先进行升级，然后才能将新元素添加到整数集合中。比如向一个底层为int16_t数组的整数集合中添加一个int64_t类型的整数值时，整数集合已有的所有元素都会被转化为int64_t类型。复杂度为O(N)
升级整数集合并添加新元素的过程如下：
1）根据新元素的类型，扩展整数集合底层数组的空间大小，并为新元素分配空间
2）将底层数组现有的所有元素都转换成与新元素相同的类型，并将类型转换后的元素放置到正确的位置上，而且在放置元素的过程中，需要继续维持底层数组的有序性质不变
3）将新元素添加到底层数组里面
举个例子，现在有一个INTSET_ENC_INT16编码的整数集合，集合中包含三个int16_t类型的元素，如下图所示：
在这里插入图片描述

因为每个元素都占用16位空间，所以整数集合底层数组的大小为3*16=48位
在这里插入图片描述
现在要将类型为int32_t的整数值65535添加到整数集合里面，因为65535的类型int32_t比整数集合当前所有元素的类型都要长，所以添加元素之前，需要先对整数集合进行升级，过程如下：
1）扩展整数集合底层数组的大小，因为每个int32_t整数值需要占用32位空间，在空间重分配之后，底层数组的大小将是32 * 4 = 128位。此时元素1、2、3仍然是int16_t类型，依旧保存在数组的前48位里面。
在这里插入图片描述

2）将3个元素转换成int32_t类型，并将转换后的元素放置到正确的位上面，在放置元素的过程中，需要维持底层数组的有序性质不变
首先元素3在4个元素中排名第三，所以它将被移动到contents数组的索引2位置上，也就是数组64位至95位的空间内，如下图所示：
在这里插入图片描述
接着元素2在4个元素中排名第二，它将被移动到contents数组索引1的位置上，也就是数组的32位至63位的空间内，如下图所示：

接着元素1在4个元素中排名第一，它将被移动到contents数组索引0的位置上，也就是数组的0位至31位的空间内，如下图所示：
在这里插入图片描述

3）最后，元素65535在4个元素中排名第四，所以它将被添加到contents数组的索引3位置上，也就是数组的96位至127位的空间内，如下图所示：
在这里插入图片描述

最后将整数集合encoding属性的值从INTSET_ENC_INT16改为INTSET_ENC_INT32，并将length属性值从3改成4，设置完后的整数集合如下图所示：
在这里插入图片描述

每次向整数集合中添加新元素都可能会引起升级，而每次升级都需要对底层数组中已有的所有元素进行类型转换，所以添加新元素的时间复杂度为O(n) 。
虽然升级会浪费内存，但是整数集合可以让集合同时保存3种类型的值，很灵活，且升级操作只会在有需要的时候进行，这可以尽量节省内存。
最后：升级之后是无法降级的。所以只支持升级，不支持降级