gvsafronov/fbx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add into clause in select statement

Browse Source
This commit is contained in:
Kenneth Cheng 2020-05-01 17:30:50 +08:00
parent f28c53786d
commit 27bb3f197c
1 changed files with 82 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

105
src/dbx.c
View File

@ -8,6 +8,8 @@
#include "../rmutil/vector.h" #include "../rmutil/vector.h"
#include "../rmutil/test_util.h" #include "../rmutil/test_util.h"
static int rn;
/* Helper function: compiles a regex, or dies complaining. */ /* Helper function: compiles a regex, or dies complaining. */
int regexCompile(RedisModuleCtx *ctx, regex_t *r, const char *t) { int regexCompile(RedisModuleCtx *ctx, regex_t *r, const char *t) {
int status = regcomp(r, t, REG_EXTENDED | REG_NOSUB | REG_NEWLINE); int status = regcomp(r, t, REG_EXTENDED | REG_NOSUB | REG_NEWLINE);
@ -100,7 +102,7 @@ void showRecord(RedisModuleCtx *ctx, RedisModuleString *key, Vector *vSelect) {
for(size_t i = 0; i < nSelected; i++) { for(size_t i = 0; i < nSelected; i++) {
Vector_Get(vSelect, i, &field); Vector_Get(vSelect, i, &field);
// If '*' is specified in selected hash list, display all hashs then // If '*' is specified in selected hash list, display all hashes then
if (strcmp(field, "*") == 0) { if (strcmp(field, "*") == 0) {
RedisModuleCallReply *tags = RedisModule_Call(ctx, "HGETALL", "s", key); RedisModuleCallReply *tags = RedisModule_Call(ctx, "HGETALL", "s", key);
size_t tf = RedisModule_CallReplyLength(tags); size_t tf = RedisModule_CallReplyLength(tags);
@ -137,6 +139,46 @@ void showRecord(RedisModuleCtx *ctx, RedisModuleString *key, Vector *vSelect) {
RedisModule_ReplySetArrayLength(ctx, n); RedisModule_ReplySetArrayLength(ctx, n);
} }
void intoRecord(RedisModuleCtx *ctx, RedisModuleString *key, Vector *vSelect, char *intoKey) {
char* field;
size_t nSelected = Vector_Size(vSelect);
char newkey[64];
sprintf(newkey, "%s:%u-%i", intoKey, (unsigned)time(NULL), rn++);
for(size_t i = 0; i < nSelected; i++) {
Vector_Get(vSelect, i, &field);
// If '*' is specified in selected hash list, display all hashes then
if (strcmp(field, "*") == 0) {
RedisModuleCallReply *tags = RedisModule_Call(ctx, "HGETALL", "s", key);
size_t tf = RedisModule_CallReplyLength(tags);
if (tf > 0) {
for(size_t j=0; j<tf; j+=2) {
RedisModuleString *rms1 = RedisModule_CreateStringFromCallReply(RedisModule_CallReplyArrayElement(tags, j));
RedisModuleString *rms2 = RedisModule_CreateStringFromCallReply(RedisModule_CallReplyArrayElement(tags, j+1));
RedisModule_Call(ctx, "HSET", "css", newkey, rms1, rms2);
RedisModule_FreeString(ctx, rms1);
RedisModule_FreeString(ctx, rms2);
}
}
RedisModule_FreeCallReply(tags);
}
else {
// Display the hash name and content
RedisModuleCallReply *tags = RedisModule_Call(ctx, "HGET", "sc", key, field);
if (RedisModule_CallReplyLength(tags) > 0) {
RedisModuleString *rms = RedisModule_CreateStringFromCallReply(tags);
RedisModule_Call(ctx, "HSET", "ccs", newkey, field, rms);
RedisModule_FreeString(ctx, rms);
}
else
RedisModule_Call(ctx, "HSET", "ccc", newkey, field, "");
RedisModule_FreeCallReply(tags);
}
}
RedisModule_ReplyWithSimpleString(ctx, newkey);
}
/* Split the string by specified delimilator */ /* Split the string by specified delimilator */
Vector* splitStringByChar(char *s, char* d) { Vector* splitStringByChar(char *s, char* d) {
size_t cap; size_t cap;
@ -178,7 +220,7 @@ Vector* splitWhereString(char *s) {
return v; return v;
} }
int processRecords(RedisModuleCtx *ctx, RedisModuleCallReply *keys, regex_t *r, Vector *vSelect, Vector *vWhere) { int processRecords(RedisModuleCtx *ctx, RedisModuleCallReply *keys, regex_t *r, Vector *vSelect, Vector *vWhere, char *intoKey) {
size_t nKeys = RedisModule_CallReplyLength(keys); size_t nKeys = RedisModule_CallReplyLength(keys);
size_t affected = 0; size_t affected = 0;
for (size_t i = 0; i < nKeys; i++) { for (size_t i = 0; i < nKeys; i++) {
@ -187,7 +229,10 @@ int processRecords(RedisModuleCtx *ctx, RedisModuleCallReply *keys, regex_t *r,
const char *s = RedisModule_StringPtrLen(key, &l); const char *s = RedisModule_StringPtrLen(key, &l);
if (!regexec(r, s, 1, NULL, 0)) { if (!regexec(r, s, 1, NULL, 0)) {
if (vWhere == NULL || whereRecord(ctx, key, vWhere)) { if (vWhere == NULL || whereRecord(ctx, key, vWhere)) {
showRecord(ctx, key, vSelect); if (strlen(intoKey) > 0)
intoRecord(ctx, key, vSelect, intoKey);
else
showRecord(ctx, key, vSelect);
affected++; affected++;
} }
} }
@ -238,6 +283,7 @@ int SelectCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
// Table // Table
RedisModuleString *fromKeys; RedisModuleString *fromKeys;
char intoKey[32] = "";
// Process the arguments // Process the arguments
size_t plen; size_t plen;
@ -281,8 +327,10 @@ int SelectCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
} }
switch(step) { switch(step) {
case 0: case 0:
if (strcmp("from", token) == 0) if (strcmp("into", token) == 0)
step = -1; step = -1;
else if (strcmp("from", token) == 0)
step = -3;
else { else {
if (strlen(stmSelect) + strlen(token) > 512) { if (strlen(stmSelect) + strlen(token) > 512) {
RedisModule_ReplyWithError(ctx, "select arguments are too long"); RedisModule_ReplyWithError(ctx, "select arguments are too long");
@ -292,25 +340,34 @@ int SelectCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
} }
break; break;
case -1: case -1:
// parse from clause // parse into clause, assume time+rand always is new key
fromKeys = RMUtil_CreateFormattedString(ctx, token); strcpy(intoKey, token);
step = 2; step = -2;
break; break;
case 2: case -2:
if (strcmp("where", token) == 0) if (strcmp("from", token) == 0)
step = -3; step = -3;
else if (strcmp("order", token) == 0)
step = -5;
else { else {
RedisModule_ReplyWithError(ctx, "where or order statement is expected"); RedisModule_ReplyWithError(ctx, "from keyword is expected");
return REDISMODULE_ERR; return REDISMODULE_ERR;
} }
break; break;
case -3: case -3:
// parse from clause
fromKeys = RMUtil_CreateFormattedString(ctx, token);
step = 4;
break;
case 4: case 4:
if (strcmp("where", token) == 0)
step = -5;
else if (strcmp("order", token) == 0)
step = -7;
break;
case -5:
case 6:
// parse where clause // parse where clause
if (strcmp("order", token) == 0) if (strcmp("order", token) == 0)
step = -5; step = -7;
else if (strcmp("and", token) == 0) else if (strcmp("and", token) == 0)
strcat(stmWhere, "&&"); strcat(stmWhere, "&&");
else { else {
@ -321,19 +378,19 @@ int SelectCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
char *p = token; char *p = token;
while (*p++) *p = *p == 7? 32: *p; while (*p++) *p = *p == 7? 32: *p;
strcat(stmWhere, token); strcat(stmWhere, token);
step = 4; step = 6;
} }
break; break;
case -5: case -7:
if (strcmp("by", token) == 0) if (strcmp("by", token) == 0)
step = -6; step = -8;
else { else {
RedisModule_ReplyWithError(ctx, "missing 'by' after order"); RedisModule_ReplyWithError(ctx, "missing 'by' after order");
return REDISMODULE_ERR; return REDISMODULE_ERR;
} }
break; break;
case -6: case -8:
case 7: case 9:
// parse order clause // parse order clause
if (strlen(stmOrder) + strlen(token) > 512) { if (strlen(stmOrder) + strlen(token) > 512) {
RedisModule_ReplyWithError(ctx, "order arguments are too long"); RedisModule_ReplyWithError(ctx, "order arguments are too long");
@ -344,7 +401,7 @@ int SelectCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
else else
if (strcmp("asc", token) != 0) if (strcmp("asc", token) != 0)
strcat(stmOrder, token); strcat(stmOrder, token);
step = 7; step = 9;
break; break;
} }
token = strtok(NULL, " "); token = strtok(NULL, " ");
@ -372,7 +429,7 @@ int SelectCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
if (Vector_Size(vOrder) > 0) { if (Vector_Size(vOrder) > 0) {
// temporary set name // temporary set name
char setName[32]; char setName[32];
sprintf(setName, "__db_tempset_%i", rand()); sprintf(setName, "__db_tempset_%i", rn++);
char stmt[128]; char stmt[128];
RedisModuleCallReply *rep; RedisModuleCallReply *rep;
@ -390,7 +447,7 @@ int SelectCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
sprintf(stmt, "*->%s", field); sprintf(stmt, "*->%s", field);
rep = RedisModule_Call(ctx, "SORT", "cccc", setName, "by", stmt, "alpha"); rep = RedisModule_Call(ctx, "SORT", "cccc", setName, "by", stmt, "alpha");
} }
size_t n = processRecords(ctx, rep, &regex, vSelect, NULL); size_t n = processRecords(ctx, rep, &regex, vSelect, NULL, intoKey);
RedisModule_FreeCallReply(rep); RedisModule_FreeCallReply(rep);
// set number of output // set number of output
@ -416,7 +473,7 @@ int SelectCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
/* Filter by pattern matching. */ /* Filter by pattern matching. */
RedisModuleCallReply *rkeys = RedisModule_CallReplyArrayElement(rep, 1); RedisModuleCallReply *rkeys = RedisModule_CallReplyArrayElement(rep, 1);
n += processRecords(ctx, rkeys, &regex, vSelect, vWhere); n += processRecords(ctx, rkeys, &regex, vSelect, vWhere, intoKey);
RedisModule_FreeCallReply(rep); RedisModule_FreeCallReply(rep);
} while (lcursor); } while (lcursor);
@ -486,7 +543,7 @@ int InsertCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
break; break;
case -1: case -1:
// parse into clause, assume time+rand always is new key // parse into clause, assume time+rand always is new key
intoKey = RMUtil_CreateFormattedString(ctx, "%s:%u-%i", token, (unsigned)time(NULL), rand()); intoKey = RMUtil_CreateFormattedString(ctx, "%s:%u-%i", token, (unsigned)time(NULL), rn++);
step = -2; step = -2;
break; break;
case -2: case -2:
@ -737,6 +794,8 @@ int ExecCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
int RedisModule_OnLoad(RedisModuleCtx *ctx) { int RedisModule_OnLoad(RedisModuleCtx *ctx) {
rn = rand();
// Register the module // Register the module
if (RedisModule_Init(ctx, "dbx", 1, REDISMODULE_APIVER_1) == REDISMODULE_ERR) if (RedisModule_Init(ctx, "dbx", 1, REDISMODULE_APIVER_1) == REDISMODULE_ERR)
return REDISMODULE_ERR; return REDISMODULE_ERR;