首页 > 代码库 > redis源码修改之zincrby，hincrby命令

redis源码修改之zincrby，hincrby命令

2024-07-12 07:55:05   221人阅读

在项目中大量使用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}
运行如下：


其实源码修改很简单，就修改了一个参数，其他的都没动，和原来想的差很多，不过目的达到了。


联系
我们