redis源码阅读—intset(整数集合)_redis _intsetvalueencoding(3)

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

需要这份系统化的资料的朋友，可以添加戳这里获取

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！

操作API

创建

创建一个整型集合，指定默认编码为INTSET_ENC_INT16

intset \*intsetNew(void) {
    intset \*is = zmalloc(sizeof(intset));
    //编码
    is->encoding = intrev32ifbe(INTSET_ENC_INT16);
    is->length = 0;
    return is;
}

插入

插入源码

将新元素添加至集合，成功返回 1，失败返回0。

/\* Insert an integer in the intset \*/
intset \*intsetAdd(intset \*is, int64_t value, uint8_t \*success) {
    uint8_t valenc = \_intsetValueEncoding(value);
    uint32_t pos;
    if (success) \*success = 1;
	// valenc 新元素适合的存储编码
	// is->encoding 当前集合的编码
	// 当valenc > is->encoding 表明当前集合无法存储新元素 那么此时需要对集合进行升级 反之检查集合是否存在该元素
    /\* Upgrade encoding if necessary. If we need to upgrade, we know that
 \* this value should be either appended (if > 0) or prepended (if < 0),
 \* because it lies outside the range of existing values. \*/
    if (valenc > intrev32ifbe(is->encoding)) {
        /\* This always succeeds, so we don't need to curry \*success. \*/
        return intsetUpgradeAndAdd(is,value);
    } else {
        /\* Abort if the value is already present in the set.
 \* This call will populate "pos" with the right position to insert
 \* the value when it cannot be found. \*/
        if (intsetSearch(is,value,&pos)) {
            if (success) \*success = 0;
            return is;
        }

        is = intsetResize(is,intrev32ifbe(is->length)+1);
        if (pos < intrev32ifbe(is->length)) intsetMoveTail(is,pos,pos+1);
    }

    \_intsetSet(is,pos,value);
    is->length = intrev32ifbe(intrev32ifbe(is->length)+1);
    return is;
}

intsetAdd() 代码流程讲解

首先， 通过调用_intsetValueEncoding 得到value合适的编码valenc。

#define INTSET\_ENC\_INT16 (sizeof(int16\_t))
#define INTSET\_ENC\_INT32 (sizeof(int32\_t))
#define INTSET\_ENC\_INT64 (sizeof(int64\_t))
/\* Return the required encoding for the provided value. \*/
static uint8_t \_intsetValueEncoding(int64_t v) {
    if (v < INT32_MIN || v > INT32_MAX)   //小于32位的最小值 大于32位的最大值
        return INTSET_ENC_INT64; // 64位
    else if (v < INT16_MIN || v > INT16_MAX) //小于1616位的最小值 大于位的最大值
        return INTSET_ENC_INT32; // 32位
    else
        return INTSET_ENC_INT16;  //16位
}

其次， 比较新元素的编码valenc与 当前集合的编码is->encoding。

1). valenc>is->encoding ： 表明当前集合无法存储新元素，需对集合进行升级。
2). valenc<=is->encoding：此时，集合无需升级。优先判断新元素是否存在。若存在，则return ； 反之，扩充集合空间，为新元素腾出位置，最后添加新元素.

intsetAdd 内部调用多个函数，其中intsetUpgradeAndAdd、intsetSearch、intsetMoveTail 尤为关键，在整个intset相关操作中，起到“承上启下”的作用，能突出地体现intset的设计思想与灵魂。

升级源码

是对集合进行升级，并添加新元素

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);
    //移动集合元素
    while(length--)
        \_intsetSet(is,length+prepend,\_intsetGetEncoded(is,length,curenc));
    //prepend = 1：表明value是最小的负数，将新元素添加到集合首位。
    //prepend = 0：表明value是最大的整数，将新元素添加至集合尾部。
    if (prepend)
        \_intsetSet(is,0,value);
    else
        \_intsetSet(is,intrev32ifbe(is->length),value);
    //设置长度
    is->length = intrev32ifbe(intrev32ifbe(is->length)+1);
    return is;
}

