synchornizing changes from RediSearch. @mnunberge please next time reflect these changes on the main repo

rmutil/strings.c

@@ -2,7 +2,6 @@
 #include <sys/param.h>
 #include <ctype.h>
 #include "strings.h"
-#include "alloc.h"
 
 #include "sds.h"
 
@@ -69,13 +68,3 @@ void RMUtil_StringToUpper(RedisModuleString *s) {
     ++c;
   }
 }
-
-void RMUtil_StringConvert(RedisModuleString **rs, const char **ss, size_t n, int options) {
-  for (size_t ii = 0; ii < n; ++ii) {
-    const char *p = RedisModule_StringPtrLen(rs[ii], NULL);
-    if (options & RMUTIL_STRINGCONVERT_COPY) {
-      p = strdup(p);
-    }
-    ss[ii] = p;
-  }
-}

rmutil/strings.h

@@ -25,14 +25,4 @@ void RMUtil_StringToLower(RedisModuleString *s);
 
 /* Converts a redis string to uppercase in place without reallocating anything */
 void RMUtil_StringToUpper(RedisModuleString *s);
-
-// If set, copy the strings using strdup rather than simply storing pointers.
-#define RMUTIL_STRINGCONVERT_COPY 1
-
-/**
- * Convert one or more RedisModuleString objects into `const char*`.
- * Both rs and ss are arrays, and should be of <n> length.
- * Options may be 0 or `RMUTIL_STRINGCONVERT_COPY`
- */
-void RMUtil_StringConvert(RedisModuleString **rs, const char **ss, size_t n, int options);
 #endif

rmutil/test_periodic.c

@@ -3,7 +3,7 @@
 #include <unistd.h>
 #include "periodic.h"
 #include "assert.h"
-#include "test.h"
+#include "test_util.h"
 
 void timerCb(RedisModuleCtx *ctx, void *p) {
   int *x = p;

rmutil/test_util.h

@@ -1,67 +1,69 @@
-#ifndef __TEST_UTIL_H__
-#define __TEST_UTIL_H__
+#ifndef __TESTUTIL_H__
+#define __TESTUTIL_H__
 
-#include "util.h"
-#include <stdarg.h>
+#include <assert.h>
+#include <stdio.h>
 #include <stdlib.h>
-#include <string.h>
 
+static int numTests = 0;
+static int numAsserts = 0;
 
-#define RMUtil_Test(f) \
-                if (argc < 2 || RMUtil_ArgExists(__STRING(f), argv, argc, 1)) { \
-                    int rc = f(ctx); \
-                    if (rc != REDISMODULE_OK) { \
-                        RedisModule_ReplyWithError(ctx, "Test " __STRING(f) " FAILED"); \
-                        return REDISMODULE_ERR;\
-                    }\
-                }
+#define TESTFUNC(f)                        \
+  printf("  Testing %s\t\t", __STRING(f)); \
+  numTests++;                              \
+  fflush(stdout);                          \
+  if (f()) {                               \
+    printf(" %s FAILED!\n", __STRING(f));  \
+    exit(1);                               \
+  } else                                   \
+    printf("[PASS]\n");
 
+#define ASSERTM(expr, ...)                                                                 \
+  if (!(expr)) {                                                                           \
+    fprintf(stderr, "%s:%d: Assertion '%s' Failed: " __VA_ARGS__ "\n", __FILE__, __LINE__, \
+            __STRING(expr));                                                               \
+    return -1;                                                                             \
+  }                                                                                        \
+  numAsserts++;
 
-#define RMUtil_Assert(expr) if (!(expr)) { fprintf (stderr, "Assertion '%s' Failed\n", __STRING(expr)); return REDISMODULE_ERR; }
+#define ASSERT(expr)                                                                      \
+  if (!(expr)) {                                                                          \
+    fprintf(stderr, "%s:%d Assertion '%s' Failed\n", __FILE__, __LINE__, __STRING(expr)); \
+    return -1;                                                                            \
+  }                                                                                       \
+  numAsserts++;
 
-#define RMUtil_AssertReplyEquals(rep, cstr) RMUtil_Assert( \
-            RMUtil_StringEquals(RedisModule_CreateStringFromCallReply(rep), RedisModule_CreateString(ctx, cstr, strlen(cstr))) \
-            )
-#            
+#define ASSERT_STRING_EQ(s1, s2) ASSERT(!strcmp(s1, s2));
 
