diff options
Diffstat (limited to 'storage/tokudb/PerconaFT/src/tests/recover-test2.cc')
| -rw-r--r-- | storage/tokudb/PerconaFT/src/tests/recover-test2.cc | 180 |
1 files changed, 180 insertions, 0 deletions
diff --git a/storage/tokudb/PerconaFT/src/tests/recover-test2.cc b/storage/tokudb/PerconaFT/src/tests/recover-test2.cc new file mode 100644 index 00000000000..167c8f5888f --- /dev/null +++ b/storage/tokudb/PerconaFT/src/tests/recover-test2.cc @@ -0,0 +1,180 @@ +/* -*- mode: C++; c-basic-offset: 4; indent-tabs-mode: nil -*- */ +// vim: ft=cpp:expandtab:ts=8:sw=4:softtabstop=4: +#ident "$Id$" +/*====== +This file is part of PerconaFT. + + +Copyright (c) 2006, 2015, Percona and/or its affiliates. All rights reserved. + + PerconaFT is free software: you can redistribute it and/or modify + it under the terms of the GNU General Public License, version 2, + as published by the Free Software Foundation. + + PerconaFT is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with PerconaFT. If not, see <http://www.gnu.org/licenses/>. + +---------------------------------------- + + PerconaFT is free software: you can redistribute it and/or modify + it under the terms of the GNU Affero General Public License, version 3, + as published by the Free Software Foundation. + + PerconaFT is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU Affero General Public License for more details. + + You should have received a copy of the GNU Affero General Public License + along with PerconaFT. If not, see <http://www.gnu.org/licenses/>. +======= */ + +#ident "Copyright (c) 2006, 2015, Percona and/or its affiliates. All rights reserved." + +// verify that the table lock log entry is handled + +#include <sys/stat.h> +#include "test.h" + + +const char *namea="a.db"; + +DB_ENV *env; +DB_TXN *tid; +DB *db; +DBT key,data; +int i; +enum {N=1000}; +char *keys[N]; +char *vals[N]; + +static void +do_x1_shutdown (void) { + int r; + toku_os_recursive_delete(TOKU_TEST_FILENAME); + r=toku_os_mkdir(TOKU_TEST_FILENAME, S_IRWXU+S_IRWXG+S_IRWXO); assert(r==0); + + r=db_env_create(&env, 0); assert(r==0); + env->set_errfile(env, stderr); + r=env->open(env, TOKU_TEST_FILENAME, DB_INIT_LOCK|DB_INIT_LOG|DB_INIT_MPOOL|DB_INIT_TXN|DB_CREATE|DB_PRIVATE|DB_THREAD, S_IRWXU+S_IRWXG+S_IRWXO); CKERR(r); + { + DB_TXN *oldest; + r=env->txn_begin(env, 0, &oldest, 0); + CKERR(r); + } + + r=db_create(&db, env, 0); CKERR(r); + r=env->txn_begin(env, 0, &tid, 0); assert(r==0); + r=db->open(db, tid, "foo.db", 0, DB_BTREE, DB_CREATE, S_IRWXU+S_IRWXG+S_IRWXO); CKERR(r); + r=tid->commit(tid, 0); assert(r==0); + + r=env->txn_begin(env, 0, &tid, 0); assert(r==0); + for (i=0; i<N; i++) { + r=db->put(db, tid, dbt_init(&key, keys[i], strlen(keys[i])+1), dbt_init(&data, vals[i], strlen(vals[i])+1), 0); assert(r==0); + } + r=tid->commit(tid, 0); assert(r==0); + + r=db->close(db, 0); assert(r==0); + + //printf("shutdown\n"); + toku_hard_crash_on_purpose(); +} + +static void +do_x1_recover (bool UU(did_commit)) { + int r; + char glob[TOKU_PATH_MAX+1]; + toku_os_recursive_delete(toku_path_join(glob, 2, TOKU_TEST_FILENAME, "*.tokudb")); + + r=db_env_create(&env, 0); assert(r==0); + env->set_errfile(env, stderr); + r=env->open(env, TOKU_TEST_FILENAME, DB_INIT_LOCK|DB_INIT_LOG|DB_INIT_MPOOL|DB_INIT_TXN|DB_CREATE|DB_PRIVATE|DB_THREAD|DB_RECOVER, S_IRWXU+S_IRWXG+S_IRWXO); CKERR(r); + r=env->txn_begin(env, 0, &tid, 0); assert(r==0); + r=db_create(&db, env, 0); CKERR(r); + r=db->open(db, tid, "foo.db", 0, DB_BTREE, 0, S_IRWXU+S_IRWXG+S_IRWXO); CKERR(r); + for (i=0; i<N; i++) { + r=db->get(db, tid, dbt_init(&key, keys[i], 1+strlen(keys[i])), dbt_init_malloc(&data), 0); assert(r==0); + assert(strcmp((char*)data.data, vals[i])==0); + toku_free(data.data); + data.data=0; + } + r=tid->commit(tid, 0); assert(r==0); + toku_free(data.data); + r=db->close(db, 0); CKERR(r); + r=env->close(env, 0); CKERR(r); +} + +bool do_commit=false, do_recover_committed=false; + +static void +x1_parse_args (int argc, char * const argv[]) { + int resultcode; + char *cmd = argv[0]; + argc--; argv++; + while (argc>0) { + if (strcmp(argv[0], "-v") == 0) { + verbose++; + } else if (strcmp(argv[0],"-q")==0) { + verbose--; + if (verbose<0) verbose=0; + } else if (strcmp(argv[0], "--test") == 0) { + do_commit=true; + } else if (strcmp(argv[0], "--recover") == 0) { + do_recover_committed=true; + } else if (strcmp(argv[0], "-h")==0) { + resultcode=0; + do_usage: + fprintf(stderr, "Usage:\n%s [-v|-q]* [-h] {--commit | --abort | --explicit-abort | --recover-committed | --recover-aborted } \n", cmd); + exit(resultcode); + } else { + fprintf(stderr, "Unknown arg: %s\n", argv[0]); + resultcode=1; + goto do_usage; + } + argc--; + argv++; + } + { + int n_specified=0; + if (do_commit) n_specified++; + if (do_recover_committed) n_specified++; + if (n_specified>1) { + printf("Specify only one of --commit or --abort or --recover-committed or --recover-aborted\n"); + resultcode=1; + goto do_usage; + } + } +} + +int +test_main (int argc, char * const argv[]) +{ + srandom(0xDEADBEEF); + for (i=0; i<N; i++) { + char ks[100]; snprintf(ks, sizeof(ks), "k%09ld.%d", random(), i); + char vs[1000]; snprintf(vs, sizeof(vs), "v%d.%0*d", i, (int)(sizeof(vs)-100), i); + keys[i]=toku_strdup(ks); + vals[i]=toku_strdup(vs); + } + x1_parse_args(argc, argv); + if (do_commit) { + do_x1_shutdown(); + } else if (do_recover_committed) { + do_x1_recover(true); + } + for (i=0; i<N; i++) { + toku_free(keys[i]); + toku_free(vals[i]); + } +#if 0 + else { + do_test(); + } +#endif + return 0; +} |