扩展Redis:增加Redis命令（redis增加key的值的命令）

一、背景

Redis因高性能、轻量的优秀特性成了互联网公司缓存的标配，有的时候我们想增加一些自定义的命令，主要是重度使用Redis的场景，像抢购的场景，要保证多个Redis命令的事务性，如果没有很好的原子性保证，很容易出现数据不一致的问题，虽然官方给出事务的方案，但如果中间需要根据某个命令的返回值做判断才做下一步处理，像批量扣库存，中间扣减某个库存失败就要返回，则事务的方案就无法应对了。

今天我们以一个实际的案例讲述如何增加一个Redis命令，这个命令主要用于防刷的场景：

经常要将某个IP或某个用户封禁一段时间，如果不用这个命令的方案如下：

先incr下，然后判断是否为1，是1则设置过期时间。可以看到这样处理在高并发时如果第1个发起incr请求的客户端挂掉，则这个key不会过期。

今天要编写的这个命令用来保证这个事务性，在服务端保证如果key的值为1的时候设置过期时间。

命令用法如下：

incexpire key expireTime maxNum

key：要处理的key

expireTime：过期时间，单位为秒，如果写10，表示这个key到10秒之后过期；

maxNum：表示增加到多少为止不增加了，如果设为10，则返回的最大值就是11，返回11表示已经超出了；

二、编写命令

1、下载redis代码

本文所用Redis版本为3.2.11；

下载代码并切到3.2.11分支

git clone https://github.com/antirez/redis
git checkout 3.2.11

2、编写增加命令代码

1）在src/server.c中redisCommandTable结构体中增加新命令

struct redisCommand redisCommandTable[] = {
  ……
 {"incexpire",incrExpireCommand,4,"wm",0,NULL,1,1,1,0,0}
 };

redisCommandTable为一个数组，每个项表示一个redis命令，其中第一个表示命令的名字，第二个为实际处理的函数，第3个为参数的个数，其它的先不详细讨论。

2）在src/server.h增加函数声明：

void incrExpireCommand(client *c);

3）然后新增一个文件为ljh.c（名字自己定)，加入以下代码：

#include "server.h"

void incrExpireCommand(client *c) {
    incrDecrExpireCommand(c,1);
}

/**
incr expire command
incr key expireTime maxNum
*/
void incrDecrExpireCommand(client *c, long long incr) {
    long long value, oldValue, when;
    robj *o, *new, *rExpireTime, *rMaxNum, *key;
  char* str;
  int expireNum, maxNum;
  u_char isExpired=0;

  rExpireTime = c->argv[2];
  rMaxNum = c->argv[3];

//检查过期时间参数
 str = (char*)rExpireTime->ptr;
  expireNum = atoi(str);
  if (expireNum < 0){
    addReplyError(c,"increxpire command expireNum param must bigger or equal than 0");
        return;
  }
  //check maxNum
  str = (char*)rMaxNum->ptr;
  maxNum = atoi(str);
  if (maxNum < 0){
    addReplyError(c,"increxpire command maxNum param must bigger or equal than 0");
        return;
  }

  //查找key是否存在
    o = lookupKeyWrite(c->db,c->argv[1]);
    if (o != NULL && checkType(c,o,OBJ_STRING)) return;
    if (getLongLongFromObjectOrReply(c,o,&value,NULL) != C_OK) return;

    oldValue = value;
    if ((incr < 0 && oldValue < 0 && incr < (LLONG_MIN-oldValue)) ||
        (incr > 0 && oldValue > 0 && incr > (LLONG_MAX-oldValue))) {
        addReplyError(c,"increment or decrement will overflow");
        return;
    }
    
  if (0 == oldValue){
    //expired 
    isExpired = 1;
  }else if (value >= (maxNum+1)){
  //如果大于等于最大值，则直接返回
    addReply(c,shared.colon);
    addReply(c, o);
    addReply(c,shared.crlf);
    return;
  }

  //添加步长值
    value += incr;
    new = createStringObjectFromLongLong(value);
    if (o){
        dbOverwrite(c->db,c->argv[1],new);
    }else{
    dbAdd(c->db,c->argv[1],new);
    //第一次不存在，添加key之后还要设置过期时间
    key = c->argv[1];
    when = mstime();
    when += expireNum * 1000;
    setExpire(c->db, key, when);
        //signalModifiedKey(c->db,key);
        notifyKeyspaceEvent(NOTIFY_GENERIC,"expire", key, c->db->id);
    }
    
    //redis事件
    signalModifiedKey(c->db,c->argv[1]);
    notifyKeyspaceEvent(NOTIFY_STRING,"isxpired",c->argv[1],c->db->id);
    server.dirty++;
    
    //返回最新的值
    addReply(c,shared.colon);
    addReply(c,new);
    addReply(c,shared.crlf);
}

4）在src/Makefile中增加新文件 ljh.o

REDIS_SERVER_OBJ=adlist.o quicklist.o ae.o anet.o dict.o server.o sds.o zmalloc.o lzf_c.o lzf_d.o pqsort.o zipmap.o sha1.o ziplist.o release.o networking
    .o util.o object.o db.o replication.o rdb.o t_string.o t_list.o t_set.o t_zset.o t_hash.o config.o aof.o pubsub.o multi.o debug.o sort.o intset.o syncio.
    o cluster.o crc16.o endianconv.o slowlog.o scripting.o bio.o rio.o rand.o memtest.o crc64.o bitops.o sentinel.o notify.o setproctitle.o blocked.o hyperlo
    glog.o latency.o sparkline.o redis-check-rdb.o geo.o slots.o slots_async.o crc32.o oneplus.o cJSON.o ljh.o

然后编译：make

执行效果如下

可以看到，key1增加到11之后不再增加，并且过了10秒之后，又变成1了。

三、写在最后

最后我们还是要问自己我们的场景中是否有增加Redis命令的必要，如果需要则要把握好各层之间的职责，不要让Redis最后变成处理业务去了；一般来说中间件层一般处理比较通用的功能，越到下面的层应该越稳定，越少改动才是正常的情况。