Redis 6.0源码学习 String类型

• • Redis 6.0源码学习 String类型
  • • 前置知识
    • 内部编码
    • • OBJ_ENCODING_INT
      • OBJ_ENCODING_RAW
      • OBJ_ENCODING_EMBSTR
    • 编码推断

  String字符串类型是RedisObject中最基础的一个类型，相对应的编码格式有3种，分别是OBJ_ENCODING_RAW、OBJ_ENCODING_INT和OBJ_ENCODING_EMBSTR。

Redis 6.0源码学习 RedisObject

  OBJ_ENCODING_INT算是所有编码格式中最特殊的一个。OBJ_ENCODING_INT编码中，redisObject中指针属性ptr存储的数据是字面量而不是内存地址。由于复用了ptr属性，所以OBJ_ENCODING_INT只能存储8位字节的长整型。

set key 8653
object encoding key

  在内存优化方面，OBJ_ENCODING_INT编码还做了享元优化。当服务器初始化的时候，会调用service.c中的createSharedObjects函数。

代码片段 初始化createSharedObjects.integers

for (j = 0; j < OBJ_SHARED_INTEGERS; j++) {
    shared.integers[j] =
        makeObjectShared(createObject(OBJ_STRING,(void*)(long)j));
    shared.integers[j]->encoding = OBJ_ENCODING_INT;
}

  OBJ_ENCODING_RAW是一种非常中规中矩的实现，它的ptr指针指向了一个SDS实例。

set key "这是一个等于45个字符的字符串...."
object encoding key

  OBJ_ENCODING_RAW中robj与sds内存空间不是连续的，基于局部性思想就诞生了robj与sds采用连续内存空间的OBJ_ENCODING_EMBSTR。当字符串长度不大于44个字符时，会采用OBJ_ENCODING_RAW编码。

set key "hello,world"
object encoding key

  setCommand是set命令的实现函数。以"SET a 111"为例，解析完请求之后，111对应的redisObject是编码为OBJ_ENCODING_RAW的字符串。tryObjectEncoding会进行一次编码推断，如果符合OBJ_ENCODING_INT的约束，还可能从享元池中复用对象。

代码片段 t_string.c setCommand

void setCommand(client *c) {
    int j;
    robj *expire = NULL;
    int unit = UNIT_SECONDS;
    int flags = OBJ_SET_NO_FLAGS;

    ...
    ...
    ...

    c->argv[2] = tryObjectEncoding(c->argv[2]);
    setGenericCommand(c,flags,c->argv[1],c->argv[2],expire,unit,NULL,NULL);
}

代码片段 object.c tryObjectEncoding

robj *tryObjectEncoding(robj *o) {
    long value;
    sds s = o->ptr;
    size_t len;

    
    serverAssertWithInfo(NULL,o,o->type == OBJ_STRING);

    
    if (!sdsEncodedObject(o)) return o;

    
     if (o->refcount > 1) return o;

    
    len = sdslen(s);
    if (len <= 20 && string2l(s,len,&value)) {
        
        if ((server.maxmemory == 0 ||
            !(server.maxmemory_policy & MAXMEMORY_FLAG_NO_SHARED_INTEGERS)) &&
            value >= 0 &&
            value < OBJ_SHARED_INTEGERS)
        {
            decrRefCount(o);
            incrRefCount(shared.integers[value]);
            return shared.integers[value];
        } else {
            if (o->encoding == OBJ_ENCODING_RAW) {
                sdsfree(o->ptr);
                o->encoding = OBJ_ENCODING_INT;
                o->ptr = (void*) value;
                return o;
            } else if (o->encoding == OBJ_ENCODING_EMBSTR) {
                decrRefCount(o);
                return createStringObjectFromLongLongForValue(value);
            }
        }
    }

    
    if (len <= OBJ_ENCODING_EMBSTR_SIZE_LIMIT) {
        robj *emb;

        if (o->encoding == OBJ_ENCODING_EMBSTR) return o;
        emb = createEmbeddedStringObject(s,sdslen(s));
        decrRefCount(o);
        return emb;
    }

    
    trimStringObjectIfNeeded(o);

    
    return o;
}