14 /* a test program for tdb - the trivial database */
21 static TDB_CONTEXT
*db
;
22 static GDBM_FILE gdbm
;
24 struct timeval tp1
,tp2
;
26 static void start_timer(void)
28 gettimeofday(&tp1
,NULL
);
31 static double end_timer(void)
33 gettimeofday(&tp2
,NULL
);
34 return((tp2
.tv_sec
- tp1
.tv_sec
) +
35 (tp2
.tv_usec
- tp1
.tv_usec
)*1.0e-6);
38 static void fatal(char *why
)
44 static void tdb_log(TDB_CONTEXT
*tdb
, int level
, const char *format
, ...)
49 vfprintf(stdout
, format
, ap
);
54 static void compare_db(void)
56 TDB_DATA d
, key
, nextkey
;
57 datum gd
, gkey
, gnextkey
;
59 key
= tdb_firstkey(db
);
61 d
= tdb_fetch(db
, key
);
63 gkey
.dsize
= key
.dsize
;
65 gd
= gdbm_fetch(gdbm
, gkey
);
67 if (!gd
.dptr
) fatal("key not in gdbm");
68 if (gd
.dsize
!= d
.dsize
) fatal("data sizes differ");
69 if (memcmp(gd
.dptr
, d
.dptr
, d
.dsize
)) {
70 fatal("data differs");
73 nextkey
= tdb_nextkey(db
, key
);
80 gkey
= gdbm_firstkey(gdbm
);
82 gd
= gdbm_fetch(gdbm
, gkey
);
84 key
.dsize
= gkey
.dsize
;
86 d
= tdb_fetch(db
, key
);
88 if (!d
.dptr
) fatal("key not in db");
89 if (d
.dsize
!= gd
.dsize
) fatal("data sizes differ");
90 if (memcmp(d
.dptr
, gd
.dptr
, gd
.dsize
)) {
91 fatal("data differs");
94 gnextkey
= gdbm_nextkey(gdbm
, gkey
);
102 static char *randbuf(int len
)
106 buf
= (char *)malloc(len
+1);
108 for (i
=0;i
<len
;i
++) {
109 buf
[i
] = 'a' + (rand() % 26);
115 static void addrec_db(void)
121 klen
= 1 + (rand() % 4);
122 dlen
= 1 + (rand() % 100);
133 if (rand() % DELETE_PROB
== 0) {
135 } else if (rand() % STORE_PROB
== 0) {
136 if (tdb_store(db
, key
, data
, TDB_REPLACE
) != 0) {
137 fatal("tdb_store failed");
140 data
= tdb_fetch(db
, key
);
141 if (data
.dptr
) free(data
.dptr
);
148 static void addrec_gdbm(void)
154 klen
= 1 + (rand() % 4);
155 dlen
= 1 + (rand() % 100);
166 if (rand() % DELETE_PROB
== 0) {
167 gdbm_delete(gdbm
, key
);
168 } else if (rand() % STORE_PROB
== 0) {
169 if (gdbm_store(gdbm
, key
, data
, GDBM_REPLACE
) != 0) {
170 fatal("gdbm_store failed");
173 data
= gdbm_fetch(gdbm
, key
);
174 if (data
.dptr
) free(data
.dptr
);
181 static int traverse_fn(TDB_CONTEXT
*tdb
, TDB_DATA key
, TDB_DATA dbuf
, void *state
)
184 printf("[%s] [%s]\n", key
.dptr
, dbuf
.dptr
);
186 tdb_delete(tdb
, key
);
190 static void merge_test(void)
196 for (i
= 0; i
< 5; i
++) {
197 sprintf(keys
[i
], "%d", i
);
204 if (tdb_store(db
, key
, data
, TDB_REPLACE
) != 0) {
205 fatal("tdb_store failed");
221 int main(int argc
, char *argv
[])
228 db
= tdb_open("test.tdb", 0, TDB_CLEAR_IF_FIRST
,
229 O_RDWR
| O_CREAT
| O_TRUNC
, 0600);
230 gdbm
= gdbm_open("test.gdbm", 512, GDBM_WRITER
|GDBM_NEWDB
|GDBM_FAST
,
234 fatal("db open failed");
237 tdb_logging_function(db
, tdb_log
);
242 for (i
=0;i
<loops
;i
++) addrec_gdbm();
243 printf("gdbm got %.2f ops/sec\n", i
/end_timer());
250 for (i
=0;i
<loops
;i
++) addrec_db();
251 printf("tdb got %.2f ops/sec\n", i
/end_timer());
255 printf("traversed %d records\n", tdb_traverse(db
, traverse_fn
, NULL
));
256 printf("traversed %d records\n", tdb_traverse(db
, traverse_fn
, NULL
));