summaryrefslogtreecommitdiff
path: root/tests/modules/scan.c
diff options
context:
space:
mode:
Diffstat (limited to 'tests/modules/scan.c')
-rw-r--r--tests/modules/scan.c62
1 files changed, 62 insertions, 0 deletions
diff --git a/tests/modules/scan.c b/tests/modules/scan.c
new file mode 100644
index 000000000..21071720a
--- /dev/null
+++ b/tests/modules/scan.c
@@ -0,0 +1,62 @@
+#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;
+}
+
+
+
+
+