source/ubftest/test_nstd_mtest3.c

0001 /* mtest3.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 */
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_mtest3)
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
0067     key.mv_size = sizeof(int);
0068     key.mv_data = kval;
0069     data.mv_size = sizeof(sval);
0070     data.mv_data = sval;
0071
0072     fprintf(stderr, "Adding %d values\n", count);
0073     for (i=0;i<count;i++) {
0074         if (!(i & 0x0f))
0075             sprintf(kval, "%03x", values[i]);
0076         sprintf(sval, "%03x %d foo bar", values[i], values[i]);
0077         if (RES(EDB_KEYEXIST, edb_put(txn, dbi, &key, &data, EDB_NODUPDATA)))
0078             j++;
0079     }
0080     if (j) fprintf(stderr, "%d duplicates skipped\n", j);
0081     E(edb_txn_commit(txn));
0082     E(edb_env_stat(env, &mst));
0083
0084     E(edb_txn_begin(env, NULL, EDB_RDONLY, &txn));
0085     E(edb_cursor_open(txn, dbi, &cursor));
0086     while ((rc = edb_cursor_get(cursor, &key, &data, EDB_NEXT)) == 0) {
0087         fprintf(stderr, "key: %p %.*s, data: %p %.*s\n",
0088             key.mv_data,  (int) key.mv_size,  (char *) key.mv_data,
0089             data.mv_data, (int) data.mv_size, (char *) data.mv_data);
0090     }
0091     CHECK(rc == EDB_NOTFOUND, "edb_cursor_get");
0092     edb_cursor_close(cursor);
0093     edb_txn_abort(txn);
0094
0095     j=0;
0096
0097     for (i= count - 1; i > -1; i-= (rand()%5)) {
0098         j++;
0099         txn=NULL;
0100         E(edb_txn_begin(env, NULL, 0, &txn));
0101         sprintf(kval, "%03x", values[i & ~0x0f]);
0102         sprintf(sval, "%03x %d foo bar", values[i], values[i]);
0103         key.mv_size = sizeof(int);
0104         key.mv_data = kval;
0105         data.mv_size = sizeof(sval);
0106         data.mv_data = sval;
0107         if (RES(EDB_NOTFOUND, edb_del(txn, dbi, &key, &data))) {
0108             j--;
0109             edb_txn_abort(txn);
0110         } else {
0111             E(edb_txn_commit(txn));
0112         }
0113     }
0114     free(values);
0115     fprintf(stderr, "Deleted %d values\n", j);
0116
0117     E(edb_env_stat(env, &mst));
0118     E(edb_txn_begin(env, NULL, EDB_RDONLY, &txn));
0119     E(edb_cursor_open(txn, dbi, &cursor));
0120     fprintf(stderr, "Cursor next\n");
0121     while ((rc = edb_cursor_get(cursor, &key, &data, EDB_NEXT)) == 0) {
0122         fprintf(stderr, "key: %.*s, data: %.*s\n",
0123             (int) key.mv_size,  (char *) key.mv_data,
0124             (int) data.mv_size, (char *) data.mv_data);
0125     }
0126     CHECK(rc == EDB_NOTFOUND, "edb_cursor_get");
0127     fprintf(stderr, "Cursor prev\n");
0128     while ((rc = edb_cursor_get(cursor, &key, &data, EDB_PREV)) == 0) {
0129         fprintf(stderr, "key: %.*s, data: %.*s\n",
0130             (int) key.mv_size,  (char *) key.mv_data,
0131             (int) data.mv_size, (char *) data.mv_data);
0132     }
0133     CHECK(rc == EDB_NOTFOUND, "edb_cursor_get");
0134     edb_cursor_close(cursor);
0135     edb_txn_abort(txn);
0136
0137     edb_dbi_close(env, dbi);
0138     edb_env_close(env);
0139     return;
0140 }
0141
0142 /**
0143  * LMDB/EDB tests
0144  * @return
0145  */
0146 TestSuite *ubf_nstd_mtest3(void)
0147 {
0148     TestSuite *suite = create_test_suite();
0149
0150     add_test(suite, test_nstd_mtest3);
0151
0152     return suite;
0153 }
0154
0155