扩展Redis:增加Redis命令

一、背景

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删除机制

一次线上Mysql死锁分析

直播评论系统分析设计

RabbitMQ网络框架代码分析二:命令分发

RabbitMQ网络框架代码分析

你可能感兴趣的