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
2、编写增加命令代码
1)在src/server.c中redisCommandTable结构体中增加新命令
struct redisCommand redisCommandTable[] = {
redisCommandTable为一个数组,每个项表示一个redis命令,其中第一个表示命令的名字,第二个为实际处理的函数,第3个为参数的个数,其它的先不详细讨论。
2)在src/server.h增加函数声明:
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
执行效果如下
可以看到,key1增加到11之后不再增加,并且过了10秒之后,又变成1了。
最后我们还是要问自己我们的场景中是否有增加Redis命令的必要,如果需要则要把握好各层之间的职责,不要让Redis最后变成处理业务去了;一般来说中间件层一般处理比较通用的功能,越到下面的层应该越稳定,越少改动才是正常的情况。
更多相关知识请关注redis入门教程栏目
以上就是增加redis命令的方法的详细内容,更多请关注自由互联其它相关文章!