redis中整数聚合的add操作

以下是inset的add操作

intset *intsetAdd(intset *is, int64_t value, uint8_t *success) {
    /* value的字节数 */
    uint8_t valenc = _intsetValueEncoding(value);
    uint32_t pos;
    if (success) *success = 1;

    if (valenc > intrev32ifbe(is->encoding)) {
        /* 升级 */
        return intsetUpgradeAndAdd(is,value);
    } else {
        /* 检查value是否已经存在 */
        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);
    }
    /* 将value放入is中的contents[pos]上 */
    _intsetSet(is,pos,value);
    /* 更新length属性 */
    is->length = intrev32ifbe(intrev32ifbe(is->length)+1);
    return is;
}

add操作中可能会涉及到“升级”操作。

什么情况下进行升级？当新增的元素的类型比现在contents数组中的所有元素的类型都长的时候，需要先将inset升级，然后添加新元素。

来看看升级的具体过程：

static intset *intsetUpgradeAndAdd(intset *is, int64_t value) {
    uint8_t curenc = intrev32ifbe(is->encoding);
    uint8_t newenc = _intsetValueEncoding(value);
    int length = intrev32ifbe(is->length);
    /* 用于确保在intset的开头或结尾都有一个空白 */
    int prepend = value < 0 ? 1 : 0;

    /* 设置新的encoding */
    is->encoding = intrev32ifbe(newenc);
    /* resize操作 */
    is = intsetResize(is,intrev32ifbe(is->length)+1);

    /* 从后往前遍历 */
    while(length--)
        /* 赋值 */
        _intsetSet(is,length+prepend,_intsetGetEncoded(is,length,curenc));

    /* 在头部添加元素 */
    if (prepend)
        _intsetSet(is,0,value);
    else
    /* 在尾部添加元素 */
        _intsetSet(is,intrev32ifbe(is->length),value);
    /* 更新length属性 */
    is->length = intrev32ifbe(intrev32ifbe(is->length)+1);
    return is;
}

我们来看看这里的intsetResize方法：很简单，就是扩展intset的空间

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

然后将旧的元素放到新的intset中去，这里是从后往前遍历，避免了有值被覆盖。

以下的_intsetSet方法，在

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);
    }
}

升级操作使得整数集合可以放置不同类型的整数且不用担心出现类型错误；在add操作时动态的升级，节约了内存，提高了灵活性。

intset并没有降级操作。