Redis模块开发示例

实现一个Redis module,支持两个扩展命令: 1) 可同时对hash的多个field进行incr操作; 2) incrby同时设置一个key的过期时间 在没有module之前,需要借助eval+lua实现相同的功能。有了module,不但可以实现逻辑复杂,且性能高的扩展,同时享受Redis的持久化和容灾能力。

  // Redis命令扩展module
  
 			#include "redismodule.h"
 
 			#include <errno.h>
 
 			#include <limits.h>
 
 			#include <stdlib.h>
 
 			#include <string.h>
 
 			#include <vector>
 
 
 
 // 带超时的INCRBY命令
 
 // 格式:INCRBYEX KEY SECONDS INCREMENT
 
 //
 
 // 示例:
 
 // ex.incrbyex k1 10 1
 
 // ex.incrbyex k1 10 2
 
 int incrbyex_RedisCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
 
 // When automatic memory management is enabled:
 
 // 1) don't need to close open keys
 
 // 2) don't need to free replies
 
 // 3) don't need to free RedisModuleString objects
 
 			    RedisModule_AutoMemory(ctx); /* Use automatic memory management. */
 
 if (argc != 4) return RedisModule_WrongArity(ctx);
 
 
 
 			    RedisModuleKey *key = (RedisModuleKey*)RedisModule_OpenKey(ctx, argv[1],
 
 			            REDISMODULE_READ|REDISMODULE_WRITE);
 
 int type = RedisModule_KeyType(key);
 
 if (type != REDISMODULE_KEYTYPE_STRING &&
 
 			        type != REDISMODULE_KEYTYPE_EMPTY)
 
 {
 
 			        return RedisModule_ReplyWithError(ctx, REDISMODULE_ERRORMSG_WRONGTYPE);
 
 }
 
 
 
 			    RedisModuleString* seconds = argv[2];
 
 			    RedisModuleString* increment = argv[3];
 
 			    long long ll_seconds; // 过期时长
 
 			    long long ll_newval; // 新的值
 
 if (RedisModule_StringToLongLong(seconds, &ll_seconds) != REDISMODULE_OK) {
 
 			        return RedisModule_ReplyWithError(ctx, "ERR value is not an integer or out of range");
 
 }
 
 if (RedisModule_StringToLongLong(increment, &ll_newval) != REDISMODULE_OK) {
 
 			        return RedisModule_ReplyWithError(ctx, "ERR value is not an integer or out of range");
 
 }
 
 
 
 			    size_t len;
 
 			    char *s = RedisModule_StringDMA(key, &len, REDISMODULE_READ|REDISMODULE_WRITE);
 
 if (0 == len || NULL == s || s == '\0') {
 
 // set必须在Expire之前,否则会冲掉Expire的作用,
 
 // 这也是else分支未用RedisModule_StringSet的原因
 
 			        RedisModule_StringSet(key, increment);
 
 			        RedisModule_SetExpire(key, ll_seconds*1000); // 以秒为单位,需要乘以1000
 
 }
 
 else {
 
 			        char* endptr;
 
 			        long long ll_oldval = strtoll(s, &endptr, 10); // s不一定是有效的数字,所以需要做检查
 
 			        ll_newval = ll_newval + ll_oldval;
 
 if ((errno == ERANGE && (ll_oldval == LLONG_MAX || ll_oldval == LLONG_MIN))
 
 || (errno != 0 && ll_oldval == 0)) {
 
 			            return RedisModule_ReplyWithError(ctx, "ERR value is not an integer or out of range");
 
 }
 
 if (endptr == s || *endptr != '\0') {
 
 			            return RedisModule_ReplyWithError(ctx, "ERR value is not an integer or out of range");
 
 }
 
 
 
 			        size_t newval_len;
 
 			        RedisModuleString* newval = RedisModule_CreateStringFromLongLong(ctx, ll_newval);
 
 const char* newval_s = RedisModule_StringPtrLen(newval, &newval_len);
 
 if (newval_len > len)
 
 			            RedisModule_StringTruncate(key, newval_len);
 
 			        strncpy(s, newval_s, newval_len);
 
 }
 
 
 
 			    RedisModule_ReplyWithLongLong(ctx, ll_newval);
 
 			    return REDISMODULE_OK;
 
 }
 
 
 
 // 实现命令HMINCRBY,同时对HASH的多个field值进行增减操作
 
 // 格式:HMINCRBY KEY FIELD1 VALUE1 FIELD2 VALUE2 FIELD3 VALUE3 ......
 
 //
 
 // 示例:
 
 // hmincrby k1 f1 1 f2 2
 
 // hmincrby k1 f5 1 f6 2 f7 3
 
 int hmincrby_RedisCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
 
 // When automatic memory management is enabled:
 
 // 1) don't need to close open keys
 
 // 2) don't need to free replies
 
 // 3) don't need to free RedisModuleString objects
 
 			    RedisModule_AutoMemory(ctx); /* Use automatic memory management. */
 
 if (argc < 4 || argc % 2 != 0) return RedisModule_WrongArity(ctx);
 
 
 
 			    RedisModuleKey *key = (RedisModuleKey*)RedisModule_OpenKey(ctx, argv[1],
 
 			            REDISMODULE_READ|REDISMODULE_WRITE);
 
 int type = RedisModule_KeyType(key);
 
 if (type != REDISMODULE_KEYTYPE_HASH &&
 
 			        type != REDISMODULE_KEYTYPE_EMPTY)
 
 {
 
 			        return RedisModule_ReplyWithError(ctx, REDISMODULE_ERRORMSG_WRONGTYPE);
 
 }
 
 
 
 const int count = argc/2 - 1; // 键值对个数
 
 			    std::vector<long long> newval_array(count); // 用来存储新值的数组
 
 for (int i=2; i<argc; i+=2) {
 
 			        RedisModuleString* field = argv[i];
 
 			        RedisModuleString* incrvalue = argv[i+1];
 
 			        long long ll_newvalue;
 
 if (RedisModule_StringToLongLong(incrvalue, &ll_newvalue) != REDISMODULE_OK) {
 
 			            return RedisModule_ReplyWithError(ctx, "ERR value is not an integer or out of range");
 
 }
 
 
 
 			        RedisModuleString *oldval;
 
 			        RedisModule_HashGet(key, REDISMODULE_HASH_NONE, field, &oldval, NULL);
 
 if (NULL == oldval) { // field不存在时
 
 			            RedisModule_HashSet(key,REDISMODULE_HASH_NONE, field, incrvalue, NULL);
 
 }
 
 else { // field已存在时
 
 			            long long ll_oldval;
 
 if (RedisModule_StringToLongLong(oldval, &ll_oldval) != REDISMODULE_OK) {
 
 			                return RedisModule_ReplyWithError(ctx, "ERR hash value is not an integer");
 
 }
 
 
 
 			            ll_newvalue = ll_newvalue + ll_oldval; // 累加得到新值
 
 			            RedisModuleString* newval = RedisModule_CreateStringFromLongLong(ctx, ll_newvalue);
 
 			            RedisModule_HashSet(key,REDISMODULE_HASH_NONE, field, newval, NULL);
 
 }
 
 
 
 			        newval_array[i] = ll_newvalue;
 
 }
 
 
 
 			    RedisModule_ReplyWithArray(ctx, count); // 返回数组类型的Reply
 
 for (std::vector<long long>::size_type i=0; i<newval_array.size(); ++i)
 
 			        RedisModule_ReplyWithLongLong(ctx, newval_array[i]);
 
 			    return REDISMODULE_OK;
 
 }
 
 
 
 			extern "C"
 
 int RedisModule_OnLoad(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
 
 if (RedisModule_Init(ctx, "command_extension", 1, REDISMODULE_APIVER_1)
 
 == REDISMODULE_ERR) return REDISMODULE_ERR;
 
 
 
 /* Log the list of parameters passing loading the module. */
 
 for (int j=0; j<argc; ++j) {
 
 const char *s = RedisModule_StringPtrLen(argv[j], NULL);
 
 			        printf("Module loaded with ARGV[%d] = %s\n", j, s);
 
 }
 
 
 
 if (RedisModule_CreateCommand(ctx, "ex.incrbyex",
 
 			            incrbyex_RedisCommand,"write", 1, 1, 1) == REDISMODULE_ERR)
 
 			        return REDISMODULE_ERR;
 
 
 
 if (RedisModule_CreateCommand(ctx, "ex.hmincrby",
 
 			            hmincrby_RedisCommand,"write", 1, 1, 1) == REDISMODULE_ERR)
 
 			        return REDISMODULE_ERR;
 
 			    return REDISMODULE_OK;
 
 } 

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏黑泽君的专栏

