summaryrefslogtreecommitdiff
path: root/src/include/engine/getkeys.c
blob: 3193e6b32ca8a62cb26ad93e120066fa59603f52 (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
104
105
106
#include <ctype.h>
#include <stdlib.h>
#include <string.h>
#include <threads.h>

#include "engine.h"
#include "utils.h"

static tabidx_t getkey(tablist_t *list, int id, char *key);

static int getkeys_helper(void *data);

struct params {
	mtx_t *mtx;
	tablist_t *list;
	tablist_t *ret;
	char **keys;
	int len;
	int lid;
	int pid;
};

static struct params *pass(mtx_t *mtx, tablist_t *list, tablist_t *ret,
						   char **keys, int len, int lid, int pid);

tablist_t *getkeys(tablist_t *list, int id, char **keys, int klen) {
	mtx_t mtx;
	if (id >= list[0].len || id < -1 ||
		mtx_init(&mtx, mtx_plain) != thrd_success)
		return NULL;
	int rc = 0;
	int len = id == -1 ? list[0].len : 1;
	tablist_t *indexes = calloc(len, sizeof(tablist_t));
	indexes[0].len = len;
	if (id >= 0)
		rc = getkeys_helper(pass(&mtx, list, indexes, keys, klen, id, 0));
	else {
		thrd_t *thrds = calloc(list[0].len, sizeof(thrd_t));
		for (int i = 0; i < list[0].len; ++i)
			thrd_create(&thrds[i], getkeys_helper,
						pass(&mtx, list, indexes, keys, klen, i, i));
		for (int i = 0; i < list[0].len; ++i) {
			if (rc)
				thrd_join(thrds[i], NULL);
			else
				thrd_join(thrds[i], &rc);
		}
		free(thrds);
	}

	mtx_destroy(&mtx);
	if (!rc)
		return indexes;
	else {
		free(indexes);
		return NULL;
	}
}

static int getkeys_helper(void *data) {
	struct params *p = (struct params *)data;
	int rc = 0;

	mtx_lock(p->mtx);
	if (p->keys == NULL) {
		for (int i = 0, j = 0; i < TABLEN; ++i)
			if (p->list[p->lid].tab[i].flag)
				p->ret[p->pid].tab[j++] = p->list[p->lid].tab[i];
	} else {
		for (int i = 0, j = 0; i < p->len; ++i) {
			tabidx_t idx = getkey(p->list, p->lid, p->keys[i]);
			if (idx.flag == 0)
				rc = 1;
			else
				p->ret[p->pid].tab[j++] = idx;
		}
	}
	mtx_unlock(p->mtx);
	free(p);
	return rc;
}

static struct params *pass(mtx_t *mtx, tablist_t *list, tablist_t *ret,
						   char **keys, int len, int lid, int pid) {
	struct params *p = calloc(1, sizeof(struct params));
	p->mtx = mtx;
	p->list = list;
	p->ret = ret;
	p->keys = keys;
	p->len = len;
	p->lid = lid;
	p->pid = pid;
	return p;
}

static tabidx_t getkey(tablist_t *list, int id, char *key) {
	int idx = hash(key);
	if (list[id].tab[idx].key == NULL)
		return (tabidx_t){.key = NULL, .flag = 0, .value = {0}};

	while (strcmp(list[id].tab[idx].key, key) && idx < TABLEN)
		idx++;
	if (idx >= TABLEN)
		return (tabidx_t){.key = NULL, .flag = 0, .value = {0}};
	return list[id].tab[idx];
}