xref: /netbsd-src/external/bsd/openldap/dist/libraries/liblmdb/mtest6.c (revision b7b7574d3bf8eeb51a1fa3977b59142ec6434a55)
1 /*	$NetBSD: mtest6.c,v 1.1.1.1 2014/05/28 09:58:42 tron Exp $	*/
2 
3 /* mtest6.c - memory-mapped database tester/toy */
4 /*
5  * Copyright 2011 Howard Chu, Symas Corp.
6  * All rights reserved.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted only as authorized by the OpenLDAP
10  * Public License.
11  *
12  * A copy of this license is available in the file LICENSE in the
13  * top-level directory of the distribution or, alternatively, at
14  * <http://www.OpenLDAP.org/license.html>.
15  */
16 
17 /* Tests for DB splits and merges */
18 #define _XOPEN_SOURCE 500		/* srandom(), random() */
19 #include <stdio.h>
20 #include <stdlib.h>
21 #include <string.h>
22 #include <time.h>
23 #include "lmdb.h"
24 
25 char dkbuf[1024];
26 
27 int main(int argc,char * argv[])
28 {
29 	int i = 0, j = 0, rc;
30 	MDB_env *env;
31 	MDB_dbi dbi;
32 	MDB_val key, data;
33 	MDB_txn *txn;
34 	MDB_stat mst;
35 	MDB_cursor *cursor;
36 	int count;
37 	int *values;
38 	long kval;
39 	char *sval;
40 
41 	srandom(time(NULL));
42 
43 	rc = mdb_env_create(&env);
44 	rc = mdb_env_set_mapsize(env, 10485760);
45 	rc = mdb_env_set_maxdbs(env, 4);
46 	rc = mdb_env_open(env, "./testdb", MDB_FIXEDMAP|MDB_NOSYNC, 0664);
47 	rc = mdb_txn_begin(env, NULL, 0, &txn);
48 	rc = mdb_open(txn, "id2", MDB_CREATE|MDB_INTEGERKEY, &dbi);
49 	rc = mdb_cursor_open(txn, dbi, &cursor);
50 	rc = mdb_stat(txn, dbi, &mst);
51 
52 	sval = calloc(1, mst.ms_psize / 4);
53 	key.mv_size = sizeof(long);
54 	key.mv_data = &kval;
55 	data.mv_size = mst.ms_psize / 4 - 30;
56 	data.mv_data = sval;
57 
58 	printf("Adding 12 values, should yield 3 splits\n");
59 	for (i=0;i<12;i++) {
60 		kval = i*5;
61 		sprintf(sval, "%08x", kval);
62 		rc = mdb_cursor_put(cursor, &key, &data, MDB_NOOVERWRITE);
63 	}
64 	printf("Adding 12 more values, should yield 3 splits\n");
65 	for (i=0;i<12;i++) {
66 		kval = i*5+4;
67 		sprintf(sval, "%08x", kval);
68 		rc = mdb_cursor_put(cursor, &key, &data, MDB_NOOVERWRITE);
69 	}
70 	printf("Adding 12 more values, should yield 3 splits\n");
71 	for (i=0;i<12;i++) {
72 		kval = i*5+1;
73 		sprintf(sval, "%08x", kval);
74 		rc = mdb_cursor_put(cursor, &key, &data, MDB_NOOVERWRITE);
75 	}
76 	rc = mdb_cursor_get(cursor, &key, &data, MDB_FIRST);
77 
78 	do {
79 		printf("key: %p %s, data: %p %.*s\n",
80 			key.mv_data,  mdb_dkey(&key, dkbuf),
81 			data.mv_data, (int) data.mv_size, (char *) data.mv_data);
82 	} while ((rc = mdb_cursor_get(cursor, &key, &data, MDB_NEXT)) == 0);
83 	mdb_cursor_close(cursor);
84 	mdb_txn_commit(txn);
85 
86 #if 0
87 	j=0;
88 
89 	for (i= count - 1; i > -1; i-= (random()%5)) {
90 		j++;
91 		txn=NULL;
92 		rc = mdb_txn_begin(env, NULL, 0, &txn);
93 		sprintf(kval, "%03x", values[i & ~0x0f]);
94 		sprintf(sval, "%03x %d foo bar", values[i], values[i]);
95 		key.mv_size = sizeof(int);
96 		key.mv_data = kval;
97 		data.mv_size = sizeof(sval);
98 		data.mv_data = sval;
99 		rc = mdb_del(txn, dbi, &key, &data);
100 		if (rc) {
101 			j--;
102 			mdb_txn_abort(txn);
103 		} else {
104 			rc = mdb_txn_commit(txn);
105 		}
106 	}
107 	free(values);
108 	printf("Deleted %d values\n", j);
109 
110 	rc = mdb_env_stat(env, &mst);
111 	rc = mdb_txn_begin(env, NULL, 1, &txn);
112 	rc = mdb_cursor_open(txn, dbi, &cursor);
113 	printf("Cursor next\n");
114 	while ((rc = mdb_cursor_get(cursor, &key, &data, MDB_NEXT)) == 0) {
115 		printf("key: %.*s, data: %.*s\n",
116 			(int) key.mv_size,  (char *) key.mv_data,
117 			(int) data.mv_size, (char *) data.mv_data);
118 	}
119 	printf("Cursor prev\n");
120 	while ((rc = mdb_cursor_get(cursor, &key, &data, MDB_PREV)) == 0) {
121 		printf("key: %.*s, data: %.*s\n",
122 			(int) key.mv_size,  (char *) key.mv_data,
123 			(int) data.mv_size, (char *) data.mv_data);
124 	}
125 	mdb_cursor_close(cursor);
126 	mdb_close(txn, dbi);
127 
128 	mdb_txn_abort(txn);
129 #endif
130 	mdb_env_close(env);
131 
132 	return 0;
133 }
134