Chinaunix首页 | 论坛 | 博客
  • 博客访问: 1879347
  • 博文数量: 184
  • 博客积分: 0
  • 博客等级: 民兵
  • 技术积分: 2388
  • 用 户 组: 普通用户
  • 注册时间: 2016-12-21 22:26
个人简介

90后空巢老码农

文章分类

全部博文(184)

文章存档

2021年(26)

2020年(56)

2019年(54)

2018年(47)

2017年(1)

我的朋友

分类: NOSQL

2019-03-26 17:02:32

之前的文章介绍了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.     for (j = 3; j < c->argc; j++) {
  8.         char *a = c->argv[j]->ptr;
  9.         robj *next = (j == c->argc-1) ? NULL : c->argv[j+1];

  10.         if ((a[0] == 'n' || a[0] == 'N') &&
  11.             (a[1] == 'x' || a[1] == 'X') && a[2] == '\0' &&
  12.             !(flags & OBJ_SET_XX))
  13.         {
  14.             flags |= OBJ_SET_NX;
  15.         } else if ((a[0] == 'x' || a[0] == 'X') &&
  16.                    (a[1] == 'x' || a[1] == 'X') && a[2] == '\0' &&
  17.                    !(flags & OBJ_SET_NX))
  18.         {
  19.             flags |= OBJ_SET_XX;
  20.         } else if ((a[0] == 'e' || a[0] == 'E') &&
  21.                    (a[1] == 'x' || a[1] == 'X') && a[2] == '\0' &&
  22.                    !(flags & OBJ_SET_PX) && next)
  23.         {
  24.             flags |= OBJ_SET_EX;
  25.             unit = UNIT_SECONDS;
  26.             expire = next;
  27.             j++;
  28.         } else if ((a[0] == 'p' || a[0] == 'P') &&
  29.                    (a[1] == 'x' || a[1] == 'X') && a[2] == '\0' &&
  30.                    !(flags & OBJ_SET_EX) && next)
  31.         {
  32.             flags |= OBJ_SET_PX;
  33.             unit = UNIT_MILLISECONDS;
  34.             expire = next;
  35.             j++;
  36.         } else {
  37.             addReply(c,shared.syntaxerr);
  38.             return;
  39.         }
  40.     }

  41.     c->argv[2] = tryObjectEncoding(c->argv[2]);
  42.     setGenericCommand(c,flags,c->argv[1],c->argv[2],expire,unit,NULL,NULL);
  43. }
这边先是解析了一部分参数,如果类型不合法,就返回,如果合法就需要调用下面的函数
这个就是整个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.     if (expire) {
  4.         if (getLongLongFromObjectOrReply(c, expire, &milliseconds, NULL) != C_OK)
  5.             return;
  6.         if (milliseconds <= 0) {
  7.             addReplyErrorFormat(c,"invalid expire time in %s",c->cmd->name);
  8.             return;
  9.         }
  10.         if (unit == UNIT_SECONDS) milliseconds *= 1000;
  11.     }

  12.     if ((flags & OBJ_SET_NX && lookupKeyWrite(c->db,key) != NULL) ||
  13.         (flags & OBJ_SET_XX && lookupKeyWrite(c->db,key) == NULL))
  14.     {
  15.         addReply(c, abort_reply ? abort_reply : shared.nullbulk);
  16.         return;
  17.     }
  18.     setKey(c->db,key,val);
  19.     server.dirty++;
  20.     if (expire) setExpire(c,c->db,key,mstime()+milliseconds);
  21.     notifyKeyspaceEvent(NOTIFY_STRING,"set",key,c->db->id);
  22.     if (expire) notifyKeyspaceEvent(NOTIFY_GENERIC,
  23.         "expire",key,c->db->id);
  24.     addReply(c, ok_reply ? ok_reply : shared.ok);
  25. }
这里可以看到显示确定各个参数的的合法性以及类型,进而采取不同的操作来返回。
这里需要注意的一点是:每次调用set命令时,redis都会先调用tryObjectEncoding这个函数来尽量将值的编码类型转换为占用存储空间较小的那一个,之后才会调用setGenericCommand函数进行实际操作~~
get操作就相对没那么复杂,只需要调用底层的查找就行了

点击(此处)折叠或打开

  1. int getGenericCommand(client *c) {
  2.     robj *o;

  3.     if ((o = lookupKeyReadOrReply(c,c->argv[1],shared.nullbulk)) == NULL)
  4.         return C_OK;

  5.     if (o->type != OBJ_STRING) {
  6.         addReply(c,shared.wrongtypeerr);
  7.         return C_ERR;
  8.     } else {
  9.         addReplyBulk(c,o);
  10.         return C_OK;
  11.     }
  12. }
之后的一些api包括setrange, getrange, mget, mset, incrXXX那些函数基本思想都同上,先要检查参数的类型以及合法性,然后调用底层db给出的接口,然后返回,这边不再赘述,关于底层db的接口,会在讲完redis对象模型之后进行阅读以及分析~
阅读(12989) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~