漫话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。

相关文章

springboot+redis实现订单过期(超时取消)功能的方法详解

springboot+redis实现订单过期(超时取消)功能的方法详解

《springboot+redis实现订单过期(超时取消)功能的方法详解》在SpringBoot中使用Redis实现订单过期（超时取消）功能,有多种成熟方案,本文为大家整理了几个详细方法,文中的示例代... 目录一、Redis键过期回调方案（推荐）1. 配置Redis监听器2. 监听键过期事件3. Redi
阅读更多...
使用Redis实现会话管理的示例代码

使用Redis实现会话管理的示例代码

《使用Redis实现会话管理的示例代码》文章介绍了如何使用Redis实现会话管理,包括会话的创建、读取、更新和删除操作,通过设置会话超时时间并重置,可以确保会话在用户持续活动期间不会过期,此外,展示了... 目录1. 会话管理的基本概念2. 使用Redis实现会话管理2.1 引入依赖2.2 会话管理基本操作
阅读更多...
Redis 命令详解与实战案例

Redis 命令详解与实战案例

《Redis命令详解与实战案例》本文详细介绍了Redis的基础知识、核心数据结构与命令、高级功能与命令、最佳实践与性能优化,以及实战应用场景,通过实战案例,展示了如何使用Redis构建高性能应用系统... 目录Redis 命令详解与实战案例一、Redis 基础介绍二、Redis 核心数据结构与命令1. 字符
阅读更多...
SpringBoot18 redis的配置方法

SpringBoot18 redis的配置方法

《SpringBoot18redis的配置方法》本文介绍在SpringBoot项目中集成和使用Redis的方法,包括添加依赖、配置文件、自定义序列化方式、使用方式、实际使用示例、常见操作总结以及注意... 目录一、Spring Boot 中使用 Redis1. 添加依赖2. 配置文件3. Redis 配置类
阅读更多...
Redis中群集三种模式的实现

Redis中群集三种模式的实现

《Redis中群集三种模式的实现》Redis群集有三种模式,分别是主从同步/复制、哨兵模式、Cluster,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面... 目录1. Redis三种模式概述2、Redis 主从复制2.1 主从复制的作用2.2 主从复制流程2
阅读更多...
Redis的安全机制详细介绍及配置方法

Redis的安全机制详细介绍及配置方法

《Redis的安全机制详细介绍及配置方法》本文介绍Redis安全机制的配置方法,包括绑定IP地址、设置密码、保护模式、禁用危险命令、防火墙限制、TLS加密、客户端连接限制、最大内存使用和日志审计等,通... 目录1. 绑定 IP 地址2. 设置密码3. 保护模式4. 禁用危险命令5. 通过防火墙限制访问6.
阅读更多...
深入理解Redis线程模型的原理及使用

深入理解Redis线程模型的原理及使用

《深入理解Redis线程模型的原理及使用》Redis的线程模型整体还是多线程的,只是后台执行指令的核心线程是单线程的,整个线程模型可以理解为还是以单线程为主,基于这种单线程为主的线程模型,不同客户端的... 目录1 Redis是单线程www.chinasem.cn还是多线程2 Redis如何保证指令原子性2.
阅读更多...
Docker + Redis 部署集群的实现步骤

Docker + Redis 部署集群的实现步骤

《Docker+Redis部署集群的实现步骤》本文详细介绍了在三台服务器上部署高可用Redis集群的完整流程,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋... 目录一、环境准备1. 服务器规划（3 台服务器）2. 防火墙配置（三台服务器均执行）3. 安装 docke
阅读更多...
Redis中删除策略的几种实现方式

Redis中删除策略的几种实现方式

《Redis中删除策略的几种实现方式》本文详细介绍了Redis的过期键删除策略和内存淘汰策略,过期键删除策略包括定时删除、惰性删除和定期删除,具有一定的参考价值,感兴趣的可以了解一下... 目录前言一、设计背景：为什么需要删除策略？二、第一类：过期键的 3 种核心删除策略1. 定时删除（Timed Dele
阅读更多...
Spring Boot整合Redis注解实现增删改查功能(Redis注解使用)

Spring Boot整合Redis注解实现增删改查功能(Redis注解使用)

《SpringBoot整合Redis注解实现增删改查功能(Redis注解使用)》文章介绍了如何使用SpringBoot整合Redis注解实现增删改查功能,包括配置、实体类、Repository、Se... 目录配置Redis连接定义实体类创建Repository接口增删改查操作示例插入数据查询数据删除数据更
阅读更多...

Copyright China编程 23002807@qq.com

沪ICP备2023033072号-2