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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
#include "ntdb.h"
#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <unistd.h>
#include <ccan/err/err.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
static void logfn(struct ntdb_context *ntdb,
enum ntdb_log_level level,
enum NTDB_ERROR ecode,
const char *message,
void *data)
{
fprintf(stderr, "ntdb:%s:%s:%s\n",
ntdb_name(ntdb), ntdb_errorstr(ecode), message);
}
int main(int argc, char *argv[])
{
unsigned int i, j, users, groups;
NTDB_DATA idxkey, idxdata;
NTDB_DATA k, d, gk;
char cmd[100];
struct ntdb_context *ntdb;
enum NTDB_ERROR ecode;
union ntdb_attribute log;
if (argc != 3) {
printf("Usage: growtdb-bench <users> <groups>\n");
exit(1);
}
users = atoi(argv[1]);
groups = atoi(argv[2]);
sprintf(cmd, "cat /proc/%i/statm", getpid());
log.base.attr = NTDB_ATTRIBUTE_LOG;
log.base.next = NULL;
log.log.fn = logfn;
ntdb = ntdb_open("/tmp/growtdb.ntdb", NTDB_DEFAULT,
O_RDWR|O_CREAT|O_TRUNC, 0600, &log);
idxkey.dptr = (unsigned char *)"User index";
idxkey.dsize = strlen("User index");
idxdata.dsize = 51;
idxdata.dptr = calloc(idxdata.dsize, 1);
if (idxdata.dptr == NULL) {
fprintf(stderr, "Unable to allocate memory for idxdata.dptr\n");
return -1;
}
/* Create users. */
k.dsize = 48;
k.dptr = calloc(k.dsize, 1);
if (k.dptr == NULL) {
fprintf(stderr, "Unable to allocate memory for k.dptr\n");
return -1;
}
d.dsize = 64;
d.dptr = calloc(d.dsize, 1);
if (d.dptr == NULL) {
fprintf(stderr, "Unable to allocate memory for d.dptr\n");
return -1;
}
ntdb_transaction_start(ntdb);
for (i = 0; i < users; i++) {
memcpy(k.dptr, &i, sizeof(i));
ecode = ntdb_store(ntdb, k, d, NTDB_INSERT);
if (ecode != NTDB_SUCCESS)
errx(1, "ntdb insert failed: %s", ntdb_errorstr(ecode));
/* This simulates a growing index record. */
ecode = ntdb_append(ntdb, idxkey, idxdata);
if (ecode != NTDB_SUCCESS)
errx(1, "ntdb append failed: %s", ntdb_errorstr(ecode));
}
if ((ecode = ntdb_transaction_commit(ntdb)) != 0)
errx(1, "ntdb commit1 failed: %s", ntdb_errorstr(ecode));
if ((ecode = ntdb_check(ntdb, NULL, NULL)) != 0)
errx(1, "ntdb_check failed after initial insert!");
system(cmd);
/* Now put them all in groups: add 32 bytes to each record for
* a group. */
gk.dsize = 48;
gk.dptr = calloc(k.dsize, 1);
if (gk.dptr == NULL) {
fprintf(stderr, "Unable to allocate memory for gk.dptr\n");
return -1;
}
gk.dptr[gk.dsize-1] = 1;
d.dsize = 32;
for (i = 0; i < groups; i++) {
ntdb_transaction_start(ntdb);
/* Create the "group". */
memcpy(gk.dptr, &i, sizeof(i));
ecode = ntdb_store(ntdb, gk, d, NTDB_INSERT);
if (ecode != NTDB_SUCCESS)
errx(1, "ntdb insert failed: %s", ntdb_errorstr(ecode));
/* Now populate it. */
for (j = 0; j < users; j++) {
/* Append to the user. */
memcpy(k.dptr, &j, sizeof(j));
if ((ecode = ntdb_append(ntdb, k, d)) != 0)
errx(1, "ntdb append failed: %s",
ntdb_errorstr(ecode));
/* Append to the group. */
if ((ecode = ntdb_append(ntdb, gk, d)) != 0)
errx(1, "ntdb append failed: %s",
ntdb_errorstr(ecode));
}
if ((ecode = ntdb_transaction_commit(ntdb)) != 0)
errx(1, "ntdb commit2 failed: %s", ntdb_errorstr(ecode));
if ((ecode = ntdb_check(ntdb, NULL, NULL)) != 0)
errx(1, "ntdb_check failed after iteration %i!", i);
system(cmd);
}
return 0;
}
|