-/**
-* Create an arg list to pass to a redis command handler manually, based on the format in fmt.
-* The accepted format specifiers are:
-*   c - for null terminated c strings
-*   s - for RedisModuleString* objects
-*   l - for longs
-*
-*  Example:  RMUtil_MakeArgs(ctx, &argc, "clc", "hello", 1337, "world");
-*
-*  Returns an array of RedisModuleString pointers. The size of the array is store in argcp
-*/
-RedisModuleString **RMUtil_MakeArgs(RedisModuleCtx *ctx, int *argcp, const char *fmt, ...) {
+#define ASSERT_EQUAL(x, y, ...)                                           \
+  if (x != y) {                                                           \
+    fprintf(stderr, "%s:%d: ", __FILE__, __LINE__);                       \
+    fprintf(stderr, "%g != %g: " __VA_ARGS__ "\n", (double)x, (double)y); \
+    return -1;                                                            \
+  }                                                                       \
+  numAsserts++;
 
-    va_list ap;
-    va_start(ap, fmt);
-    RedisModuleString **argv = calloc(strlen(fmt), sizeof(RedisModuleString*));
-    int argc = 0;
-    const char *p = fmt;
-    while(*p) {
-        if (*p == 'c') {
-            char *cstr = va_arg(ap,char*);
-            argv[argc++] = RedisModule_CreateString(ctx, cstr, strlen(cstr));
-        } else if (*p == 's') {
-            argv[argc++] = va_arg(ap,void*);;
-        } else if (*p == 'l') {
-            long ll = va_arg(ap,long long);
-            argv[argc++] = RedisModule_CreateStringFromLongLong(ctx, ll);
-        } else {
-            goto fmterr;
-        }
-        p++;
+#define FAIL(fmt, ...)                                                            \
+  {                                                                               \
+    fprintf(stderr, "%s:%d: FAIL: " fmt "\n", __FILE__, __LINE__, ##__VA_ARGS__); \
+    return -1;                                                                    \
   }
-    *argcp = argc;
 
-    return argv;
-fmterr:
-    free(argv);
-    return NULL;
+#define RETURN_TEST_SUCCESS return 0;
+#define TEST_CASE(x, block) \
+  int x {                   \
+    block;                  \
+    return 0                \
   }
 
+#define PRINT_TEST_SUMMARY printf("\nTotal: %d tests and %d assertions OK\n", numTests, numAsserts);
+
+#define TEST_MAIN(body)                         \
+  int main(int argc, char **argv) {             \
+    printf("Starting Test '%s'...\n", argv[0]); \
+    body;                                       \
+    PRINT_TEST_SUMMARY;                         \
+    printf("\n--------------------\n\n");       \
+    return 0;                                   \
+  }
 #endif

rmutil/test_vector.c

@@ -1,6 +1,6 @@
 #include "vector.h"
 #include <stdio.h>
-#include "test.h"
+#include "test_util.h"
 
 int testVector() {
 

rmutil/util.c

@@ -275,3 +275,18 @@ RedisModuleString **RMUtil_ParseVarArgs(RedisModuleString **argv, int argc, int
   *nargs = n;
   return argv + 1;
 }
+
+void RMUtil_DefaultAofRewrite(RedisModuleIO *aof, RedisModuleString *key, void *value) {
+  RedisModuleCallReply *rep =
+      RedisModule_Call(RedisModule_GetContextFromIO(aof), "DUMP", "%s", key);
+  if (rep != NULL && RedisModule_CallReplyType(rep) == REDISMODULE_REPLY_STRING) {
+    size_t n;
+    const char *s = RedisModule_CallReplyStringPtr(rep, &n);
+    RedisModule_EmitAOF(aof, "RESTORE", "%sb", key, s, n);
+  } else {
+    RedisModule_Log(RedisModule_GetContextFromIO(aof), "warning", "Failed to emit AOF");
+  }
+  if (rep != NULL) {
+    RedisModule_FreeCallReply(rep);
+  }
+}

rmutil/util.h

@@ -72,6 +72,13 @@ int rmutil_vparseArgs(RedisModuleString **argv, int argc, int offset, const char
 RedisModuleString **RMUtil_ParseVarArgs(RedisModuleString **argv, int argc, int offset,
                                         const char *keyword, size_t *nargs);
 
+/**
+ * Default implementation of an AoF rewrite function that simply calls DUMP/RESTORE
+ * internally. To use this function, pass it as the .aof_rewrite value in
+ * RedisModuleTypeMethods
+ */
+void RMUtil_DefaultAofRewrite(RedisModuleIO *aof, RedisModuleString *key, void *value);
+
 // A single key/value entry in a redis info map
 typedef struct {
   const char *key;

rmutil/vector.h

@@ -2,7 +2,6 @@
 #define __VECTOR_H__
 #include <stdlib.h>
 #include <string.h>
-#include <stdarg.h>
 
 /*
 * Generic resizable vector that can be used if you just want to store stuff