漫话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智听未来一站式有声阅读平台听书系统小程序源码

JAVA智听未来一站式有声阅读平台听书系统小程序源码

智听未来，一站式有声阅读平台听书系统 🌟&nbsp;开篇：遇见未来，从“智听”开始 在这个快节奏的时代，你是否渴望在忙碌的间隙，找到一片属于自己的宁静角落？是否梦想着能随时随地，沉浸在知识的海洋，或是故事的奇幻世界里？今天，就让我带你一起探索“智听未来”——这一站式有声阅读平台听书系统，它正悄悄改变着我们的阅读方式，让未来触手可及！ 📚&nbsp;第一站：海量资源，应有尽有 走进“智听
阅读更多...
零基础学习Redis(10) -- zset类型命令使用

零基础学习Redis(10) -- zset类型命令使用

zset是有序集合，内部除了存储元素外，还会存储一个score，存储在zset中的元素会按照score的大小升序排列，不同元素的score可以重复，score相同的元素会按照元素的字典序排列。 1. zset常用命令 1.1 zadd  zadd key [NX | XX] [GT | LT]   [CH] [INCR] score member [score member ...]
阅读更多...
Java ArrayList扩容机制 （源码解读）

Java ArrayList扩容机制 （源码解读）

结论：初始长度为10，若所需长度小于1.5倍原长度，则按照1.5倍扩容。若不够用则按照所需长度扩容。 一. 明确类内部重要变量含义         1：数组默认长度         2:这是一个共享的空数组实例，用于明确创建长度为0时的ArrayList ，比如通过 new ArrayList<>(0)，ArrayList 内部的数组 elementData 会指向这个 EMPTY_EL
阅读更多...
如何在Visual Studio中调试.NET源码

如何在Visual Studio中调试.NET源码

今天偶然在看别人代码时，发现在他的代码里使用了Any判断List<T>是否为空。 我一般的做法是先判断是否为null，再判断Count。 看了一下Count的源码如下： 1 [__DynamicallyInvokable]2 public int Count3 {4 [__DynamicallyInvokable]5 get
阅读更多...
工厂ERP管理系统实现源码（JAVA）

工厂ERP管理系统实现源码（JAVA）

工厂进销存管理系统是一个集采购管理、仓库管理、生产管理和销售管理于一体的综合解决方案。该系统旨在帮助企业优化流程、提高效率、降低成本，并实时掌握各环节的运营状况。 在采购管理方面，系统能够处理采购订单、供应商管理和采购入库等流程，确保采购过程的透明和高效。仓库管理方面，实现库存的精准管理，包括入库、出库、盘点等操作，确保库存数据的准确性和实时性。 生产管理模块则涵盖了生产计划制定、物料需求计划、
阅读更多...
Spring 源码解读：自定义实现Bean定义的注册与解析

Spring 源码解读：自定义实现Bean定义的注册与解析

引言 在Spring框架中，Bean的注册与解析是整个依赖注入流程的核心步骤。通过Bean定义，Spring容器知道如何创建、配置和管理每个Bean实例。本篇文章将通过实现一个简化版的Bean定义注册与解析机制，帮助你理解Spring框架背后的设计逻辑。我们还将对比Spring中的BeanDefinition和BeanDefinitionRegistry，以全面掌握Bean注册和解析的核心原理。
阅读更多...
音视频入门基础：WAV专题（10）——FFmpeg源码中计算WAV音频文件每个packet的pts、dts的实现

音视频入门基础：WAV专题（10）——FFmpeg源码中计算WAV音频文件每个packet的pts、dts的实现

一、引言 从文章《音视频入门基础：WAV专题（6）——通过FFprobe显示WAV音频文件每个数据包的信息》中我们可以知道，通过FFprobe命令可以打印WAV音频文件每个packet（也称为数据包或多媒体包）的信息，这些信息包含该packet的pts、dts： 打印出来的“pts”实际是AVPacket结构体中的成员变量pts，是以AVStream->time_base为单位的显
阅读更多...
kubelet组件的启动流程源码分析

kubelet组件的启动流程源码分析

概述 摘要: 本文将总结kubelet的作用以及原理，在有一定基础认识的前提下，通过阅读kubelet源码，对kubelet组件的启动流程进行分析。 正文 kubelet的作用 这里对kubelet的作用做一个简单总结。 节点管理 节点的注册 节点状态更新 容器管理(pod生命周期管理) 监听apiserver的容器事件 容器的创建、删除(CRI) 容器的网络的创建与删除
阅读更多...
Redis中使用布隆过滤器解决缓存穿透问题

Redis中使用布隆过滤器解决缓存穿透问题

一、缓存穿透(失效)问题 缓存穿透是指查询一个一定不存在的数据，由于缓存中没有命中，会去数据库中查询，而数据库中也没有该数据，并且每次查询都不会命中缓存，从而每次请求都直接打到了数据库上，这会给数据库带来巨大压力。 二、布隆过滤器原理 布隆过滤器（Bloom Filter）是一种空间效率很高的随机数据结构，它利用多个不同的哈希函数将一个元素映射到一个位数组中的多个位置，并将这些位置的值置
阅读更多...
red5-server源码

red5-server源码

red5-server源码：https://github.com/Red5/red5-server
阅读更多...

Copyright China编程 23002807@qq.com

沪ICP备2023033072号-2