1 /* -*- mode: C++; c-basic-offset: 4; indent-tabs-mode: nil -*- */
2 // vim: ft=cpp:expandtab:ts=8:sw=4:softtabstop=4:
3 #ident "$Id$"
4 /*======
5 This file is part of PerconaFT.
6
7
8 Copyright (c) 2006, 2015, Percona and/or its affiliates. All rights reserved.
9
10 PerconaFT is free software: you can redistribute it and/or modify
11 it under the terms of the GNU General Public License, version 2,
12 as published by the Free Software Foundation.
13
14 PerconaFT is distributed in the hope that it will be useful,
15 but WITHOUT ANY WARRANTY; without even the implied warranty of
16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 GNU General Public License for more details.
18
19 You should have received a copy of the GNU General Public License
20 along with PerconaFT. If not, see <http://www.gnu.org/licenses/>.
21
22 ----------------------------------------
23
24 PerconaFT is free software: you can redistribute it and/or modify
25 it under the terms of the GNU Affero General Public License, version 3,
26 as published by the Free Software Foundation.
27
28 PerconaFT is distributed in the hope that it will be useful,
29 but WITHOUT ANY WARRANTY; without even the implied warranty of
30 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
31 GNU Affero General Public License for more details.
32
33 You should have received a copy of the GNU Affero General Public License
34 along with PerconaFT. If not, see <http://www.gnu.org/licenses/>.
35 ======= */
36
37 #ident "Copyright (c) 2006, 2015, Percona and/or its affiliates. All rights reserved."
38
39 // test that aborting an update broadcast works correctly
40
41 #include "test.h"
42
43 const int envflags = DB_INIT_MPOOL|DB_CREATE|DB_THREAD |DB_INIT_LOCK|DB_INIT_LOG|DB_INIT_TXN|DB_PRIVATE;
44
45 DB_ENV *env;
46
47 const unsigned int NUM_KEYS = 100;
48
_v(const unsigned int i)49 static inline unsigned int _v(const unsigned int i) { return 10 - i; }
_e(const unsigned int i)50 static inline unsigned int _e(const unsigned int i) { return i + 4; }
_u(const unsigned int v,const unsigned int e)51 static inline unsigned int _u(const unsigned int v, const unsigned int e) { return v * v * e; }
52
update_fun(DB * UU (db),const DBT * key,const DBT * old_val,const DBT * extra,void (* set_val)(const DBT * new_val,void * set_extra),void * set_extra)53 static int update_fun(DB *UU(db),
54 const DBT *key,
55 const DBT *old_val, const DBT *extra,
56 void (*set_val)(const DBT *new_val,
57 void *set_extra),
58 void *set_extra) {
59 unsigned int *k, *ov, e, v;
60 assert(key->size == sizeof(*k));
61 CAST_FROM_VOIDP(k, key->data);
62 assert(old_val->size == sizeof(*ov));
63 CAST_FROM_VOIDP(ov, old_val->data);
64 assert(extra->size == 0);
65 e = _e(*k);
66 v = _u(*ov, e);
67
68 {
69 DBT newval;
70 set_val(dbt_init(&newval, &v, sizeof(v)), set_extra);
71 }
72
73 return 0;
74 }
75
setup(void)76 static void setup (void) {
77 toku_os_recursive_delete(TOKU_TEST_FILENAME);
78 { int chk_r = toku_os_mkdir(TOKU_TEST_FILENAME, S_IRWXU+S_IRWXG+S_IRWXO); CKERR(chk_r); }
79 { int chk_r = db_env_create(&env, 0); CKERR(chk_r); }
80 env->set_errfile(env, stderr);
81 env->set_update(env, update_fun);
82 { int chk_r = env->open(env, TOKU_TEST_FILENAME, envflags, S_IRWXU+S_IRWXG+S_IRWXO); CKERR(chk_r); }
83 }
84
cleanup(void)85 static void cleanup (void) {
86 { int chk_r = env->close(env, 0); CKERR(chk_r); }
87 }
88
do_inserts(DB_TXN * txn,DB * db)89 static int do_inserts(DB_TXN *txn, DB *db) {
90 int r = 0;
91 DBT key, val;
92 unsigned int i, v;
93 DBT *keyp = dbt_init(&key, &i, sizeof(i));
94 DBT *valp = dbt_init(&val, &v, sizeof(v));
95 for (i = 0; i < NUM_KEYS; ++i) {
96 v = _v(i);
97 r = db->put(db, txn, keyp, valp, 0); CKERR(r);
98 }
99 return r;
100 }
101
do_updates(DB_TXN * txn,DB * db,uint32_t flags)102 static int do_updates(DB_TXN *txn, DB *db, uint32_t flags) {
103 DBT extra;
104 DBT *extrap = dbt_init(&extra, NULL, 0);
105 int r = db->update_broadcast(db, txn, extrap, flags); CKERR(r);
106 return r;
107 }
108
chk_updated(const unsigned int k,const unsigned int v)109 static void chk_updated(const unsigned int k, const unsigned int v) {
110 assert(v == _u(_v(k), _e(k)));
111 }
112
chk_original(const unsigned int k,const unsigned int v)113 static void chk_original(const unsigned int k, const unsigned int v) {
114 assert(v == _v(k));
115 }
116
do_verify_results(DB_TXN * txn,DB * db,void (* check_val)(const unsigned int k,const unsigned int v))117 static int do_verify_results(DB_TXN *txn, DB *db, void (*check_val)(const unsigned int k, const unsigned int v)) {
118 int r = 0;
119 DBT key, val;
120 unsigned int i, *vp;
121 DBT *keyp = dbt_init(&key, &i, sizeof(i));
122 DBT *valp = dbt_init(&val, NULL, 0);
123 for (i = 0; i < NUM_KEYS; ++i) {
124 r = db->get(db, txn, keyp, valp, 0); CKERR(r);
125 assert(val.size == sizeof(*vp));
126 CAST_FROM_VOIDP(vp, val.data);
127 check_val(i, *vp);
128 }
129 return r;
130 }
131
run_test(bool is_resetting)132 static void run_test(bool is_resetting) {
133 DB *db;
134 uint32_t update_flags = is_resetting ? DB_IS_RESETTING_OP : 0;
135
136 IN_TXN_COMMIT(env, NULL, txn_1, 0, {
137 { int chk_r = db_create(&db, env, 0); CKERR(chk_r); }
138 { int chk_r = db->open(db, txn_1, "foo.db", NULL, DB_BTREE, DB_CREATE, 0666); CKERR(chk_r); }
139
140 { int chk_r = do_inserts(txn_1, db); CKERR(chk_r); }
141
142 IN_TXN_COMMIT(env, txn_1, txn_11, 0, {
143 { int chk_r = do_verify_results(txn_11, db, chk_original); CKERR(chk_r); }
144 });
145 });
146
147 IN_TXN_ABORT(env, NULL, txn_2, 0, {
148 { int chk_r = do_updates(txn_2, db, update_flags); CKERR(chk_r); }
149
150 IN_TXN_COMMIT(env, txn_2, txn_21, 0, {
151 { int chk_r = do_verify_results(txn_21, db, chk_updated); CKERR(chk_r); }
152 });
153 });
154
155 IN_TXN_COMMIT(env, NULL, txn_3, 0, {
156 { int chk_r = do_verify_results(txn_3, db, chk_original); CKERR(chk_r); }
157 });
158
159 IN_TXN_COMMIT(env, NULL, txn_4, 0, {
160 { int chk_r = do_updates(txn_4, db, update_flags); CKERR(chk_r); }
161
162 IN_TXN_COMMIT(env, txn_4, txn_41, 0, {
163 { int chk_r = do_verify_results(txn_41, db, chk_updated); CKERR(chk_r); }
164 });
165 });
166
167 IN_TXN_COMMIT(env, NULL, txn_5, 0, {
168 { int chk_r = do_verify_results(txn_5, db, chk_updated); CKERR(chk_r); }
169 });
170
171 { int chk_r = db->close(db, 0); CKERR(chk_r); }
172 }
173
test_main(int argc,char * const argv[])174 int test_main(int argc, char * const argv[]) {
175 parse_args(argc, argv);
176 setup();
177 run_test(true);
178 run_test(false);
179 cleanup();
180
181 return 0;
182 }
183