漫话Redis源码之一百零六

2024-02-06 09:32
文章标签 源码 redis 漫话 一百零六

本文主要是介绍漫话Redis源码之一百零六，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

blocking command相关的操作，简单看一下就行了：

#define UNUSED(x) (void)(x)/* Reply callback for blocking command BLOCK.DEBUG */
int HelloBlock_Reply(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {UNUSED(argv);UNUSED(argc);int *myint = RedisModule_GetBlockedClientPrivateData(ctx);return RedisModule_ReplyWithLongLong(ctx,*myint);
}/* Timeout callback for blocking command BLOCK.DEBUG */
int HelloBlock_Timeout(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {UNUSED(argv);UNUSED(argc);RedisModuleBlockedClient *bc = RedisModule_GetBlockedClientHandle(ctx);RedisModule_BlockedClientMeasureTimeEnd(bc);return RedisModule_ReplyWithSimpleString(ctx,"Request timedout");
}/* Private data freeing callback for BLOCK.DEBUG command. */
void HelloBlock_FreeData(RedisModuleCtx *ctx, void *privdata) {UNUSED(ctx);RedisModule_Free(privdata);
}/* The thread entry point that actually executes the blocking part* of the command BLOCK.DEBUG. */
void *BlockDebug_ThreadMain(void *arg) {void **targ = arg;RedisModuleBlockedClient *bc = targ[0];long long delay = (unsigned long)targ[1];long long enable_time_track = (unsigned long)targ[2];if (enable_time_track)RedisModule_BlockedClientMeasureTimeStart(bc);RedisModule_Free(targ);struct timespec ts;ts.tv_sec = delay / 1000;ts.tv_nsec = (delay % 1000) * 1000000;nanosleep(&ts, NULL);int *r = RedisModule_Alloc(sizeof(int));*r = rand();if (enable_time_track)RedisModule_BlockedClientMeasureTimeEnd(bc);RedisModule_UnblockClient(bc,r);return NULL;
}/* The thread entry point that actually executes the blocking part* of the command BLOCK.DOUBLE_DEBUG. */
void *DoubleBlock_ThreadMain(void *arg) {void **targ = arg;RedisModuleBlockedClient *bc = targ[0];long long delay = (unsigned long)targ[1];RedisModule_BlockedClientMeasureTimeStart(bc);RedisModule_Free(targ);struct timespec ts;ts.tv_sec = delay / 1000;ts.tv_nsec = (delay % 1000) * 1000000;nanosleep(&ts, NULL);int *r = RedisModule_Alloc(sizeof(int));*r = rand();RedisModule_BlockedClientMeasureTimeEnd(bc);/* call again RedisModule_BlockedClientMeasureTimeStart() and* RedisModule_BlockedClientMeasureTimeEnd and ensure that the* total execution time is 2x the delay. */RedisModule_BlockedClientMeasureTimeStart(bc);nanosleep(&ts, NULL);RedisModule_BlockedClientMeasureTimeEnd(bc);RedisModule_UnblockClient(bc,r);return NULL;
}void HelloBlock_Disconnected(RedisModuleCtx *ctx, RedisModuleBlockedClient *bc) {RedisModule_Log(ctx,"warning","Blocked client %p disconnected!",(void*)bc);
}/* BLOCK.DEBUG <delay_ms> <timeout_ms> -- Block for <count> milliseconds, then reply with* a random number. Timeout is the command timeout, so that you can test* what happens when the delay is greater than the timeout. */
int HelloBlock_RedisCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {if (argc != 3) return RedisModule_WrongArity(ctx);long long delay;long long timeout;if (RedisModule_StringToLongLong(argv[1],&delay) != REDISMODULE_OK) {return RedisModule_ReplyWithError(ctx,"ERR invalid count");}if (RedisModule_StringToLongLong(argv[2],&timeout) != REDISMODULE_OK) {return RedisModule_ReplyWithError(ctx,"ERR invalid count");}pthread_t tid;RedisModuleBlockedClient *bc = RedisModule_BlockClient(ctx,HelloBlock_Reply,HelloBlock_Timeout,HelloBlock_FreeData,timeout);/* Here we set a disconnection handler, however since this module will* block in sleep() in a thread, there is not much we can do in the* callback, so this is just to show you the API. */RedisModule_SetDisconnectCallback(bc,HelloBlock_Disconnected);/* Now that we setup a blocking client, we need to pass the control* to the thread. However we need to pass arguments to the thread:* the delay and a reference to the blocked client handle. */void **targ = RedisModule_Alloc(sizeof(void*)*3);targ[0] = bc;targ[1] = (void*)(unsigned long) delay;// pass 1 as flag to enable time trackingtarg[2] = (void*)(unsigned long) 1;if (pthread_create(&tid,NULL,BlockDebug_ThreadMain,targ) != 0) {RedisModule_AbortBlock(bc);return RedisModule_ReplyWithError(ctx,"-ERR Can't start thread");}return REDISMODULE_OK;
}/* BLOCK.DEBUG_NOTRACKING <delay_ms> <timeout_ms> -- Block for <count> milliseconds, then reply with* a random number. Timeout is the command timeout, so that you can test* what happens when the delay is greater than the timeout.* this command does not track background time so the background time should no appear in stats*/
int HelloBlockNoTracking_RedisCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {if (argc != 3) return RedisModule_WrongArity(ctx);long long delay;long long timeout;if (RedisModule_StringToLongLong(argv[1],&delay) != REDISMODULE_OK) {return RedisModule_ReplyWithError(ctx,"ERR invalid count");}if (RedisModule_StringToLongLong(argv[2],&timeout) != REDISMODULE_OK) {return RedisModule_ReplyWithError(ctx,"ERR invalid count");}pthread_t tid;RedisModuleBlockedClient *bc = RedisModule_BlockClient(ctx,HelloBlock_Reply,HelloBlock_Timeout,HelloBlock_FreeData,timeout);/* Here we set a disconnection handler, however since this module will* block in sleep() in a thread, there is not much we can do in the* callback, so this is just to show you the API. */RedisModule_SetDisconnectCallback(bc,HelloBlock_Disconnected);/* Now that we setup a blocking client, we need to pass the control* to the thread. However we need to pass arguments to the thread:* the delay and a reference to the blocked client handle. */void **targ = RedisModule_Alloc(sizeof(void*)*3);targ[0] = bc;targ[1] = (void*)(unsigned long) delay;// pass 0 as flag to enable time trackingtarg[2] = (void*)(unsigned long) 0;if (pthread_create(&tid,NULL,BlockDebug_ThreadMain,targ) != 0) {RedisModule_AbortBlock(bc);return RedisModule_ReplyWithError(ctx,"-ERR Can't start thread");}return REDISMODULE_OK;
}/* BLOCK.DOUBLE_DEBUG <delay_ms> -- Block for 2 x <count> milliseconds,* then reply with a random number.* This command is used to test multiple calls to RedisModule_BlockedClientMeasureTimeStart()* and RedisModule_BlockedClientMeasureTimeEnd() within the same execution. */
int HelloDoubleBlock_RedisCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {if (argc != 2) return RedisModule_WrongArity(ctx);long long delay;if (RedisModule_StringToLongLong(argv[1],&delay) != REDISMODULE_OK) {return RedisModule_ReplyWithError(ctx,"ERR invalid count");}pthread_t tid;RedisModuleBlockedClient *bc = RedisModule_BlockClient(ctx,HelloBlock_Reply,HelloBlock_Timeout,HelloBlock_FreeData,0);/* Now that we setup a blocking client, we need to pass the control* to the thread. However we need to pass arguments to the thread:* the delay and a reference to the blocked client handle. */void **targ = RedisModule_Alloc(sizeof(void*)*2);targ[0] = bc;targ[1] = (void*)(unsigned long) delay;if (pthread_create(&tid,NULL,DoubleBlock_ThreadMain,targ) != 0) {RedisModule_AbortBlock(bc);return RedisModule_ReplyWithError(ctx,"-ERR Can't start thread");}return REDISMODULE_OK;
}RedisModuleBlockedClient *blocked_client = NULL;/* BLOCK.BLOCK [TIMEOUT] -- Blocks the current client until released* or TIMEOUT seconds. If TIMEOUT is zero, no timeout function is* registered.*/
int Block_RedisCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {if (RedisModule_IsBlockedReplyRequest(ctx)) {RedisModuleString *r = RedisModule_GetBlockedClientPrivateData(ctx);return RedisModule_ReplyWithString(ctx, r);} else if (RedisModule_IsBlockedTimeoutRequest(ctx)) {RedisModule_UnblockClient(blocked_client, NULL); /* Must be called to avoid leaks. */blocked_client = NULL;return RedisModule_ReplyWithSimpleString(ctx, "Timed out");}if (argc != 2) return RedisModule_WrongArity(ctx);long long timeout;if (RedisModule_StringToLongLong(argv[1], &timeout) != REDISMODULE_OK) {return RedisModule_ReplyWithError(ctx, "ERR invalid timeout");}if (blocked_client) {return RedisModule_ReplyWithError(ctx, "ERR another client already blocked");}/* Block client. We use this function as both a reply and optional timeout* callback and differentiate the different code flows above.*/blocked_client = RedisModule_BlockClient(ctx, Block_RedisCommand,timeout > 0 ? Block_RedisCommand : NULL, NULL, timeout);return REDISMODULE_OK;
}/* BLOCK.IS_BLOCKED -- Returns 1 if we have a blocked client, or 0 otherwise.*/
int IsBlocked_RedisCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {UNUSED(argv);UNUSED(argc);RedisModule_ReplyWithLongLong(ctx, blocked_client ? 1 : 0);return REDISMODULE_OK;
}/* BLOCK.RELEASE [reply] -- Releases the blocked client and produce the specified reply.*/
int Release_RedisCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {if (argc != 2) return RedisModule_WrongArity(ctx);if (!blocked_client) {return RedisModule_ReplyWithError(ctx, "ERR No blocked client");}RedisModuleString *replystr = argv[1];RedisModule_RetainString(ctx, replystr);int err = RedisModule_UnblockClient(blocked_client, replystr);blocked_client = NULL;RedisModule_ReplyWithSimpleString(ctx, "OK");return REDISMODULE_OK;
}int RedisModule_OnLoad(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {UNUSED(argv);UNUSED(argc);if (RedisModule_Init(ctx,"block",1,REDISMODULE_APIVER_1)== REDISMODULE_ERR) return REDISMODULE_ERR;if (RedisModule_CreateCommand(ctx,"block.debug",HelloBlock_RedisCommand,"",0,0,0) == REDISMODULE_ERR)return REDISMODULE_ERR;if (RedisModule_CreateCommand(ctx,"block.double_debug",HelloDoubleBlock_RedisCommand,"",0,0,0) == REDISMODULE_ERR)return REDISMODULE_ERR;if (RedisModule_CreateCommand(ctx,"block.debug_no_track",HelloBlockNoTracking_RedisCommand,"",0,0,0) == REDISMODULE_ERR)return REDISMODULE_ERR;if (RedisModule_CreateCommand(ctx, "block.block",Block_RedisCommand, "", 0, 0, 0) == REDISMODULE_ERR)return REDISMODULE_ERR;if (RedisModule_CreateCommand(ctx,"block.is_blocked",IsBlocked_RedisCommand,"",0,0,0) == REDISMODULE_ERR)return REDISMODULE_ERR;if (RedisModule_CreateCommand(ctx,"block.release",Release_RedisCommand,"",0,0,0) == REDISMODULE_ERR)return REDISMODULE_ERR;return REDISMODULE_OK;
}

这篇关于漫话Redis源码之一百零六的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！


http://www.chinasem.cn/article/683860。

相关文章

Java 正则表达式URL 匹配与源码全解析

Java 正则表达式URL 匹配与源码全解析

《Java正则表达式URL匹配与源码全解析》在Web应用开发中,我们经常需要对URL进行格式验证,今天我们结合Java的Pattern和Matcher类,深入理解正则表达式在实际应用中... 目录1.正则表达式分解：2. 添加域名匹配 (2)3. 添加路径和查询参数匹配 (3) 4. 最终优化版本5.设计思
阅读更多...
Redis在windows环境下如何启动

Redis在windows环境下如何启动

《Redis在windows环境下如何启动》：本文主要介绍Redis在windows环境下如何启动的实现方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录Redis在Windows环境下启动1.在redis的安装目录下2.输入·redis-server.exe
阅读更多...
Redis实现延迟任务的三种方法详解

Redis实现延迟任务的三种方法详解

《Redis实现延迟任务的三种方法详解》延迟任务（DelayedTask）是指在未来的某个时间点,执行相应的任务,本文为大家整理了三种常见的实现方法,感兴趣的小伙伴可以参考一下... 目录1.前言2.Redis如何实现延迟任务3.代码实现3.1. 过期键通知事件实现3.2. 使用ZSet实现延迟任务3.3
阅读更多...
Java调用C++动态库超详细步骤讲解(附源码)

Java调用C++动态库超详细步骤讲解(附源码)

《Java调用C++动态库超详细步骤讲解(附源码)》C语言因其高效和接近硬件的特性,时常会被用在性能要求较高或者需要直接操作硬件的场合,：本文主要介绍Java调用C++动态库的相关资料,文中通过代... 目录一、直接调用C++库第一步：动态库生成（vs2017+qt5.12.10）第二步：Java调用C++
阅读更多...
Redis分片集群的实现

Redis分片集群的实现

《Redis分片集群的实现》Redis分片集群是一种将Redis数据库分散到多个节点上的方式,以提供更高的性能和可伸缩性,本文主要介绍了Redis分片集群的实现,具有一定的参考价值,感兴趣的可以了解一... 目录1. Redis Cluster的核心概念哈希槽（Hash Slots）主从复制与故障转移2.
阅读更多...
Python实现无痛修改第三方库源码的方法详解

Python实现无痛修改第三方库源码的方法详解

《Python实现无痛修改第三方库源码的方法详解》很多时候,我们下载的第三方库是不会有需求不满足的情况,但也有极少的情况,第三方库没有兼顾到需求,本文将介绍几个修改源码的操作,大家可以根据需求进行选择... 目录需求不符合模拟示例 1. 修改源文件2. 继承修改3. 猴子补丁4. 追踪局部变量需求不符合很
阅读更多...
Redis 中的热点键和数据倾斜示例详解

Redis 中的热点键和数据倾斜示例详解

《Redis中的热点键和数据倾斜示例详解》热点键是指在Redis中被频繁访问的特定键,这些键由于其高访问频率,可能导致Redis服务器的性能问题,尤其是在高并发场景下,本文给大家介绍Redis中的热... 目录Redis 中的热点键和数据倾斜热点键（Hot Key）定义特点应对策略示例数据倾斜（Data S
阅读更多...
redis+lua实现分布式限流的示例

redis+lua实现分布式限流的示例

《redis+lua实现分布式限流的示例》本文主要介绍了redis+lua实现分布式限流的示例,可以实现复杂的限流逻辑,如滑动窗口限流,并且避免了多步操作导致的并发问题,具有一定的参考价值,感兴趣的可... 目录为什么使用Redis+Lua实现分布式限流使用ZSET也可以实现限流，为什么选择lua的方式实现
阅读更多...
Redis中管道操作pipeline的实现

Redis中管道操作pipeline的实现

《Redis中管道操作pipeline的实现》RedisPipeline是一种优化客户端与服务器通信的技术,通过批量发送和接收命令减少网络往返次数,提高命令执行效率,本文就来介绍一下Redis中管道操... 目录什么是pipeline场景一:我要向Redis新增大批量的数据分批处理事务（ MULTI/EXE
阅读更多...
Redis中高并发读写性能的深度解析与优化

Redis中高并发读写性能的深度解析与优化

《Redis中高并发读写性能的深度解析与优化》Redis作为一款高性能的内存数据库,广泛应用于缓存、消息队列、实时统计等场景,本文将深入探讨Redis的读写并发能力,感兴趣的小伙伴可以了解下... 目录引言一、Redis 并发能力概述1.1 Redis 的读写性能1.2 影响 Redis 并发能力的因素二、
阅读更多...

Copyright China编程 23002807@qq.com

沪ICP备2023033072号-2