diff options
Diffstat (limited to 'examples/redis-unstable/tests/modules/scan.c')
| -rw-r--r-- | examples/redis-unstable/tests/modules/scan.c | 121 |
1 files changed, 0 insertions, 121 deletions
diff --git a/examples/redis-unstable/tests/modules/scan.c b/examples/redis-unstable/tests/modules/scan.c deleted file mode 100644 index 1723d30..0000000 --- a/examples/redis-unstable/tests/modules/scan.c +++ /dev/null @@ -1,121 +0,0 @@ -#include "redismodule.h" - -#include <string.h> -#include <assert.h> -#include <unistd.h> - -typedef struct { - size_t nkeys; -} scan_strings_pd; - -void scan_strings_callback(RedisModuleCtx *ctx, RedisModuleString* keyname, RedisModuleKey* key, void *privdata) { - scan_strings_pd* pd = privdata; - int was_opened = 0; - if (!key) { - key = RedisModule_OpenKey(ctx, keyname, REDISMODULE_READ); - was_opened = 1; - } - - if (RedisModule_KeyType(key) == REDISMODULE_KEYTYPE_STRING) { - size_t len; - char * data = RedisModule_StringDMA(key, &len, REDISMODULE_READ); - RedisModule_ReplyWithArray(ctx, 2); - RedisModule_ReplyWithString(ctx, keyname); - RedisModule_ReplyWithStringBuffer(ctx, data, len); - pd->nkeys++; - } - if (was_opened) - RedisModule_CloseKey(key); -} - -int scan_strings(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) -{ - REDISMODULE_NOT_USED(argv); - REDISMODULE_NOT_USED(argc); - scan_strings_pd pd = { - .nkeys = 0, - }; - - RedisModule_ReplyWithArray(ctx, REDISMODULE_POSTPONED_LEN); - - RedisModuleScanCursor* cursor = RedisModule_ScanCursorCreate(); - while(RedisModule_Scan(ctx, cursor, scan_strings_callback, &pd)); - RedisModule_ScanCursorDestroy(cursor); - - RedisModule_ReplySetArrayLength(ctx, pd.nkeys); - return REDISMODULE_OK; -} - -typedef struct { - RedisModuleCtx *ctx; - size_t nreplies; -} scan_key_pd; - -void scan_key_callback(RedisModuleKey *key, RedisModuleString* field, RedisModuleString* value, void *privdata) { - REDISMODULE_NOT_USED(key); - scan_key_pd* pd = privdata; - RedisModule_ReplyWithArray(pd->ctx, 2); - size_t fieldCStrLen; - - // The implementation of RedisModuleString is robj with lots of encodings. - // We want to make sure the robj that passes to this callback in - // String encoded, this is why we use RedisModule_StringPtrLen and - // RedisModule_ReplyWithStringBuffer instead of directly use - // RedisModule_ReplyWithString. - const char* fieldCStr = RedisModule_StringPtrLen(field, &fieldCStrLen); - RedisModule_ReplyWithStringBuffer(pd->ctx, fieldCStr, fieldCStrLen); - if(value){ - size_t valueCStrLen; - const char* valueCStr = RedisModule_StringPtrLen(value, &valueCStrLen); - RedisModule_ReplyWithStringBuffer(pd->ctx, valueCStr, valueCStrLen); - } else { - RedisModule_ReplyWithNull(pd->ctx); - } - - pd->nreplies++; -} - -int scan_key(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) -{ - if (argc != 2) { - RedisModule_WrongArity(ctx); - return REDISMODULE_OK; - } - scan_key_pd pd = { - .ctx = ctx, - .nreplies = 0, - }; - - RedisModuleKey *key = RedisModule_OpenKey(ctx, argv[1], REDISMODULE_READ); - if (!key) { - RedisModule_ReplyWithError(ctx, "not found"); - return REDISMODULE_OK; - } - - RedisModule_ReplyWithArray(ctx, REDISMODULE_POSTPONED_ARRAY_LEN); - - RedisModuleScanCursor* cursor = RedisModule_ScanCursorCreate(); - while(RedisModule_ScanKey(key, cursor, scan_key_callback, &pd)); - RedisModule_ScanCursorDestroy(cursor); - - RedisModule_ReplySetArrayLength(ctx, pd.nreplies); - RedisModule_CloseKey(key); - return REDISMODULE_OK; -} - -int RedisModule_OnLoad(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) { - REDISMODULE_NOT_USED(argv); - REDISMODULE_NOT_USED(argc); - if (RedisModule_Init(ctx, "scan", 1, REDISMODULE_APIVER_1)== REDISMODULE_ERR) - return REDISMODULE_ERR; - - if (RedisModule_CreateCommand(ctx, "scan.scan_strings", scan_strings, "", 0, 0, 0) == REDISMODULE_ERR) - return REDISMODULE_ERR; - - if (RedisModule_CreateCommand(ctx, "scan.scan_key", scan_key, "", 0, 0, 0) == REDISMODULE_ERR) - return REDISMODULE_ERR; - - return REDISMODULE_OK; -} - - |
