漫话Redis源码之九十七

2024-02-06 09:32
文章标签 源码 redis 漫话 九十七

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

这里都是Event相关的代码，非核心内容，不用细看：

#include "redismodule.h"
#include <stdio.h>
#include <string.h>/* We need to store events to be able to test and see what we got, and we can't* store them in the key-space since that would mess up rdb loading (duplicates)* and be lost of flushdb. */
RedisModuleDict *event_log = NULL;typedef struct EventElement {long count;RedisModuleString *last_val_string;long last_val_int;
} EventElement;void LogStringEvent(RedisModuleCtx *ctx, const char* keyname, const char* data) {EventElement *event = RedisModule_DictGetC(event_log, (void*)keyname, strlen(keyname), NULL);if (!event) {event = RedisModule_Alloc(sizeof(EventElement));memset(event, 0, sizeof(EventElement));RedisModule_DictSetC(event_log, (void*)keyname, strlen(keyname), event);}if (event->last_val_string) RedisModule_FreeString(ctx, event->last_val_string);event->last_val_string = RedisModule_CreateString(ctx, data, strlen(data));event->count++;
}void LogNumericEvent(RedisModuleCtx *ctx, const char* keyname, long data) {REDISMODULE_NOT_USED(ctx);EventElement *event = RedisModule_DictGetC(event_log, (void*)keyname, strlen(keyname), NULL);if (!event) {event = RedisModule_Alloc(sizeof(EventElement));memset(event, 0, sizeof(EventElement));RedisModule_DictSetC(event_log, (void*)keyname, strlen(keyname), event);}event->last_val_int = data;event->count++;
}void FreeEvent(RedisModuleCtx *ctx, EventElement *event) {if (event->last_val_string)RedisModule_FreeString(ctx, event->last_val_string);RedisModule_Free(event);
}int cmdEventCount(RedisModuleCtx *ctx, RedisModuleString **argv, int argc)
{if (argc != 2){RedisModule_WrongArity(ctx);return REDISMODULE_OK;}EventElement *event = RedisModule_DictGet(event_log, argv[1], NULL);RedisModule_ReplyWithLongLong(ctx, event? event->count: 0);return REDISMODULE_OK;
}int cmdEventLast(RedisModuleCtx *ctx, RedisModuleString **argv, int argc)
{if (argc != 2){RedisModule_WrongArity(ctx);return REDISMODULE_OK;}EventElement *event = RedisModule_DictGet(event_log, argv[1], NULL);if (event && event->last_val_string)RedisModule_ReplyWithString(ctx, event->last_val_string);else if (event)RedisModule_ReplyWithLongLong(ctx, event->last_val_int);elseRedisModule_ReplyWithNull(ctx);return REDISMODULE_OK;
}void clearEvents(RedisModuleCtx *ctx)
{RedisModuleString *key;EventElement *event;RedisModuleDictIter *iter = RedisModule_DictIteratorStart(event_log, "^", NULL);while((key = RedisModule_DictNext(ctx, iter, (void**)&event)) != NULL) {event->count = 0;event->last_val_int = 0;if (event->last_val_string) RedisModule_FreeString(ctx, event->last_val_string);event->last_val_string = NULL;RedisModule_DictDel(event_log, key, NULL);RedisModule_Free(event);}RedisModule_DictIteratorStop(iter);
}int cmdEventsClear(RedisModuleCtx *ctx, RedisModuleString **argv, int argc)
{REDISMODULE_NOT_USED(argc);REDISMODULE_NOT_USED(argv);clearEvents(ctx);return REDISMODULE_OK;
}/* Client state change callback. */
void clientChangeCallback(RedisModuleCtx *ctx, RedisModuleEvent e, uint64_t sub, void *data)
{REDISMODULE_NOT_USED(e);RedisModuleClientInfo *ci = data;char *keyname = (sub == REDISMODULE_SUBEVENT_CLIENT_CHANGE_CONNECTED) ?"client-connected" : "client-disconnected";LogNumericEvent(ctx, keyname, ci->id);
}void flushdbCallback(RedisModuleCtx *ctx, RedisModuleEvent e, uint64_t sub, void *data)
{REDISMODULE_NOT_USED(e);RedisModuleFlushInfo *fi = data;char *keyname = (sub == REDISMODULE_SUBEVENT_FLUSHDB_START) ?"flush-start" : "flush-end";LogNumericEvent(ctx, keyname, fi->dbnum);
}void roleChangeCallback(RedisModuleCtx *ctx, RedisModuleEvent e, uint64_t sub, void *data)
{REDISMODULE_NOT_USED(e);REDISMODULE_NOT_USED(data);RedisModuleReplicationInfo *ri = data;char *keyname = (sub == REDISMODULE_EVENT_REPLROLECHANGED_NOW_MASTER) ?"role-master" : "role-replica";LogStringEvent(ctx, keyname, ri->masterhost);
}void replicationChangeCallback(RedisModuleCtx *ctx, RedisModuleEvent e, uint64_t sub, void *data)
{REDISMODULE_NOT_USED(e);REDISMODULE_NOT_USED(data);char *keyname = (sub == REDISMODULE_SUBEVENT_REPLICA_CHANGE_ONLINE) ?"replica-online" : "replica-offline";LogNumericEvent(ctx, keyname, 0);
}void rasterLinkChangeCallback(RedisModuleCtx *ctx, RedisModuleEvent e, uint64_t sub, void *data)
{REDISMODULE_NOT_USED(e);REDISMODULE_NOT_USED(data);char *keyname = (sub == REDISMODULE_SUBEVENT_MASTER_LINK_UP) ?"masterlink-up" : "masterlink-down";LogNumericEvent(ctx, keyname, 0);
}void persistenceCallback(RedisModuleCtx *ctx, RedisModuleEvent e, uint64_t sub, void *data)
{REDISMODULE_NOT_USED(e);REDISMODULE_NOT_USED(data);char *keyname = NULL;switch (sub) {case REDISMODULE_SUBEVENT_PERSISTENCE_RDB_START: keyname = "persistence-rdb-start"; break;case REDISMODULE_SUBEVENT_PERSISTENCE_AOF_START: keyname = "persistence-aof-start"; break;case REDISMODULE_SUBEVENT_PERSISTENCE_SYNC_RDB_START: keyname = "persistence-syncrdb-start"; break;case REDISMODULE_SUBEVENT_PERSISTENCE_ENDED: keyname = "persistence-end"; break;case REDISMODULE_SUBEVENT_PERSISTENCE_FAILED: keyname = "persistence-failed"; break;}/* modifying the keyspace from the fork child is not an option, using log instead */RedisModule_Log(ctx, "warning", "module-event-%s", keyname);if (sub == REDISMODULE_SUBEVENT_PERSISTENCE_SYNC_RDB_START)LogNumericEvent(ctx, keyname, 0);
}void loadingCallback(RedisModuleCtx *ctx, RedisModuleEvent e, uint64_t sub, void *data)
{REDISMODULE_NOT_USED(e);REDISMODULE_NOT_USED(data);char *keyname = NULL;switch (sub) {case REDISMODULE_SUBEVENT_LOADING_RDB_START: keyname = "loading-rdb-start"; break;case REDISMODULE_SUBEVENT_LOADING_AOF_START: keyname = "loading-aof-start"; break;case REDISMODULE_SUBEVENT_LOADING_REPL_START: keyname = "loading-repl-start"; break;case REDISMODULE_SUBEVENT_LOADING_ENDED: keyname = "loading-end"; break;case REDISMODULE_SUBEVENT_LOADING_FAILED: keyname = "loading-failed"; break;}LogNumericEvent(ctx, keyname, 0);
}void loadingProgressCallback(RedisModuleCtx *ctx, RedisModuleEvent e, uint64_t sub, void *data)
{REDISMODULE_NOT_USED(e);RedisModuleLoadingProgress *ei = data;char *keyname = (sub == REDISMODULE_SUBEVENT_LOADING_PROGRESS_RDB) ?"loading-progress-rdb" : "loading-progress-aof";LogNumericEvent(ctx, keyname, ei->progress);
}void shutdownCallback(RedisModuleCtx *ctx, RedisModuleEvent e, uint64_t sub, void *data)
{REDISMODULE_NOT_USED(e);REDISMODULE_NOT_USED(data);REDISMODULE_NOT_USED(sub);RedisModule_Log(ctx, "warning", "module-event-%s", "shutdown");
}void cronLoopCallback(RedisModuleCtx *ctx, RedisModuleEvent e, uint64_t sub, void *data)
{REDISMODULE_NOT_USED(e);REDISMODULE_NOT_USED(sub);RedisModuleCronLoop *ei = data;LogNumericEvent(ctx, "cron-loop", ei->hz);
}void moduleChangeCallback(RedisModuleCtx *ctx, RedisModuleEvent e, uint64_t sub, void *data)
{REDISMODULE_NOT_USED(e);RedisModuleModuleChange *ei = data;char *keyname = (sub == REDISMODULE_SUBEVENT_MODULE_LOADED) ?"module-loaded" : "module-unloaded";LogStringEvent(ctx, keyname, ei->module_name);
}void swapDbCallback(RedisModuleCtx *ctx, RedisModuleEvent e, uint64_t sub, void *data)
{REDISMODULE_NOT_USED(e);REDISMODULE_NOT_USED(sub);RedisModuleSwapDbInfo *ei = data;LogNumericEvent(ctx, "swapdb-first", ei->dbnum_first);LogNumericEvent(ctx, "swapdb-second", ei->dbnum_second);
}/* This function must be present on each Redis module. It is used in order to* register the commands into the Redis server. */
int RedisModule_OnLoad(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
#define VerifySubEventSupported(e, s) \if (!RedisModule_IsSubEventSupported(e, s)) { \return REDISMODULE_ERR; \}REDISMODULE_NOT_USED(argv);REDISMODULE_NOT_USED(argc);if (RedisModule_Init(ctx,"testhook",1,REDISMODULE_APIVER_1)== REDISMODULE_ERR) return REDISMODULE_ERR;/* Example on how to check if a server sub event is supported */if (!RedisModule_IsSubEventSupported(RedisModuleEvent_ReplicationRoleChanged, REDISMODULE_EVENT_REPLROLECHANGED_NOW_MASTER)) {return REDISMODULE_ERR;}/* replication related hooks */RedisModule_SubscribeToServerEvent(ctx,RedisModuleEvent_ReplicationRoleChanged, roleChangeCallback);RedisModule_SubscribeToServerEvent(ctx,RedisModuleEvent_ReplicaChange, replicationChangeCallback);RedisModule_SubscribeToServerEvent(ctx,RedisModuleEvent_MasterLinkChange, rasterLinkChangeCallback);/* persistence related hooks */RedisModule_SubscribeToServerEvent(ctx,RedisModuleEvent_Persistence, persistenceCallback);RedisModule_SubscribeToServerEvent(ctx,RedisModuleEvent_Loading, loadingCallback);RedisModule_SubscribeToServerEvent(ctx,RedisModuleEvent_LoadingProgress, loadingProgressCallback);/* other hooks */RedisModule_SubscribeToServerEvent(ctx,RedisModuleEvent_ClientChange, clientChangeCallback);RedisModule_SubscribeToServerEvent(ctx,RedisModuleEvent_FlushDB, flushdbCallback);RedisModule_SubscribeToServerEvent(ctx,RedisModuleEvent_Shutdown, shutdownCallback);RedisModule_SubscribeToServerEvent(ctx,RedisModuleEvent_CronLoop, cronLoopCallback);RedisModule_SubscribeToServerEvent(ctx,RedisModuleEvent_ModuleChange, moduleChangeCallback);RedisModule_SubscribeToServerEvent(ctx,RedisModuleEvent_SwapDB, swapDbCallback);event_log = RedisModule_CreateDict(ctx);if (RedisModule_CreateCommand(ctx,"hooks.event_count", cmdEventCount,"",0,0,0) == REDISMODULE_ERR)return REDISMODULE_ERR;if (RedisModule_CreateCommand(ctx,"hooks.event_last", cmdEventLast,"",0,0,0) == REDISMODULE_ERR)return REDISMODULE_ERR;if (RedisModule_CreateCommand(ctx,"hooks.clear", cmdEventsClear,"",0,0,0) == REDISMODULE_ERR)return REDISMODULE_ERR;return REDISMODULE_OK;
}int RedisModule_OnUnload(RedisModuleCtx *ctx) {clearEvents(ctx);RedisModule_FreeDict(ctx, event_log);event_log = NULL;return REDISMODULE_OK;
}

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


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

相关文章

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