整数集合( intset) 是集合键的底层实现之一,当一个集合只包含整数值元素,并且这个集合的元素数量不多时, Redis 就会使用整数集合作为集合键的底层实现。
整数集合 (intset) 是 Redis 用于保存整数值的集合抽象数据结构,它可以保存类型为int16 、int32 或者 int64 的整数值,并且保证集合中不会出现重复元素。
数据结构
typedef struct intset {
//编码方式
uint32_t encoding;
//集合元素数量
uint32_t length;
//元素数组
int8_t contents[];
} intset;
contents 数组是整数集合的底层实现:整数集合的每个元素都是 contents 数组的一个数组项,各个项在数组中按值的大小从小到大有序地排列,并且数组中不包含任何重复项。
length 属性记录了整数集合包含的元素数量,也即是 contents 数组的长度。
虽然 intset 结构将 contents 属性声明为 int8 类型的数组,但实际上 contents 数组并不保存任何 int8 类型的值,contents 数组的真正类型取决于 encoding 属性的值
数组升级
每当我们要将一个新元素添加到整数集合里面,并且新元素的类型比整数集合现有所有 元素的类型都要长时,整数集合需要先进行升级 (upgrade ),然后才能将新元素添加到整数 集合里面。
整数集合的升级策略好处,一个是提升整数集合的灵活性,另一个是尽可能地节约内存。
升级整数集合并添加新元素共分为三步进行:
1 根据新元素的类型,扩展整数集合底层数组的空间大小,并为新元素分配空间。
2 将底层数组现有的所有元素都转换成与新元素相同的类型,并将类型转换后的元素 放置到正确的位上,而且在放置元素的过程中,需要继续维持底层数组的有序性质不变。
3 将新元素添加到底层数组里面
//升级代码
static intset *intsetUpgradeAndAdd(intset *is, int64_t value) {
uint8_t curenc = intrev32ifbe(is->encoding);
uint8_t newenc = _intsetValueEncoding(value);
int length = intrev32ifbe(is->length);
int prepend = value < 0 ? 1 : 0;
//设置新的编码,分配升级后需要使用的内存
is->encoding = intrev32ifbe(newenc);
is = intsetResize(is,intrev32ifbe(is->length)+1);
//从后往前移动原先的值到正确的位置,这样不会覆盖原先的值,prepend判断元素插入位置时第一位还是最后一位,因为引起升级,说明插入的值要么最大,要么最小
while(length--)
_intsetSet(is,length+prepend,_intsetGetEncoded(is,length,curenc));
//将插入的值放入正确的位置
if (prepend)
_intsetSet(is,0,value);
else
_intsetSet(is,intrev32ifbe(is->length),value);
is->length = intrev32ifbe(intrev32ifbe(is->length)+1);
return is;
}