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

img
img
img

既有适合小白学习的零基础资料,也有适合3年以上经验的小伙伴深入学习提升的进阶课程,涵盖了95%以上Go语言开发知识点,真正体系化!

由于文件比较多,这里只是将部分目录截图出来,全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频,并且后续会持续更新

如果你需要这些资料,可以戳这里获取

encoding类型字节
INTSET_ENC_INT16int16_t2
INTSET_ENC_INT32int32_t4
INTSET_ENC_INT64int64_t8

length : 集合元素大小,时间复杂度由O(n)->O(1)。
contents : 元素数组。

操作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;
    }
}

采用 “二分法”的算法查找搜索

img
img
img

既有适合小白学习的零基础资料,也有适合3年以上经验的小伙伴深入学习提升的进阶课程,涵盖了95%以上Go语言开发知识点,真正体系化!

由于文件比较多,这里只是将部分目录截图出来,全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频,并且后续会持续更新

如果你需要这些资料,可以戳这里获取

img-kQLzLGch-1715909895497)]
[外链图片转存中…(img-J1jhpXPA-1715909895498)]

既有适合小白学习的零基础资料,也有适合3年以上经验的小伙伴深入学习提升的进阶课程,涵盖了95%以上Go语言开发知识点,真正体系化!

由于文件比较多,这里只是将部分目录截图出来,全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频,并且后续会持续更新

如果你需要这些资料,可以戳这里获取

  • 5
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值