15 /* a test program for tdb - the trivial database */
22 static TDB_CONTEXT *db;
23 static GDBM_FILE gdbm;
25 struct timeval tp1,tp2;
27 static void start_timer(void)
29 gettimeofday(&tp1,NULL);
32 static double end_timer(void)
34 gettimeofday(&tp2,NULL);
35 return((tp2.tv_sec - tp1.tv_sec) +
36 (tp2.tv_usec - tp1.tv_usec)*1.0e-6);
39 static void fatal(char *why)
45 static void tdb_log(TDB_CONTEXT *tdb, int level, const char *format, ...)
50 vfprintf(stdout, format, ap);
55 static void compare_db(void)
57 TDB_DATA d, key, nextkey;
58 datum gd, gkey, gnextkey;
60 key = tdb_firstkey(db);
62 d = tdb_fetch(db, key);
64 gkey.dsize = key.dsize;
66 gd = gdbm_fetch(gdbm, gkey);
68 if (!gd.dptr) fatal("key not in gdbm");
69 if (gd.dsize != d.dsize) fatal("data sizes differ");
70 if (memcmp(gd.dptr, d.dptr, d.dsize)) {
71 fatal("data differs");
74 nextkey = tdb_nextkey(db, key);
81 gkey = gdbm_firstkey(gdbm);
83 gd = gdbm_fetch(gdbm, gkey);
85 key.dsize = gkey.dsize;
87 d = tdb_fetch(db, key);
89 if (!d.dptr) fatal("key not in db");
90 if (d.dsize != gd.dsize) fatal("data sizes differ");
91 if (memcmp(d.dptr, gd.dptr, gd.dsize)) {
92 fatal("data differs");
95 gnextkey = gdbm_nextkey(gdbm, gkey);
103 static char *randbuf(int len)
107 buf = (char *)malloc(len+1);
109 for (i=0;i<len;i++) {
110 buf[i] = 'a' + (rand() % 26);
116 static void addrec_db(void)
122 klen = 1 + (rand() % 4);
123 dlen = 1 + (rand() % 100);
134 if (rand() % DELETE_PROB == 0) {
136 } else if (rand() % STORE_PROB == 0) {
137 if (tdb_store(db, key, data, TDB_REPLACE) != 0) {
138 fatal("tdb_store failed");
141 data = tdb_fetch(db, key);
142 if (data.dptr) free(data.dptr);
149 static void addrec_gdbm(void)
155 klen = 1 + (rand() % 4);
156 dlen = 1 + (rand() % 100);
167 if (rand() % DELETE_PROB == 0) {
168 gdbm_delete(gdbm, key);
169 } else if (rand() % STORE_PROB == 0) {
170 if (gdbm_store(gdbm, key, data, GDBM_REPLACE) != 0) {
171 fatal("gdbm_store failed");
174 data = gdbm_fetch(gdbm, key);
175 if (data.dptr) free(data.dptr);
182 static int traverse_fn(TDB_CONTEXT *tdb, TDB_DATA key, TDB_DATA dbuf, void *state)
185 printf("[%s] [%s]\n", key.dptr, dbuf.dptr);
187 tdb_delete(tdb, key);
191 static void merge_test(void)
197 for (i = 0; i < 5; i++) {
198 sprintf(keys[i], "%d", i);
205 if (tdb_store(db, key, data, TDB_REPLACE) != 0) {
206 fatal("tdb_store failed");
222 int main(int argc, char *argv[])
229 db = tdb_open("test.tdb", 0, TDB_CLEAR_IF_FIRST,
230 O_RDWR | O_CREAT | O_TRUNC, 0600);
231 gdbm = gdbm_open("test.gdbm", 512, GDBM_WRITER|GDBM_NEWDB|GDBM_FAST,
235 fatal("db open failed");
238 tdb_logging_function(db, tdb_log);
243 for (i=0;i<loops;i++) addrec_gdbm();
244 printf("gdbm got %.2f ops/sec\n", i/end_timer());
251 for (i=0;i<loops;i++) addrec_db();
252 printf("tdb got %.2f ops/sec\n", i/end_timer());
256 printf("traversed %d records\n", tdb_traverse(db, traverse_fn, NULL));
257 printf("traversed %d records\n", tdb_traverse(db, traverse_fn, NULL));