常用的 default.properties 文件 + 常用的 struts-default.xml 文件 + 常用的 struts-plugin.xml 文件 + 常用的 struts.xml 文件

常用的 default.properties 文件,所在位置:\struts-2.3.15.3-all\struts-2.3.15.3\apps\struts2...

924
来自专栏用户2442861的专栏

Intellij IDEA 14中使用MyBatis-generator 自动生成MyBatis代码

  Intellij IDEA 14 作为Java IDE 神器,接触后发现,非常好用,对它爱不释手,打算离开eclipse和myeclipse,投入Inte...

6261
来自专栏程序员与猫

Log system architecture

Keywords: Collector, Processor, Aggregator

1761
来自专栏逍遥剑客的游戏开发

做了Nebula3的应用程序向导

1263
来自专栏有刻

Java 小记 — Spring Boot 注解

37813
来自专栏微信公众号:Java团长

详解Intellij IDEA搭建SpringBoot

Spring Boot是由Pivotal团队提供的全新框架,其设计目的是用来简化新Spring应用的初始搭建以及开发过程。该框架使用了特定的方式来进行配置,从而...

5374
来自专栏Ryan Miao

Spring-AOP实践 - 统计访问时间

公司的项目有的页面超级慢,20s以上,不知道用户会不会疯掉,于是老大说这个页面要性能优化。于是,首先就要搞清楚究竟是哪一步耗时太多。 我采用spring aop...

5108
来自专栏杨建荣的学习笔记

关于CPU使用率高的awr分析(r8笔记第46天)

今天看到一个报警信息,大体是CPU使用异常。 ZABBIX-监控系统: ------------------------------------ 报警内容: ...

3345
来自专栏大大的微笑

深入理解JVM原理之编译openjdk7

Java虚拟机种类很多例如:HotSpot 丶 JRockit 丶 J9等 我们一般用的就是HotSpot,如果我们电脑上装了JDK,可以运行java -ver...

2869
来自专栏Albert陈凯

2018-11-17 面试必问问题TransactionalJava事务之一——Java事务的基本问题

关于加@Transactional注解的方法之间调用,事务是否生效的问题 https://blog.csdn.net/blacktal/article/det...

3052

扫码关注云+社区

领取腾讯云代金券