11c6ce812a
The implementation expose the following new functions: 1. RedisModule_CursorCreate - allow to create a new cursor object for keys scanning 2. RedisModule_CursorRestart - restart an existing cursor to restart the scan 3. RedisModule_CursorDestroy - destroy an existing cursor 4. RedisModule_Scan - scan keys The RedisModule_Scan function gets a cursor object, a callback and void* (used as user private data). The callback will be called for each key in the database proving the key name and the value as RedisModuleKey.
63 lines
1.6 KiB
C
63 lines
1.6 KiB
C
#define REDISMODULE_EXPERIMENTAL_API
|
|
#include "redismodule.h"
|
|
|
|
#include <string.h>
|
|
#include <assert.h>
|
|
#include <unistd.h>
|
|
|
|
#define UNUSED(V) ((void) V)
|
|
|
|
typedef struct scan_pd{
|
|
size_t nkeys;
|
|
RedisModuleCtx *ctx;
|
|
} scan_pd;
|
|
|
|
void scan_callback(void *privdata, RedisModuleString* keyname, RedisModuleKey* key){
|
|
scan_pd* pd = privdata;
|
|
RedisModule_ReplyWithArray(pd->ctx, 2);
|
|
|
|
RedisModule_ReplyWithString(pd->ctx, keyname);
|
|
if(key && RedisModule_KeyType(key) == REDISMODULE_KEYTYPE_STRING){
|
|
size_t len;
|
|
char * data = RedisModule_StringDMA(key, &len, REDISMODULE_READ);
|
|
RedisModule_ReplyWithStringBuffer(pd->ctx, data, len);
|
|
}else{
|
|
RedisModule_ReplyWithNull(pd->ctx);
|
|
}
|
|
pd->nkeys++;
|
|
}
|
|
|
|
int scan_keys_values(RedisModuleCtx *ctx, RedisModuleString **argv, int argc)
|
|
{
|
|
scan_pd pd = {
|
|
.nkeys = 0,
|
|
.ctx = ctx,
|
|
};
|
|
|
|
RedisModule_ReplyWithArray(ctx, REDISMODULE_POSTPONED_ARRAY_LEN);
|
|
|
|
RedisModuleCursor* cursor = RedisModule_CursorCreate();
|
|
while(RedisModule_Scan(ctx, cursor, scan_callback, &pd));
|
|
RedisModule_CursorDestroy(cursor);
|
|
|
|
RedisModule_ReplySetArrayLength(ctx, pd.nkeys);
|
|
return 0;
|
|
}
|
|
|
|
int RedisModule_OnLoad(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
|
|
UNUSED(argv);
|
|
UNUSED(argc);
|
|
if (RedisModule_Init(ctx, "scan", 1, REDISMODULE_APIVER_1)== REDISMODULE_ERR)
|
|
return REDISMODULE_ERR;
|
|
|
|
if (RedisModule_CreateCommand(ctx, "scan.scankeysvalues", scan_keys_values, "", 0, 0, 0) == REDISMODULE_ERR)
|
|
return REDISMODULE_ERR;
|
|
|
|
return REDISMODULE_OK;
|
|
}
|
|
|
|
|
|
|
|
|
|
|