forked from beanstalkd/beanstalkd
-
Notifications
You must be signed in to change notification settings - Fork 0
/
ms.c
110 lines (93 loc) · 1.88 KB
/
ms.c
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
107
108
109
110
#include "dat.h"
#include <stdint.h>
#include <string.h>
#include <stdlib.h>
void
ms_init(Ms *a, ms_event_fn oninsert, ms_event_fn onremove)
{
a->len = a->cap = a->last = 0;
a->items = NULL;
a->oninsert = oninsert;
a->onremove = onremove;
}
static int
grow(Ms *a)
{
void **nitems;
size_t ncap = a->cap << 1;
if (!ncap)
ncap = 1;
nitems = malloc(ncap * sizeof(void *));
if (!nitems)
return 0;
memcpy(nitems, a->items, a->len * sizeof(void *));
free(a->items);
a->items = nitems;
a->cap = ncap;
return 1;
}
int
ms_append(Ms *a, void *item)
{
if (a->len >= a->cap && !grow(a))
return 0;
a->items[a->len++] = item;
if (a->oninsert)
a->oninsert(a, item, a->len - 1);
return 1;
}
static int
ms_delete(Ms *a, size_t i)
{
void *item;
if (i >= a->len)
return 0;
item = a->items[i];
a->items[i] = a->items[--a->len];
/* it has already been removed now */
if (a->onremove)
a->onremove(a, item, i);
return 1;
}
void
ms_clear(Ms *a)
{
while (ms_delete(a, 0));
free(a->items);
ms_init(a, a->oninsert, a->onremove);
}
int
ms_remove(Ms *a, void *item)
{
size_t i;
for (i = 0; i < a->len; i++) {
if (a->items[i] == item)
return ms_delete(a, i);
}
return 0;
}
int
ms_contains(Ms *a, void *item)
{
size_t i;
for (i = 0; i < a->len; i++) {
if (a->items[i] == item)
return 1;
}
return 0;
}
void *
ms_take(Ms *a)
{
void *item;
if (!a->len)
return NULL;
// The result of last behaviour is that ms_take returns the oldest elements
// first, exception is a row of multiple take calls without inserts on ms
// of even number of elements. See the test.
a->last = a->last % a->len;
item = a->items[a->last];
ms_delete(a, a->last);
++a->last;
return item;
}