Redis因高性能、輕量的優(yōu)秀特性成了互聯(lián)網(wǎng)公司緩存的標配，有的時候我們想增加一些自定義的命令，主要是重度使用Redis的場景，像搶購的場景，要保證多個Redis命令的事務性，如果沒有很好的原子性保證，很容易出現(xiàn)數(shù)據(jù)不一致的問題。

雖然官方給出事務的方案，但如果中間需要根據(jù)某個命令的返回值做判斷才做下一步處理，則事務的方案就無法應對了。

今天我們以一個實際的案例講述如何增加一個Redis命令，這個命令主要用于防刷的場景：

經(jīng)常要將某個IP或某個用戶封禁一段時間，如果不用這個命令的方案如下：

先incr下，然后判斷是否為1，是1則設置過期時間?？梢钥吹竭@樣處理在高并發(fā)時如果第1個發(fā)起incr請求的客戶端掛掉，則這個key不會過期。

今天要編寫的這個命令用來保證這個事務性，在服務端保證如果key的值為1的時候設置過期時間。

命令用法如下：

incexpire key expireTime maxNum

key：要處理的key

expireTime：過期時間，單位為秒，如果寫10，表示這個key到10秒之后過期；

maxNum：表示增加到多少為止不增加了，如果設為10，則返回的最大值就是11，返回11表示已經(jīng)超出了；

二、編寫命令

1、下載redis代碼

本文所用Redis版本為3.2.11；

下載代碼并切到3.2.11分支

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

2、編寫增加命令代碼

1）在src/server.c中redisCommandTable結構體中增加新命令

struct redisCommand redisCommandTable[] = {

redisCommandTable為一個數(shù)組，每個項表示一個redis命令，其中第一個表示命令的名字，第二個為實際處理的函數(shù)，第3個為參數(shù)的個數(shù)，其它的先不詳細討論。

2）在src/server.h增加函數(shù)聲明：

void incrExpireCommand(client *c);

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

#include "server.h"

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

執(zhí)行效果如下

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

最后我們還是要問自己我們的場景中是否有增加Redis命令的必要，如果需要則要把握好各層之間的職責，不要讓Redis最后變成處理業(yè)務去了；一般來說中間件層一般處理比較通用的功能，越到下面的層應該越穩(wěn)定，越少改動才是正常的情況。