1 files changed, 148 insertions, 0 deletions
diff --git a/src/include/engine/setkeys.c b/src/include/engine/setkeys.c
new file mode 100644
index 0000000..34ef3d2
--- /dev/null
+++ b/src/include/engine/setkeys.c
@@ -0,0 +1,148 @@
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+#include <threads.h>
+
+#include "engine.h"
+#include "utils.h"
+
+static int setkey_helper(void *thr_data);
+static int setkey(tablist_t **, int, char *);
+static char **getkv(char *pair);
+
+struct params {
+  mtx_t *mtx;
+  tablist_t **copy;
+  char *pair;
+  int id;
+};
+
+static struct params *pass(mtx_t *mtx, tablist_t **copy, char *pair, int id);
+
+int setkeys(tablist_t **list, int id, char **pairs, int len)
+{
+  mtx_t mtx;
+  if (id < -1 || pairs == NULL || len <= 0)
+    return -1;
+  if (mtx_init(&mtx, mtx_plain) != thrd_success)
+    return -2;
+  int rc = 0;
+  tablist_t *copy = calloc((*list)[0].len, sizeof(tablist_t));
+  copytab(copy, *list);
+
+  for (int i = 0; i < len; ++i) {
+    char *pair = pairs[i];
+    if (id == -1) {
+      thrd_t *thrds = calloc((*list)[0].len, sizeof(thrd_t));
+      for (int i = 0; i < copy[0].len; ++i)
+        thrd_create(&thrds[i], setkey_helper, pass(&mtx, &copy, pair, i));
+      for (int i = 0; i < copy[0].len; ++i) {
+        if (rc)
+          thrd_join(thrds[i], NULL);
+        else
+          thrd_join(thrds[i], &rc);
+      }
+      free(thrds);
+    }
+    else
+     rc = setkey_helper(pass(&mtx, &copy, pair, id));
+  }
+
+  if (!rc) {
+    if (copy[0].len > (*list)[0].len)
+      *list = realloc(*list, copy[0].len * sizeof(tablist_t));
+    dellist(*list);
+    copytab(*list, copy);
+  }
+  mtx_destroy(&mtx);
+  dellist(copy);
+  free(copy);
+  return rc;
+}
+
+static int setkey_helper(void *thr_data)
+{
+  int rc;
+  struct params *p = (struct params *) thr_data;
+
+  mtx_lock(p->mtx);
+  rc = setkey(p->copy, p->id, p->pair);
+  mtx_unlock(p->mtx);
+
+  free(p);
+  return rc;
+}
+
+static int setkey(tablist_t **list, int id, char *pair)
+{
+  if (pair == NULL)
+    return 1;
+  if (id >= (*list)[0].len) {
+    *list = realloc(*list, (id + 1) * sizeof(tablist_t));
+    for (int i = (*list)[0].len; i <= id; ++i) {
+      for (int j = 0; j < TABLEN; ++j)
+        (*list)[i].tab[j] = (tabidx_t) { NULL, 0, { 0 } };
+    }
+    (*list)[0].len = id + 1;
+  }
+  char **kv = getkv(pair);
+  if (kv == NULL)
+    return 2;
+
+  int idx = hash(kv[0]);
+  while ((*list)[id].tab[idx].key != NULL &&
+    strcmp((*list)[id].tab[idx].key, kv[0]) &&
+    idx < TABLEN) idx++;
+  if (idx >= TABLEN)
+    return 2;
+  if (!(*list)[id].tab[idx].key)
+    (*list)[id].tab[idx].key = kv[0];
+  else {
+    free(kv[0]);
+    if ((*list)[id].tab[idx].flag == 3)
+      free((*list)[id].tab[idx].value.str);
+  }
+  char *end;
+  double num = strtod(kv[1], &end);
+  if (*end == '\0') {
+    (*list)[id].tab[idx].flag = 1;
+    (*list)[id].tab[idx].value.num = num;
+  } else if (!strcmp(kv[1], "true") || !strcmp(kv[1], "false")) {
+    (*list)[id].tab[idx].flag = 2;
+    (*list)[id].tab[idx].value.boolean = !strcmp(kv[1], "true");
+  } else {
+    (*list)[id].tab[idx].flag = 3;
+    (*list)[id].tab[idx].value.str = calloc(strlen(kv[1]) + 1, sizeof(char));
+    strcpy((*list)[id].tab[idx].value.str, kv[1]);
+  }
+  free(kv[1]);
+  free(kv);
+  return 0;
+}
+
+static char **getkv(char *pair)
+{
+  char **kv = calloc(2, sizeof(char *));
+  int i = 0;
+  while (pair[i] != ':' && i < strlen(pair))
+    i++;
+  if (i >= strlen(pair)) {
+    free(kv);
+    return NULL;
+  }
+  kv[0] = calloc(i + 1, sizeof(char));
+  strncpy(kv[0], pair, i);
+  kv[1] = calloc(strlen(pair) - i, sizeof(char));
+  strcpy(kv[1], pair + i + 1);
+  return kv;
+}
+
+static struct params *pass(mtx_t *mtx, tablist_t **copy, char *pair, int id)
+{
+  struct params *p = calloc(1, sizeof(struct params));
+  p->mtx =  mtx;
+  p->copy = copy;
+  p->pair = pair;
+  p->id = id;
+  return p;
+}