Redis设计与实现读书笔记——简单动态字符串

前言

项目里用到了redis数据结构,不想只是简单的调用api,这里对我的读书笔记做一下记录。原文地址: http://www.redisbook.com/en/latest/internal-datastruct/sds.html

数据类型定义

与sds实现有关的数据类型有两个,一个是 sds:

// 字符串类型的别名
typedef char *sds;

另一个是 sdshdr:

// 持有sds的结构
struct sdshdr {
	// buf中已经被使用的字符串空间数量
	int len;
	// buf中预留字符串的空间数量
	int free;
	// 实际存储字符串的地方
	char buf[];
};

其中,sds只是字符串数组类型char*的别名,而sdshdr用于持有和保存sds的信息

比如,sdshdr.len可以用于在O(1)的复杂度下获取sdshdr.buf中存储的字符串的实际长度,而sdshdr.free则用于保存sdshdr.buf中还有多少预留空间

(这里sdshdr应该是sds handler的缩写)

将sdshdr用作sds

sds模块对sdshdr结构使用了一点小技巧:通过指针运算,它使得sdshdr结构可以像sds类型一样被传值和处理,并在需要的时候恢复成sdshdr类型

通过下面的函数定义来理解这个技巧

sdsnewlen 函数返回一个新的sds值,实际上,它创建的却是一个sdshdr结构:

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

	if (init) {
		// 创建
		sh = malloc(sizeof(struct sdshdr) + initlen + 1);
	} else {
		// 重分配
		sh = calloc(1, sizeof(struct sdshdr) + initlen + 1);
	}

	if (sh == NULL) return NULL;

	sh->len = initlen;
	sh->free = 0;	// 刚开始free为0

	if (initlen && init) {
		memcpy(sh->buf, init, initlen);
	}
	sh->buf[initlen] = '';

	// 只返回sh->buf这个字符串部分
	return (char *)sh->buf;
}

通过使用变量持有一个sds的值,在遇到那些只处理sds值本身的函数时,可以直接将sds传给它们。比如说,sdstoupper 函数就是其中的一个例子:

static inline size_t sdslen(const sds s)
{
	// 从sds中计算出相应的sdshdr结构
	struct sdshdr *sh = (void *)(s - (sizeof(struct sdshdr)));

	return sh->len;
}


void sdstoupper(sds s)
{
	int len = sdslen(s), j;

	for (j = 0; j < len; j ++)
		s[j] = toupper(s[j]);
}

这里有一个技巧,通过指针运算,可以从sds值中计算出相应的sdshdr结构:

sds虽然是指向char *的buf(ps:并且空数组不占用内存空间,数组名即为内存地址),但是分配的时候是分配sizeof(struct sdshdr) + initlen + 1的,通过sds - sizeof(struct sdshdr)可以计算出struct sdshdr的首地址,从而可以得到len和free的信息




sdsavail 函数就是使用这中技巧的一个例子:

static inline size_t sdsavail(const sds s)
{
	struct sdshdr *sh = (void *)(s - (sizeof(struct sdshdr)));

	return sh->free;
}

内存分配函数实现

和Reids 的实现决策相关的函数是 sdsMakeRoomFor :

sds sdsMakeRoomFor(sds s, size_t addlen)
{
	struct sdshdr *sh, *newsh;
	size_t free = sdsavail(s);
	size_t len, newlen;

	// 预留空间可以满足本地拼接	
	if (free >= addlen) return s;

	len = sdslen(s);
	sh = (void *)(s - (sizeof(struct sdshdr)));

	// 设置新sds的字符串长度
	// 这个长度比完成本次拼接实际所需的长度要大
	// 通过预留空间优化下次拼接操作
	newlen = (len + addlen);
	if (newlen < 1024 * 1024)
		newlen *= 2;
	else
		newlen += 1024;

	// 重新分配sdshdr
	newsh = realloc(sh, sizeof(struct sdshdr) + newlen + 1);
	if (newsh == NULL) return NULL;

	newsh->free = newlen - len;

	// 只返回字符串部分
	return newsh->buf;
}

这种内存分配策略表明,在对sds 值进行扩展(expand)时,总会预留额外的空间,通过花费更多的内存,减少了对内存进行重分配(reallocate)的次数,并优化下次扩展操作的处理速度

再把redis的如果实现对sds字符串扩展的方法贴一下,很不错的思路:

/**
 * 按长度len扩展sds,并将t拼接到sds的末尾
 */
sds sdscatlen(sds s, const void *t, size_t len)
{
	struct sdshdr *sh;

	size_t curlen = sdslen(s);

	// O(N)
	s = sdsMakeRoomFor(s, len);
	if (s == NULL) return NULL;

	// 复制
	memcpy(s + curlen, t, len);

	// 更新len和free属性
	sh = (void *)(s - (sizeof(struct sdshdr)));
	sh->len = curlen + len;
	sh->free = sh->free - len;

	// 终结符
	s[curlen + len] = '';

	return s;
}

/**
 * 将一个char数组拼接到sds 末尾
 */
sds sdscat(sds s, const char *t)
{
	return sdscatlen(s, t, strlen(t));
}


OK,这里暂时对sds(简单动态字符串)的学习告一段落,继续写业务逻辑代码,很好奇hashs和sets结构是如何实现!!
原文地址:https://www.cnblogs.com/xinyuyuanm/p/3177887.html