source/ubftest/test_nstd_mtest5.c

0001 /* mtest5.c - memory-mapped database tester/toy */
0002 /*
0003  * Copyright 2011-2017 Howard Chu, Symas Corp.
0004  * All rights reserved.
0005  *
0006  * Redistribution and use in source and binary forms, with or without
0007  * modification, are permitted only as authorized by the OpenLDAP
0008  * Public License.
0009  *
0010  * A copy of this license is available in the file LICENSE in the
0011  * top-level directory of the distribution or, alternatively, at
0012  * <http://www.OpenLDAP.org/license.html>.
0013  */
0014
0015 /* Tests for sorted duplicate DBs using cursor_put */
0016 #include <stdio.h>
0017 #include <stdlib.h>
0018 #include <string.h>
0019 #include <time.h>
0020 #include <cgreen/cgreen.h>
0021 #include <ndebug.h>
0022 #include <edbutil.h>
0023 #include "exdb.h"
0024
0025 #define E(expr) CHECK((rc = (expr)) == EDB_SUCCESS, #expr)
0026 #define RES(err, expr) ((rc = expr) == (err) || (CHECK(!rc, #expr), 0))
0027 #define CHECK(test, msg) ((test) ? (void)0 : ((void)fprintf(stderr, \
0028     "%s:%d: %s: %s\n", __FILE__, __LINE__, msg, edb_strerror(rc)), abort()))
0029
0030 Ensure(test_nstd_mtest5)
0031 {
0032     int i = 0, j = 0, rc;
0033     EDB_env *env;
0034     EDB_dbi dbi;
0035     EDB_val key, data;
0036     EDB_txn *txn;
0037     EDB_stat mst;
0038     EDB_cursor *cursor;
0039     int count;
0040     int *values;
0041     char sval[32];
0042     char kval[sizeof(int)];
0043         char errdet[PATH_MAX];
0044
0045     srand(time(NULL));
0046
0047     memset(sval, 0, sizeof(sval));
0048
0049     count = (rand()%384) + 64;
0050     values = (int *)malloc(count*sizeof(int));
0051
0052     for(i = 0;i<count;i++) {
0053         values[i] = rand()%1024;
0054     }
0055
0056         E(ndrx_mdb_unlink("./testdb", errdet, sizeof(errdet),
0057             LOG_FACILITY_UBF));
0058
0059     E(edb_env_create(&env));
0060     E(edb_env_set_mapsize(env, 10485760));
0061     E(edb_env_set_maxdbs(env, 4));
0062     E(edb_env_open(env, "./testdb", EDB_FIXEDMAP|EDB_NOSYNC, 0664));
0063
0064     E(edb_txn_begin(env, NULL, 0, &txn));
0065     E(edb_dbi_open(txn, "id2", EDB_CREATE|EDB_DUPSORT, &dbi));
0066     E(edb_cursor_open(txn, dbi, &cursor));
0067
0068     key.mv_size = sizeof(int);
0069     key.mv_data = kval;
0070     data.mv_size = sizeof(sval);
0071     data.mv_data = sval;
0072
0073     fprintf(stderr, "Adding %d values\n", count);
0074     for (i=0;i<count;i++) {
0075         if (!(i & 0x0f))
0076             sprintf(kval, "%03x", values[i]);
0077         sprintf(sval, "%03x %d foo bar", values[i], values[i]);
0078         if (RES(EDB_KEYEXIST, edb_cursor_put(cursor, &key, &data, EDB_NODUPDATA)))
0079             j++;
0080     }
0081     if (j) fprintf(stderr, "%d duplicates skipped\n", j);
0082     edb_cursor_close(cursor);
0083     E(edb_txn_commit(txn));
0084     E(edb_env_stat(env, &mst));
0085
0086     E(edb_txn_begin(env, NULL, EDB_RDONLY, &txn));
0087     E(edb_cursor_open(txn, dbi, &cursor));
0088     while ((rc = edb_cursor_get(cursor, &key, &data, EDB_NEXT)) == 0) {
0089         fprintf(stderr, "key: %p %.*s, data: %p %.*s\n",
0090             key.mv_data,  (int) key.mv_size,  (char *) key.mv_data,
0091             data.mv_data, (int) data.mv_size, (char *) data.mv_data);
0092     }
0093     CHECK(rc == EDB_NOTFOUND, "edb_cursor_get");
0094     edb_cursor_close(cursor);
0095     edb_txn_abort(txn);
0096
0097     j=0;
0098
0099     for (i= count - 1; i > -1; i-= (rand()%5)) {
0100         j++;
0101         txn=NULL;
0102         E(edb_txn_begin(env, NULL, 0, &txn));
0103         sprintf(kval, "%03x", values[i & ~0x0f]);
0104         sprintf(sval, "%03x %d foo bar", values[i], values[i]);
0105         key.mv_size = sizeof(int);
0106         key.mv_data = kval;
0107         data.mv_size = sizeof(sval);
0108         data.mv_data = sval;
0109         if (RES(EDB_NOTFOUND, edb_del(txn, dbi, &key, &data))) {
0110             j--;
0111             edb_txn_abort(txn);
0112         } else {
0113             E(edb_txn_commit(txn));
0114         }
0115     }
0116     free(values);
0117     fprintf(stderr, "Deleted %d values\n", j);
0118
0119     E(edb_env_stat(env, &mst));
0120     E(edb_txn_begin(env, NULL, EDB_RDONLY, &txn));
0121     E(edb_cursor_open(txn, dbi, &cursor));
0122     fprintf(stderr, "Cursor next\n");
0123     while ((rc = edb_cursor_get(cursor, &key, &data, EDB_NEXT)) == 0) {
0124         fprintf(stderr, "key: %.*s, data: %.*s\n",
0125             (int) key.mv_size,  (char *) key.mv_data,
0126             (int) data.mv_size, (char *) data.mv_data);
0127     }
0128     CHECK(rc == EDB_NOTFOUND, "edb_cursor_get");
0129     fprintf(stderr, "Cursor prev\n");
0130     while ((rc = edb_cursor_get(cursor, &key, &data, EDB_PREV)) == 0) {
0131         fprintf(stderr, "key: %.*s, data: %.*s\n",
0132             (int) key.mv_size,  (char *) key.mv_data,
0133             (int) data.mv_size, (char *) data.mv_data);
0134     }
0135     CHECK(rc == EDB_NOTFOUND, "edb_cursor_get");
0136     edb_cursor_close(cursor);
0137     edb_txn_abort(txn);
0138
0139     edb_dbi_close(env, dbi);
0140     edb_env_close(env);
0141     return;
0142 }
0143
0144
0145 /**
0146  * LMDB/EDB tests
0147  * @return
0148  */
0149 TestSuite *ubf_nstd_mtest5(void)
0150 {
0151     TestSuite *suite = create_test_suite();
0152
0153     add_test(suite, test_nstd_mtest5);
0154
0155     return suite;
0156 }
0157
0158