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