Redis源码sds学习四-sdscpy和sdsdup

Redis源码sds学习四-sdscpy和sdsdup

在sds动态字符串学习二中，学习了sdscat函数，该函数是字符串拼接函数，C语言中是strcat函数是字符串拼接函数。字符串操作中很多函数，其中在C语言中，还有个常用的字符串拷贝strcpy函数，今天来学习一下redis sds中对应的字符串拷贝函数：sdscpy和sdsdup。

题外话：

学习了几个redis sds字符串操作函数，而C语言中字符串操作函数命名一般为 str*，sds中命名一般为 sds*，*部分对应是一样的，不知道后续其它动态字符串函数是否为此规律，有待验证。

1. sdscpy函数

sdscpy函数参数与sdscat一致，s为源字符串首地址，t为要拷贝覆盖s的字符串，作用是用一个串覆盖掉另外一个串。

sds sdscpy(sds s, const char *t) {
    return sdscpylen(s, t, strlen(t));
}

sdscpylen函数在sdscpy的参数基础上，加上了字符串t的长度值：

sds sdscpylen(sds s, const char *t, size_t len) {
    struct sdshdr *sh = (void*) (s-(sizeof(struct sdshdr)));

    // sds 现有 buf 的长度
    size_t totlen = sh->free+sh->len;

    // 如果 s 的 buf 长度不满足 len ，那么扩展它
    if (totlen < len) {
        s = sdsMakeRoomFor(s,len-sh->len);
        if (s == NULL) return NULL;
        sh = (void*) (s-(sizeof(struct sdshdr)));
        totlen = sh->free+sh->len;
    }

    // 复制内容
    memcpy(s, t, len);

    // 添加终结符号
    s[len] = '\0';

    // 更新属性
    sh->len = len;
    sh->free = totlen-len;

    // 返回新的 sds
    return s;
}

先计算出源字符串 s 的存储区的总长度，至于相关指针和长度的关系，为何要减去 (sizeof(struct sdshdr)），请参看已发文章Redis sds动态字符串学习三,该文章有图片详细给出相关指针的关系图示，有助于理解上述代码。

s的总长度包括s串长度和空闲free的值，两者之和即为 totlen，如果该值小于 t 串的长度，则就要为 s 扩展存储区，以便能够存下t的内容，调用 sdsMakeRoomFor函数，在学习三文章中，讲到过该函数，其内会调用remalloc函数，同时还会对redis的 used_memory值进行修改等操作。

如果总长度 totlen 大于 t 的长度，则直接调用c库函数 memcpy，用t串覆盖掉s串，即从 sh->buf存储区开始填充t串的内容。记的末尾加’\0’，并修改sh的len和free值。

2. sdsdup函数

sdsdup 函数从字面理解，应该是duplicate的头几个字符，复制的意思，就是把现有的串复制拷贝一份，生成原有串的一个副本。据以往文章可以猜到，副本要保存，肯定事先要申请对应的存储区来存储。

参数s为要复制的源字符串，返回值sds为接收s串副本的存储区地址。

sds sdsdup(const sds s) {
    return sdsnewlen(s, sdslen(s));
}

sdsnewlen在前文中记录过，详见Redis sds动态字符串学习一

sds sdsnewlen(const void *init, size_t initlen) {
    struct sdshdr *sh;

    if (init) {
        sh = zmalloc(sizeof(struct sdshdr)+initlen+1); //8 + 3 + 1
    } else {
        sh = zcalloc(sizeof(struct sdshdr)+initlen+1);
    }
    if (sh == NULL) return NULL;

    sh->len = initlen;
    sh->free = 0;
    if (initlen && init)
        memcpy(sh->buf, init, initlen);

    sh->buf[initlen] = '\0';
    return (char*)sh->buf;
}

该函数中调用zmalloc重新分配了存储区，并将串的内容拷贝到新存储sh->buf中，最后返回sh->buf的地址，即char *地址也即 sds，因为typedef char *sds;。调用该函数也必须对应有char *指针接收该地址。

上面就是sds模块中的拷贝函数sdscpy和sdsdup，前者是将一个串拷贝覆盖掉另一个串；后者是拷贝串，生成一个该串的副本。

本人才疏学浅，理解水平有限，如果有错误及不当之处，请批评指正。

如果对您有一点点帮助，请帮忙转发和关注工作号：hongmaolinux，感激不尽!