Redis整型数组Intset升级源码解析

在看Redis设计于实现的时候看到了一句话：
由于引发升级的新元素长度总是比整数集合现有的所有元素的长度都大，因此这个新元素的值要么就大于所有现有元素，要么就小于所有现有元素。
我的推断是：当插入数字的时候，如果这个数字小于int16那么就会给这个值类型设置为int16，如果大于int 16小于 int32就会设置为int 32,所以才会出现上述情况。但是觉得还是看源码靠谱，所以看看源码吧。

//Reids升级的源码
static intset *intsetUpgradeAndAdd(intset *is, int64_t value) {
	//以前的编码方式
    uint8_t curenc = intrev32ifbe(is->encoding);
    //通过 _intsetValueEncoding 判断现在的编码格式
    uint8_t newenc = _intsetValueEncoding(value);
    int length = intrev32ifbe(is->length);
    int prepend = value < 0 ? 1 : 0;

    /* First set new encoding and resize */
    is->encoding = intrev32ifbe(newenc);
    //让is指向新分配的空间
    is = intsetResize(is,intrev32ifbe(is->length)+1);

    /* Upgrade back-to-front so we don't overwrite values.
     * Note that the "prepend" variable is used to make sure we have an empty
     * space at either the beginning or the end of the intset. */
    while(length--)
        _intsetSet(is,length+prepend,_intsetGetEncoded(is,length,curenc));

    /* Set the value at the beginning or the end. */
    if (prepend)
    	//如果小于0，那么就放在队头
        _intsetSet(is,0,value);
    else
    	//最大的话就放在队尾
        _intsetSet(is,intrev32ifbe(is->length),value);
        //设置长度加1
    is->length = intrev32ifbe(intrev32ifbe(is->length)+1);
    return is;
}

/* Return the required encoding for the provided value. */
//判断v的编码格式，从这里可以看到我的猜想是正确的
static uint8_t _intsetValueEncoding(int64_t v) {
    if (v < INT32_MIN || v > INT32_MAX)
        return INTSET_ENC_INT64;
    else if (v < INT16_MIN || v > INT16_MAX)
        return INTSET_ENC_INT32;
    else
        return INTSET_ENC_INT16;
}

/* Resize the intset */
//重新分配指定大小的空间
static intset *intsetResize(intset *is, uint32_t len) {
    uint64_t size = (uint64_t)len*intrev32ifbe(is->encoding);
    assert(size <= SIZE_MAX - sizeof(intset));
    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);
    }
}