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

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

今天我們以一個(gè)實(shí)際的案例講述如何增加一個(gè)Redis命令，這個(gè)命令主要用于防刷的場(chǎng)景：

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

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

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

命令用法如下：

incexpire key expireTime maxNum

key：要處理的key

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

maxNum：表示增加到多少為止不增加了，如果設(shè)為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結(jié)構(gòu)體中增加新命令

struct redisCommand redisCommandTable[] = {

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

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

void incrExpireCommand(client *c);

3）然后新增一個(gè)文件為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了。

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