转载

让redis支持动态库

So, let’s put some fun in redis

redis的命令列表是写死在代码里面,编译后就没法修改(跟nginx一个德行)

硬编码一些额外的方法到redis中

最初写的一个hello world级的测试命令

void wendalCommand(redisClient *c) {     struct timeval tv;           // 从timeCommand拷贝的...     addReplyMultiBulkLen(c,3);   // 共返回3个结果     addReplyBulkCString(c, "Hi,Wendal"); // 输出个字符串啦     gettimeofday(&tv,NULL);              // 获取时间, 也是从timeCommand拷贝的     addReplyBulkLongLong(c,tv.tv_sec);   // 不解释了     addReplyBulkLongLong(c,tv.tv_usec);  // 不解释了 }

然后在redisCommandTable中声明一个新的命令,就搞定了

{"wendal", wendalCommand,1,"rR",0,NULL,0,0,0,0,0}

做得更灵活?上动态库,哈哈

so,我声明了2个新的命令, 加载/卸载动态库

    {"loadlib",loadlibCommand,2,"wmaR",0,NULL,0,0,0,0,0},     {"unloadlib",unloadlibCommand,2,"wmaR",0,NULL,0,0,0,0,0}

其中, 加载动态库,就是接受一个路径,并执行其redis_lib_init方法

// 简化版的loadlibCommand void loadlibCommand(redisClient *c) {     void *handle;     char *error;     int (*redis_lib_init)(redisClient*, dict*);     int re;          handle = dlopen(c->argv[1]->ptr, RTLD_LAZY);     if (!handle) {         addReplyError(c, dlerror());         return;     }     redis_lib_init = dlsym(handle, "redis_lib_init"); // 查找redis_lib_init     if ((error = dlerror()) != NULL)  {         dlclose(handle);         addReplyError(c, error);         return;     }     re = redis_lib_init(c, server.commands); // 执行之, 返回0就成功     if (re) {         dlclose(handle);         addReplyError(c, "lib init error");         return;     }     addReply(c,shared.ok);     server.dirty ++;                        // 使dirty自增,这样就能同步到slave了 }

redis动态库示例

禁用flushdb命令

#include <redis.h>  static void* flushdb;  extern int redis_lib_init(redisClient *c) {     flushdb = dictFetchValue(server.commands, sdsnew("FLUSHDB")); // 命令列表就是server.commands中,一个dict     if (!flushdb) return 1; // 找不到? 被其他库删掉了?     dictDeleteNoFree(server.commands, sdsnew("FLUSHDB")); // 删除但不要执行free     return 0; }  extern void redis_lib_depose(redisClient *c) {     dictAdd(server.commands, sdsnew("FLUSHDB"), flushdb); // 卸载的时候就赋值回去,呵呵     return; }