redis源码阅读之字符串对象模型

之前的文章介绍了redis当中的模型，今天首先来拿字符串对象说说事~
1. 设置命令
首先，redis定义了5个宏来确定当前设置的一些属性，具体如下

点击(此处)折叠或打开

1. #define OBJ_SET_NO_FLAGS 0
   
2. #define OBJ_SET_NX (1<<0) /* Set if key not exists. */
   
3. #define OBJ_SET_XX (1<<1) /* Set if key exists. */
   
4. #define OBJ_SET_EX (1<<2) /* Set if time in seconds is given */
   
5. #define OBJ_SET_PX (1<<3) /* Set if time in ms in given */

常规的set接口如下：

点击(此处)折叠或打开

1. /* SET key value [NX] [XX] [EX <seconds>] [PX <milliseconds>] */
   
2. void setCommand(client *c) {
   
3.     int j;
   
4.     robj *expire = NULL;
   
5.     int unit = UNIT_SECONDS;
   
6.     int flags = OBJ_SET_NO_FLAGS;
   
7. 
   
8.     for (j = 3; j < c->argc; j++) {
   
9.         char *a = c->argv[j]->ptr;
   
10.         robj *next = (j == c->argc-1) ? NULL : c->argv[j+1];
    
11. 
    
12.         if ((a[0] == 'n' || a[0] == 'N') &&
    
13.             (a[1] == 'x' || a[1] == 'X') && a[2] == '\0' &&
    
14.             !(flags & OBJ_SET_XX))
    
15.         {
    
16.             flags |= OBJ_SET_NX;
    
17.         } else if ((a[0] == 'x' || a[0] == 'X') &&
    
18.                    (a[1] == 'x' || a[1] == 'X') && a[2] == '\0' &&
    
19.                    !(flags & OBJ_SET_NX))
    
20.         {
    
21.             flags |= OBJ_SET_XX;
    
22.         } else if ((a[0] == 'e' || a[0] == 'E') &&
    
23.                    (a[1] == 'x' || a[1] == 'X') && a[2] == '\0' &&
    
24.                    !(flags & OBJ_SET_PX) && next)
    
25.         {
    
26.             flags |= OBJ_SET_EX;
    
27.             unit = UNIT_SECONDS;
    
28.             expire = next;
    
29.             j++;
    
30.         } else if ((a[0] == 'p' || a[0] == 'P') &&
    
31.                    (a[1] == 'x' || a[1] == 'X') && a[2] == '\0' &&
    
32.                    !(flags & OBJ_SET_EX) && next)
    
33.         {
    
34.             flags |= OBJ_SET_PX;
    
35.             unit = UNIT_MILLISECONDS;
    
36.             expire = next;
    
37.             j++;
    
38.         } else {
    
39.             addReply(c,shared.syntaxerr);
    
40.             return;
    
41.         }
    
42.     }
    
43. 
    
44.     c->argv[2] = tryObjectEncoding(c->argv[2]);
    
45.     setGenericCommand(c,flags,c->argv[1],c->argv[2],expire,unit,NULL,NULL);
    
46. }

这边先是解析了一部分参数，如果类型不合法，就返回，如果合法就需要调用下面的函数
这个就是整个set命令当中的重头戏，其余的所有set操作，在确定一些其他的参数后，都会调用这个函数，具体代码如下：

点击(此处)折叠或打开

1. void setGenericCommand(client *c, int flags, robj *key, robj *val, robj *expire, int unit, robj *ok_reply, robj *abort_reply) {
   
2.     long long milliseconds = 0; /* initialized to avoid any harmness warning */
   
3. 
   
4.     if (expire) {
   
5.         if (getLongLongFromObjectOrReply(c, expire, &milliseconds, NULL) != C_OK)
   
6.             return;
   
7.         if (milliseconds <= 0) {
   
8.             addReplyErrorFormat(c,"invalid expire time in %s",c->cmd->name);
   
9.             return;
   
10.         }
    
11.         if (unit == UNIT_SECONDS) milliseconds *= 1000;
    
12.     }
    
13. 
    
14.     if ((flags & OBJ_SET_NX && lookupKeyWrite(c->db,key) != NULL) ||
    
15.         (flags & OBJ_SET_XX && lookupKeyWrite(c->db,key) == NULL))
    
16.     {
    
17.         addReply(c, abort_reply ? abort_reply : shared.nullbulk);
    
18.         return;
    
19.     }
    
20.     setKey(c->db,key,val);
    
21.     server.dirty++;
    
22.     if (expire) setExpire(c,c->db,key,mstime()+milliseconds);
    
23.     notifyKeyspaceEvent(NOTIFY_STRING,"set",key,c->db->id);
    
24.     if (expire) notifyKeyspaceEvent(NOTIFY_GENERIC,
    
25.         "expire",key,c->db->id);
    
26.     addReply(c, ok_reply ? ok_reply : shared.ok);
    
27. }

这里可以看到显示确定各个参数的值的合法性以及类型，进而采取不同的操作来返回。
这里需要注意的一点是：每次调用set命令时，redis都会先调用tryObjectEncoding这个函数来尽量将值的编码类型转换为占用存储空间较小的那一个，之后才会调用setGenericCommand函数进行实际操作~~
get操作就相对没那么复杂，只需要调用底层的查找就行了

点击(此处)折叠或打开

1. int getGenericCommand(client *c) {
   
2.     robj *o;
   
3. 
   
4.     if ((o = lookupKeyReadOrReply(c,c->argv[1],shared.nullbulk)) == NULL)
   
5.         return C_OK;
   
6. 
   
7.     if (o->type != OBJ_STRING) {
   
8.         addReply(c,shared.wrongtypeerr);
   
9.         return C_ERR;
   
10.     } else {
    
11.         addReplyBulk(c,o);
    
12.         return C_OK;
    
13.     }
    
14. }

之后的一些api包括setrange, getrange, mget, mset, incrXXX那些函数基本思想都同上，先要检查参数的类型以及合法性，然后调用底层db给出的接口，然后返回，这边不再赘述，关于底层db的接口，会在讲完redis对象模型之后进行阅读以及分析~