在项目中大量使用zincrby命令,原因就是要统计日志中某个指标的计数值,且需要按顺序返回topn。
正常来说,一个指标调用一次zincrby(zincrby default:type 1 typeA) 就可以正常工作。
实际情况是由于日志生成的太快,redis cpu利用率经常100%,而且还丢数据。
是否可以一次性增加多次指标的累计值,比如zincrby default:type 1 typeA 1 typeB 1 typeC 。。。,这样将多次通信压缩到一次通信中,肯定能提高处理能力。
无奈 zincrby只支持4个参数,redis官方文档上写的清楚:
ZINCRBY key increment member
所以,想到了修改redis(2.4.17)源码来实现我自己的功能。
打开redis.c源码文件可以到 redis的支持的命令表, 部分内容如下,
struct redisCommand readonlyCommandTable[] = {
{"get",getCommand,2,0,NULL,1,1,1},
{"set",setCommand,3,REDIS_CMD_DENYOOM,NULL,0,0,0},
{"setnx",setnxCommand,3,REDIS_CMD_DENYOOM,NULL,0,0,0},
{"setex",setexCommand,4,REDIS_CMD_DENYOOM,NULL,0,0,0},
{"append",appendCommand,3,REDIS_CMD_DENYOOM,NULL,1,1,1},
{"strlen",strlenCommand,2,0,NULL,1,1,1},
。。。
{"zincrby",zincrbyCommand,4,REDIS_CMD_DENYOOM,NULL,1,1,1}
}
这个版本和2.6以上版本命令参数有不同,但是前几个参数都是一致的,第一个表示命令关键字,第二个表示函数名,第三个表示输入参数个数,正数表示参数数只能等于,负数表示参数个数至少等于。
由于zincrby命令指定的参数是4,看了一下后续的实现代码,发现里面当参数大于4的时候也进行了处理,于是将4改为-4,然后make,进行测试。
{"zincrby",zincrbyCommand,-4,REDIS_CMD_DENYOOM,NULL,1,1,1}
运行如下:
其实源码修改很简单,就修改了一个参数,其他的都没动,和原来想的差很多,不过目的达到了。
redis源码修改之zincrby,hincrby命令,布布扣,bubuko.com