执行流程：扩充集合空间（intsetResize）->集合元素设置新编码（_intsetSet）->向集合头/尾部添加新元素（_intsetSet）。

/\* Resize the intset \*/
static intset \*intsetResize(intset \*is, uint32_t len) {
    uint32_t size = len\*intrev32ifbe(is->encoding);
    is = zrealloc(is,sizeof(intset)+size);
    return is;
}

/\* Set the value at pos, using the configured encoding. \*/
static void \_intsetSet(intset \*is, int pos, int64_t value) {
    uint32_t encoding = intrev32ifbe(is->encoding);

    if (encoding == INTSET_ENC_INT64) {
        ((int64_t\*)is->contents)[pos] = value;
        memrev64ifbe(((int64_t\*)is->contents)+pos);
    } else if (encoding == INTSET_ENC_INT32) {
        ((int32_t\*)is->contents)[pos] = value;
        memrev32ifbe(((int32_t\*)is->contents)+pos);
    } else {
        ((int16_t\*)is->contents)[pos] = value;
        memrev16ifbe(((int16_t\*)is->contents)+pos);
    }
}

新元素的位置是由 prepend 变量控制。而prepend是根据 value 值定义的。
当value<0 时，表明 value 小于当前集合所有元素 , 系统将 value 作为集合第一个元素。
当value>=0时, 表明 value 大于当前集合所有元素 , 系统将 value 添加到集合的尾部。

查找源码

查找集合元素，判断value是否已存在。若元素存在，则返回1; 反之, 返回0。

static uint8_t intsetSearch(intset \*is, int64_t value, uint32_t \*pos) {
    int min = 0, max = intrev32ifbe(is->length)-1, mid = -1;
    int64_t cur = -1;
    if (intrev32ifbe(is->length) == 0) {
        if (pos) \*pos = 0;
        return 0;
    } else {
        if (value > \_intsetGet(is,intrev32ifbe(is->length)-1)) {
            //大于集合中最大的数
            if (pos) \*pos = intrev32ifbe(is->length);
            return 0;
        } else if (value < \_intsetGet(is,0)) {
            //小于集合中最小的数
            if (pos) \*pos = 0;
            return 0;
        }
    }
    //采用二分查找算法进行搜索
    while(max >= min) {
        mid = ((unsigned int)min + (unsigned int)max) >> 1;
        cur = \_intsetGet(is,mid);
        if (value > cur) {
            min = mid+1;
        } else if (value < cur) {
            max = mid-1;
        } else {
            break;
        }
    }
    if (value == cur) {
        //找到相同的元素
        if (pos) \*pos = mid;
        return 1;
    } else {
        //未找到
        if (pos) \*pos = min;
        return 0;
    }
}

采用 “二分法”的算法查找搜索
因此，可以推断出,intset是一种有序的集合。当元素存在时，pos 代表该元素所在的集合下标。当元素不存在时，pos 表示该元素添加的位置。

移动源码

static void intsetMoveTail(intset \*is, uint32_t from, uint32_t to) {
    void \*src, \*dst;
    uint32_t bytes = intrev32ifbe(is->length)-from;
    uint32_t encoding = intrev32ifbe(is->encoding);


![img](https://img-blog.csdnimg.cn/img_convert/88617824b0920b29a1c690feb7105c45.png)
![img](https://img-blog.csdnimg.cn/img_convert/e05ba11e7c983c41971173e38c31a88a.png)

**网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。**

**[需要这份系统化的资料的朋友，可以添加戳这里获取](https://bbs.csdn.net/topics/618658159)**


**一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！**

外链图片转存中...(img-4oHUK298-1715909860833)]

**网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。**

**[需要这份系统化的资料的朋友，可以添加戳这里获取](https://bbs.csdn.net/topics/618658159)**


**一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！**