redis6.0源码分析：简单动态字符串sds
文章目录
- sds简介与特性(面试)
- sds结构模型
-
- 数据结构
- 苛刻的数据优化
-
- 数据结构优化
- uintX_t
- 对齐填充
- sds优势
-
- O(1)时间复杂度获取字符串长度
- 二进制安全
- 杜绝缓冲区溢出
-
-
- 自动扩容机制——sdsMakeRoomFor方法
-
- 内存重分配次数优化
- sds最长是多少
- 部分API源码解读
-
- 创建sds
- 释放sds
sds简介与特性(面试)
Redis 面试中大概率会提及相关的数据结构，SDS 的八股文大部分人倒背如流，可是没有读过源码，知其然不知其所以然，这可万万使不得呀！！
sds结构模型
本文阅读的Redis源码为最新的 Redis6.2.6 和 Redis3.0.0版本
相信各位看官在听到 Redis 中的字符串不是简简单单的C语言中的字符串，是 SDS（Simple Dynamic String，简单动态字符串）时以为是造出了啥新类型呢，其实 SDS 内容的源码底层就是typedef char *sds;。
数据结构
Redis6.x 的 SDS 的数据结构定义与 Redis3.0.0 相差比较大，但是核心思想不变。先从简单版本（Redis3.x）开始吧~
struct sdshdr {
//记录buf数组中已使用字节的数量
//等于SDS所保存字符串的长度
unsigned int len;
//记录buf数组中未使用字节的数量
unsigned int free;
//char数组，用于保存字符串
char buf[];
};
如下图所示为字符串"Aobing"在Redis中的存储形式：
- len 为6，表示这个 SDS 保存了一个长度为6的字符串；
- free 为0，表示这个 SDS 没有剩余空间；
- buf 是个char类型的数组，注意末尾保存了一个空字符’\0’。
buf 尾部自动追加一个’\0’字符并不会计算在 SDS 的len中，这是为了遵循 C 字符串以空字符串结尾的惯例，使得 SDS 可以直接使用一部分string.h库中的函数，如strlen
#include 苛刻的数据优化
数据结构优化
目前我们似乎得到了一个结构不错的 SDS ，但是我们能否继续进行优化呢？
在 Redis3.x 版本中不同长度的字符串占用的头部是相同的，如果某一字符串很短但是头部却占用了更多的空间，这未免太浪费了。所以我们将 SDS 分为三种级别的字符串：
- 短字符串(长度小于32)，len和free的长度用1字节即可；
- 长字符串，用2字节或者4字节；
- 超长字符串，用8字节。
共有五种类型的SDS（长度小于1字节、1字节、2字节、4字节、8字节）
我们可以在 SDS 中新增一个 type 字段来标识类型，但是没必要使用一个 4 字节的int类型去做！可以使用 1 字节的char类型，通过位运算（3位即可标识2^3种类型）来获取类型。
如下所示为短字符串(长度小于32)的优化形式：
低三位存储类型，高5位存储长度，最多能标识的长度为32，所以短字符串的长度必定小于32。
无需free字段了，32-len即为free
接下来看看Redis6.x中是怎么做的吧！
// 注意：sdshdr5从未被使用，Redis中只是访问flags。
struct __attribute__ ((__packed__)) sdshdr5 {
unsigned char flags; /* 低3位存储类型, 高5位存储长度 */
char buf[];
};
struct __attribute__ ((__packed__)) sdshdr8 {
uint8_t len; /* 已使用 */
uint8_t alloc; /* 总长度，用1字节存储 */
unsigned char flags; /* 低3位存储类型, 高5位预留 */
char buf[];
};
struct __attribute__ ((__packed__)) sdshdr16 {
uint16_t len; /* 已使用 */
uint16_t alloc; /* 总长度，用2字节存储 */
unsigned char flags; /* 低3位存储类型, 高5位预留 */
char buf[];
};
struct __attribute__ ((__packed__)) sdshdr32 {
uint32_t len; /* 已使用 */
uint32_t alloc; /* 总长度，用4字节存储 */
unsigned char flags; /* 低3位存储类型, 高5位预留 */
char buf[];
};
struct __attribute__ ((__packed__)) sdshdr64 {
uint64_t len; /* 已使用 */
uint64_t alloc; /* 总长度，用8字节存储 */
unsigned char flags; /* 低3位存储类型, 高5位预留 */
char buf[];
};
数据结构和我们分析的差不多嘛！也是加一个标识字段而已，并且不是int类型，而是1字节的char类型，使用其中的3位表示具体的类型。
同时，Redis 中也声明了5个常量分别表示五种类型的 SDS ，与我们分析的也不谋而合。
#define SDS_TYPE_5 0
#define SDS_TYPE_8 1
#define SDS_TYPE_16 2
#define SDS_TYPE_32 3
#define SDS_TYPE_64 4
uintX_t
对比前后两版代码，不难发现在 Redis6.x 中 int 类型也多出了几种：uint8_t / uint16_t / uint32_t /uint64_t。乍一看以为是新增类型呢，毕竟 C语言里面可没有这些类型呀！查看相关源码，如下：
typedef unsigned char uint8_t;
typedef unsigned short uint16_t;
typedef unsigned int uint32_t;
typedef unsigned long long uint64_t;
对齐填充
在 Redis6.x 的源码中 SDS 的结构体为struct __attribute__ ((__packed__))与struct有较大的差别，这其实和我们熟知的对齐填充有关。
（1）举个例子
考虑如下结构体：
typedef struct{
char c1;
short s;
char c2;
int i;
} s;
若此结构体中的成员都是紧凑排列的，假设c1的起始地址为0，则s的地址为1，c2的地址为3，i的地址为4。下面用代码论证一下我们的假设。
#include 尴尬了，和假设差的不是一星半点呀！这就是对齐填充搞的鬼，这啥啥啥呀~
（2）什么是字节对齐
现代计算机中，内存空间按照字节划分，理论上可以从任何起始地址访问任意类型的变量。但实际中在访问特定类型变量时经常在特定的内存地址访问，这就需要各种类型数据按照一定的规则在空间上排列，而不是顺序一个接一个地存放，这就是对齐。
（3）对齐原因
为什么需要对齐填充是由于各个硬件平台对存储空间的处理上有很大的不同。一些平台对某些特定类型的数据只能从某些特定地址开始存取。
最常见的是如果不按照适合其平台的要求对数据存放进行对齐，会在存取效率上带来损失。
比如有些平台每次读都是从偶地址开始，如果一个int型（假设为 32位）存放在偶地址开始的地方，那么一个读周期就可以读出，而如果存放在奇地址开始的地方，就可能会需要2个读周期，并对两次读出的结果的高低字节进行拼凑才能得到该int数据，导致在读取效率上下降很多。
（4）更改对齐方式
注意：我们写程序的时候，不需要考虑对齐问题。编译器会替我们选择适合目标平台的对齐策略。
如果我们一定要手动更改对齐方式，一般可以通过下面的方法来改变缺省的对界条件：
-
使用伪指令#pragma pack(n)：C编译器将按照n个字节对齐；
-
使用伪指令#pragma pack()： 取消自定义字节对齐方式。
另外，还有如下的一种方式(GCC特有语法)：
-
__attribute((aligned (n)))： 让所作用的结构成员对齐在n字节自然边界上。如果结构体中有成员的长度大于n，则按照最大成员的长度来对齐。
-
attribute ((packed))： 取消结构在编译过程中的优化对齐，按照实际占用字节数进行对齐。
将上述示例代码的结构体更改如下(取消对齐)，再次执行，可以发现取消对齐后和我们的假设就一致了。
typedef struct __attribute__ ((__packed__))
{
char c1;
short s;
char c2;
int i;
} s;
// 再次执行：c1 -> 0, s -> 1, c2 -> 3, i -> 4
（5）redis为什么不对齐呢？
综上所述我们知道了对齐填充可以提高 CPU 的数据读取效率，作为 IO 频繁的 Redis 为什么选择不对齐呢？
我们再次回顾 Redis6.x 中的 SDS 结构：
有个细节各位需要知道，即 SDS 的指针并不是指向 SDS 的起始位置（len位置），而是直接指向buf[]，使得 SDS 可以直接使用 C 语言string.h库中的某些函数，做到了兼容，十分nice~。
如果不进行对齐填充，那么在获取当前 SDS 的类型时则只需要后退一步即可flagsPointer = ((unsigned char*)s)-1；
相反，若进行对齐填充，由于 Padding 的存在，我们在不同的系统中不知道退多少才能获得flags，并且我们也不能将 sds 的指针指向flags，这样就无法兼容 C 语言的函数了，也不知道前进多少才能得到 buf[]。
sds优势
O(1)时间复杂度获取字符串长度
由于C字符串不记录自身的长度，所以为了获取一个字符串的长度程序必须遍历这个字符串，直至遇到’0’为止，整个操作的时间复杂度为O(N)。而我们使用SDS封装字符串则直接获取len属性值即可，时间复杂度为O(1)。
二进制安全
什么是二进制安全？
通俗地讲，C语言中，用’0’表示字符串的结束，如果字符串本身就有’0’字符，字符串就会被截断，即非二进制安全；若通过某种机制，保证读写字符串时不损害其内容，则是二进制安全。
C字符串中的字符除了末尾字符为’\0’外其他字符不能为空字符，否则会被认为是字符串结尾(即使实际上不是)。这限制了C字符串只能保存文本数据，而不能保存二进制数据。而SDS使用len属性的值判断字符串是否结束，所以不会受’\0’的影响。
杜绝缓冲区溢出
字符串的拼接操作是使用十分频繁的，在C语言开发中使用char *strcat(char *dest,const char *src)方法将src字符串中的内容拼接到dest字符串的末尾。
由于C字符串不记录自身的长度，所有strcat方法已经认为用户在执行此函数时已经为dest分配了足够多的内存，足以容纳src字符串中的所有内容，而一旦这个条件不成立就会产生缓冲区溢出，会把其他数据覆盖掉，Dangerous~。
// strcat 源码
char * __cdecl strcat (char * dst, const char * src)
{
char * cp = dst;
while( *cp )
cp++; /* 找到 dst 的结尾 */
while( *cp++ = *src++ ) ; /* 无脑将 src 复制到 dst 中 */
return( dst ); /* 返回 dst */
}
如下图所示为一次缓冲区溢出：
与C字符串不同，SDS 的自动扩容机制完全杜绝了发生缓冲区溢出的可能性：当SDS API需要对SDS进行修改时，API会先检查 SDS 的空间是否满足修改所需的要求，如果不满足，API会自动将SDS的空间扩展至执行修改所需的大小，然后才执行实际的修改操作，所以使用 SDS 既不需要手动修改SDS的空间大小，也不会出现缓冲区溢出问题。
SDS 的sds sdscat(sds s, const char *t)方法在字符串拼接时会进行扩容相关操作。
sds sdscatsds(sds s, const sds t) {
return sdscatlen(s, t, sdslen(t));
}
/* s: 源字符串
* t: 待拼接字符串
* len: 待拼接字符串长度
*/
sds sdscatlen(sds s, const void *t, size_t len) {
// 获取源字符串长度
size_t curlen = sdslen(s);
// SDS 分配空间（自动扩容机制）
s = sdsMakeRoomFor(s,len);
if (s == NULL) return NULL;
// 将目标字符串拷贝至源字符串末尾
memcpy(s+curlen, t, len);
// 更新 SDS 长度
sdssetlen(s, curlen+len);
// 追加结束符
s[curlen+len] = '\0';
return s;
}
自动扩容机制——sdsMakeRoomFor方法
strcatlen中调用sdsMakeRoomFor完成字符串的容量检查及扩容操作，重点分析此方法：
/* s: 源字符串
* addlen: 新增长度
*/
sds sdsMakeRoomFor(sds s, size_t addlen) {
void *sh, *newsh;
// sdsavail: s->alloc - s->len, 获取 SDS 的剩余长度
size_t avail = sdsavail(s);
size_t len, newlen, reqlen;
// 根据 flags 获取 SDS 的类型 oldtype
char type, oldtype = s[-1] & SDS_TYPE_MASK;
int hdrlen;
size_t usable;
/* Return ASAP if there is enough space left. */
// 剩余空间大于等于新增空间，无需扩容，直接返回源字符串
if (avail >= addlen) return s;
// 获取当前长度
len = sdslen(s);
//
sh = (char*)s-sdsHdrSize(oldtype);
// 新长度
reqlen = newlen = (len+addlen);
// 断言新长度比原长度长，否则终止执行
assert(newlen > len); /* 防止数据溢出 */
// SDS_MAX_PREALLOC = 1024*1024, 即1MB
if (newlen < SDS_MAX_PREALLOC)
// 新增后长度小于 1MB ，则按新长度的两倍扩容
newlen *= 2;
else
// 新增后长度大于 1MB ，则按新长度加上 1MB 扩容
newlen += SDS_MAX_PREALLOC;
// 重新计算 SDS 的类型
type = sdsReqType(newlen);
/* Don't use type 5: the user is appending to the string and type 5 is
* not able to remember empty space, so sdsMakeRoomFor() must be called
* at every appending operation. */
// 不使用 sdshdr5
if (type == SDS_TYPE_5) type = SDS_TYPE_8;
// 获取新的 header 大小
hdrlen = sdsHdrSize(type);
assert(hdrlen + newlen + 1 > reqlen); /* Catch size_t overflow */
if (oldtype==type) {
// 类型没变
// 调用 s_realloc_usable 重新分配可用内存，返回新 SDS 的头部指针
// usable 会被设置为当前分配的大小
newsh = s_realloc_usable(sh, hdrlen+newlen+1, &usable);
if (newsh == NULL) return NULL; // 分配失败直接返回NULL
// 获取指向 buf 的指针
s = (char*)newsh+hdrlen;
} else {
// 类型变化导致 header 的大小也变化，需要向前移动字符串，不能使用 realloc
newsh = s_malloc_usable(hdrlen+newlen+1, &usable);
if (newsh == NULL) return NULL;
// 将原字符串copy至新空间中
memcpy((char*)newsh+hdrlen, s, len+1);
// 释放原字符串内存
s_free(sh);
s = (char*)newsh+hdrlen;
// 更新 SDS 类型
s[-1] = type;
// 设置长度
sdssetlen(s, len);
}
// 获取 buf 总长度(待定)
usable = usable-hdrlen-1;
if (usable > sdsTypeMaxSize(type))
// 若可用空间大于当前类型支持的最大长度则截断
usable = sdsTypeMaxSize(type);
// 设置 buf 总长度
sdssetalloc(s, usable);
return s;
}
自动扩容机制总结：
扩容阶段：
- 若 SDS 中剩余空闲空间 avail 大于新增内容的长度 addlen，则无需扩容；
- 若 SDS 中剩余空闲空间 avail 小于或等于新增内容的长度 addlen：
- 若新增后总长度 len+addlen < 1MB，则按新长度的两倍扩容；
- 若新增后总长度 len+addlen > 1MB，则按新长度加上 1MB 扩容。
内存分配阶段：
- 根据扩容后的长度选择对应的 SDS 类型：
- 若类型不变，则只需通过
s_realloc_usable(即realloc函数)扩大 buf 数组即可； - 若类型变化，则需要为整个 SDS 重新分配内存，并将原来的 SDS 内容拷贝至新位置。
- 若类型不变，则只需通过
自动扩容流程图如下所示：
扩容后的 SDS 不会恰好容纳下新增的字符，而是多分配了一些空间(预分配策略)，这减少了修改字符串时带来的内存重分配次数
内存重分配次数优化
（1）空间预分配策略
因为 SDS 的空间预分配策略， SDS 字符串在增长过程中不会频繁的进行空间分配。通过这种分配策略，SDS 将连续增长N次字符串所需的内存重分配次数从必定N次降低为最多N次。
（2）惰性空间释放机制
空间预分配策略用于优化 SDS 增长时频繁进行空间分配，而惰性空间释放机制则用于优化 SDS 字符串缩短时并不立即使用内存重分配来回收缩短后多出来的空间，而仅仅更新 SDS 的len属性，多出来的空间供将来使用。
SDS 中调用sdstrim方法来缩短字符串：
/* sdstrim 方法删除字符串首尾中在 cset 中出现过的字符
* 比如:
* s = sdsnew("AA...AA.a.aa.aHelloWorld :::");
* s = sdstrim(s,"Aa. :");
* printf("%s\n", s);
*
* SDS 变成了 "HelloWorld"
*/
sds sdstrim(sds s, const char *cset) {
char *start, *end, *sp, *ep;
size_t len;
sp = start = s;
ep = end = s+sdslen(s)-1;
// strchr()函数用于查找给定字符串中某一个特定字符
while(sp <= end && strchr(cset, *sp)) sp++;
while(ep > sp && strchr(cset, *ep)) ep--;
len = (sp > ep) ? 0 : ((ep-sp)+1);
if (s != sp) memmove(s, sp, len);
s[len] = '\0';
// 仅仅更新了len
sdssetlen(s,len);
return s;
}
勘误
在《Redis的设计与实现》一书中针对 sdstrim方法的讲解为：删除字符串中 cset 出现的所有字符，而不是首尾。
比如：调用sdstrim(“XYXaYYbcXyY”,“XY”)，后移除了所有的’X’和’Y’。这是错误❌的~
SDS 并没有释放多出来的5字节空间，仅仅将 len 设置成了7，剩余空间为5。如果后续字符串增长时则可以派上用场（可能不需要再分配内存）。
也许各位看官又会有疑问了，这没真正释放空间，是否会导致内存泄漏呢？放心，SDS为我们提供了真正释放SDS未使用空间的方法sdsRemoveFreeSpace。
sds sdsRemoveFreeSpace(sds s) {
void *sh, *newsh;
// 获取类型
char type, oldtype = s[-1] & SDS_TYPE_MASK;
// 获取 header 大小
int hdrlen, oldhdrlen = sdsHdrSize(oldtype);
// 获取原字符串长度
size_t len = sdslen(s);
// 获取可用长度
size_t avail = sdsavail(s);
// 获取指向头部的指针
sh = (char*)s-oldhdrlen;
/* Return ASAP if there is no space left. */
if (avail == 0) return s;
// 查找适合这个字符串长度的最优 SDS 类型
type = sdsReqType(len);
hdrlen = sdsHdrSize(type);
/* 如果类型相同，或者至少仍然需要一个足够大的类型，我们只需 realloc buf即可；
* 否则，说明变化很大，则手动重新分配字符串以使用不同的头文件类型。
*/
if (oldtype==type || type > SDS_TYPE_8) {
newsh = s_realloc(sh, oldhdrlen+len+1);
if (newsh == NULL) return NULL;
s = (char*)newsh+oldhdrlen;
} else {
newsh = s_malloc(hdrlen+len+1);
if (newsh == NULL) return NULL;
memcpy((char*)newsh+hdrlen, s, len+1);
// 释放内存
s_free(sh);
s = (char*)newsh+hdrlen;
s[-1] = type;
sdssetlen(s, len);
}
// 重新设置总长度为len
sdssetalloc(s, len);
return s;
}
sds最长是多少
Redis 官方给出了最大的字符串容量为 512MB。这是为什么呢？
在 Reids3.x 版本中len是使用int修饰的，这就会导致 buf 最长就是2147483647，无形中限制了字符串的最大长度。
任何细节在源码中都能发现，在_sdsnewlen方法创建 SDS 中都会调用sdsTypeMaxSize方法获取每种类型所能创建的最大buf长度，不难发现此方法最大的返回值为2147483647，即512MB。
static inline size_t sdsTypeMaxSize(char type) {
if (type == SDS_TYPE_5)
return (1<<5) - 1;
if (type == SDS_TYPE_8)
return (1<<8) - 1;
if (type == SDS_TYPE_16)
return (1<<16) - 1;
#if (LONG_MAX == LLONG_MAX)
if (type == SDS_TYPE_32)
return (1ll<<32) - 1; // 不管方法啥意思，最大返回2147483647。OVER~
#endif
return -1; /* this is equivalent to the max SDS_TYPE_64 or SDS_TYPE_32 */
}
此方法在 Redis3.0.0中是不存在的
部分API源码解读
创建sds
Redis 通过sdsnewlen方法创建 SDS。在方法中会根据字符串初始化长度选择合适的类型。
sds _sdsnewlen(const void *init, size_t initlen, int trymalloc) {
void *sh;
sds s;
// 根据初始化长度判断 SDS 的类型
char type = sdsReqType(initlen);
// SDS_TYPE_5 强制转换为 SDS_TYPE_8
// 这样侧面验证了 sdshdr5 从未被使用，创建这一步就GG了 ੯ੁૂ‧̀͡u\
if (type == SDS_TYPE_5 && initlen == 0) type = SDS_TYPE_8;
// 获取头部大学
int hdrlen = sdsHdrSize(type);
// 指向 flags 的指针
unsigned char *fp; /* flags pointer. */
// 分配的空间
size_t usable;
// 防止溢出
assert(initlen + hdrlen + 1 > initlen); /* Catch size_t overflow */
// 分配空间
// s_trymalloc_usable: 尝试分配内存，失败则返回NULL
// s_malloc_usable: 分配内存或者抛异常[不友好]
sh = trymalloc?
s_trymalloc_usable(hdrlen+initlen+1, &usable) :
s_malloc_usable(hdrlen+initlen+1, &usable);
if (sh == NULL) return NULL;
if (init==SDS_NOINIT)
init = NULL;
else if (!init)
memset(sh, 0, hdrlen+initlen+1);
// s 此时指向buf
s = (char*)sh+hdrlen;
// 指向flags
fp = ((unsigned char*)s)-1;
usable = usable-hdrlen-1;
// 对不同类型的 SDS 可分配空间进行截断
if (usable > sdsTypeMaxSize(type))
usable = sdsTypeMaxSize(type);
switch(type) {
case SDS_TYPE_5: {
*fp = type | (initlen << SDS_TYPE_BITS);
break;
}
case SDS_TYPE_8: {
SDS_HDR_VAR(8,s);
sh->len = initlen;
sh->alloc = usable;
*fp = type;
break;
}
// ... 省略
}
if (initlen && init)
memcpy(s, init, initlen);
// 末尾添加\0
s[initlen] = '\0';
return s;
}
通过sdsnewlen方法我们可以获得以下信息：
SDS_TYPE_5会被强制转换为SDS_TYPE_8类型；- 创建时默认会在末尾加
'\0'； - 返回值是指向 SDS 结构中 buf 的指针；
- 返回值是
char *sds类型，可以兼容部分C函数。
释放sds
为了优化性能，SDS 提供了不直接释放内存，而是通过重置len达到清空 SDS 目的的方法——sdsclear。改方法仅仅将 SDS 的len归零，而buf的空间并为真正被清空，新的数据可以复写，而不用重新申请内存。
void sdsclear(sds s) {
sdssetlen(s, 0);// 设置len为0
s[0] = '\0';//“清空”buf
}
若真正想清空 SDS 则可以调用sdsfree方法，底层通过调用s_free释放内存。
void sdsfree(sds s) {
if (s == NULL) return;
s_free((char*)s-sdsHdrSize(s[-1]));
}