【Redis源碼】append命令

簡(jiǎn)介
數據庫已經(jīng)有了key,它的值為value。當我們發(fā)現value值需要追加字符串卻又不想直接用set命令覆蓋原值時(shí),可以用append命令來(lái)實(shí)現。
命令格式:
append key value
- 說(shuō)明: 將value追加到原值的末尾,如果key不存在,此命令等同于set key value命令。
append 實(shí)現
現在介紹在key已經(jīng)存在的情況下進(jìn)行的操作。我們知道,只有value為字符串時(shí)才可以追加字符串,數字是不可以追加的,所以當key存在時(shí),首先判斷下value的類(lèi)型是否為string類(lèi)型。如果不為string類(lèi)型時(shí)會(huì )報錯。
if (checkType(c,o,OBJ_STRING))
return;
在追加字符串時(shí),需要判斷追加后的字符串長(cháng)度必須小于512MB,否則會(huì )報錯。
append = c->argv[2];
totlen = stringObjectLen(o)+sdslen(append->ptr);//檢查長(cháng)度
if (checkStringLength(c,totlen) != C_OK)
checkStringLength函數原型如下:
static int checkStringLength(client *c, long long size) {
if (size > 512*1024*1024) {
addReplyError(c,"string exceeds maximum allowed size (512MB)");
return C_ERR;
}
return C_OK;
}
這里我們不禁要問(wèn),為什么在追加字符串時(shí)才考慮追加后的長(cháng)度不能大于512 MB,那么在set命令時(shí)為什么沒(méi)有限制最大長(cháng)度呢?在networking.c中找到如下代碼:
ok = string2ll(c->querybuf+1+c->qb_pos,newline-(c->querybuf+1+c->qb_pos),&ll);
if (!ok || ll > 1024*1024) {
addReplyError(c,"Protocol error: invalid multibulk length");
setProtocolError("invalid mbulk count",c);
return C_ERR;
}
由此可見(jiàn),在服務(wù)端接收到命令的時(shí)候,就已經(jīng)判斷了命令的最大長(cháng)度不能大于1 MB,所以set命令不需要再次判斷了。
字符串追加會(huì )修改原字符串的值,所以必須保證字符串是非共享的。如果字符串是共享的,則需要解除共享,新創(chuàng )建一個(gè)值對象。實(shí)現代碼為:
robj *dbUnshareStringValue(redisDb *db, robj *key, robj *o) {
serverAssert(o->type == OBJ_STRING);
if (o->refcount != 1 || o->encoding != OBJ_ENCODING_RAW) {
// 如果是共享的,則需要解除共享,創(chuàng )建新的字符串
robj *decoded = getDecodedObject(o);
o = createRawStringObject(decoded->ptr, sdslen(decoded->ptr));
decrRefCount(decoded);
dbOverwrite(db,key,o);
}
return o;
}
值對象創(chuàng )建好之后,將新字符串追加到原字符串末尾。
o->ptr = sdscatlen(o->ptr,append->ptr,sdslen(append->ptr));
這樣就完成了字符串的append操作。
評論
0 評論