summaryrefslogtreecommitdiff
path: root/src/lib/batch.c
blob: b8b7629370cb34ed3f2ed803b498b5d04c0e8b38 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <threads.h>

#include "mdb.h"

static int setkey_helper(void *thr_data);
static int delkey_helper(void *thr_data);
static int *clone(int n);

tablist_t *setkey_copy;
char *setkey_str;

mtx_t setkey_mtx;
int setkeys(tablist_t **list, char *pair)
{
  int rc = 0;
  setkey_copy = calloc(list[0]->len, sizeof(tablist_t));
  memcpy(setkey_copy, *list, list[0]->len * sizeof(tablist_t));
  setkey_str = pair;

  thrd_t *thrds = calloc(list[0]->len, sizeof(thrd_t));
  if (mtx_init(&setkey_mtx, mtx_plain) != thrd_success)
    return 1;

  for (int i = 0; i < setkey_copy[0].len; ++i)
    thrd_create(&thrds[i], setkey_helper, clone(i));
  for (int i = 0; i < setkey_copy[0].len; ++i) {
    if (rc)
      thrd_join(thrds[i], NULL);
    else
      thrd_join(thrds[i], &rc);
  }
  if (!rc)
    memcpy(*list, setkey_copy, setkey_copy[0].len * sizeof(tablist_t));
  free(thrds);
  free(setkey_copy);
  return rc;
}

tablist_t *delkey_copy;
char *delkey_str;

mtx_t delkey_mtx;
int delkeys(tablist_t *list, char *key)
{
  int rc = 0;
  delkey_copy = calloc(list[0].len, sizeof(tablist_t));
  memcpy(delkey_copy, list, list[0].len * sizeof(tablist_t));
  delkey_str = key;

  thrd_t *thrds = calloc(list[0].len, sizeof(thrd_t));
  if (mtx_init(&delkey_mtx, mtx_plain) != thrd_success)
    return 1;

  for (int i = 0; i < delkey_copy[0].len; ++i)
    thrd_create(&thrds[i], delkey_helper, clone(i));
  for (int i = 0; i < delkey_copy[0].len; ++i) {
    if (rc)
      thrd_join(thrds[i], NULL);
    else
      thrd_join(thrds[i], &rc);
  }
  if (!rc)
    memcpy(list, delkey_copy, delkey_copy[0].len * sizeof(tablist_t));
  free(thrds);
  free(delkey_copy);
  return rc;
}

static int setkey_helper(void *thr_data)
{
  int rc = 0;
  
  mtx_lock(&setkey_mtx);
  int *id = (int *) thr_data;
  rc = setkey(&setkey_copy, *id, setkey_str);
  free(id);
  mtx_unlock(&setkey_mtx);
  
  return rc;
}

static int delkey_helper(void *thr_data)
{
  int rc;

  mtx_lock(&delkey_mtx);
  int *id = (int *) thr_data;
  rc = delkey(delkey_copy, *id, delkey_str);
  free(id);
  mtx_unlock(&delkey_mtx);

  return rc;
}

static int *clone(int n)
{
  int *c = calloc(1, sizeof(int));
  *c = n;
  return c;
}