diff options
Diffstat (limited to 'storage/tokudb/PerconaFT/ft/tests/cachetable-5097.cc')
| -rw-r--r-- | storage/tokudb/PerconaFT/ft/tests/cachetable-5097.cc | 191 |
1 files changed, 191 insertions, 0 deletions
diff --git a/storage/tokudb/PerconaFT/ft/tests/cachetable-5097.cc b/storage/tokudb/PerconaFT/ft/tests/cachetable-5097.cc new file mode 100644 index 00000000000..c1b6e0a94f2 --- /dev/null +++ b/storage/tokudb/PerconaFT/ft/tests/cachetable-5097.cc @@ -0,0 +1,191 @@ +/* -*- 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." + +#include "test.h" + +CACHEFILE f1; +CACHEFILE f2; + +bool check_flush; +bool dirty_flush_called; +bool check_pe_callback; +bool pe_callback_called; +bool enable_partial_eviction; + +CACHETABLE ct; + +static int +pe_callback ( + void *ftnode_pv __attribute__((__unused__)), + PAIR_ATTR bytes_to_free __attribute__((__unused__)), + void* extraargs __attribute__((__unused__)), + void (*finalize)(PAIR_ATTR bytes_freed, void *extra), + void *finalize_extra + ) +{ + if (check_pe_callback) { + pe_callback_called = true; + } + usleep(4*1024*1024); + finalize(make_pair_attr(1), finalize_extra); + return 0; +} + +static void +flush (CACHEFILE f __attribute__((__unused__)), + int UU(fd), + CACHEKEY k __attribute__((__unused__)), + void *v __attribute__((__unused__)), + void **dd __attribute__((__unused__)), + void *e __attribute__((__unused__)), + PAIR_ATTR s __attribute__((__unused__)), + PAIR_ATTR* new_size __attribute__((__unused__)), + bool w __attribute__((__unused__)), + bool keep __attribute__((__unused__)), + bool c __attribute__((__unused__)), + bool UU(is_clone) + ) { + if (check_flush && w) { + dirty_flush_called = true; + } +} + +static void *f2_pin(void *arg) { + int r; + void* v1; + long s1; + CACHETABLE_WRITE_CALLBACK wc = def_write_callback(NULL); + // + // these booleans for pe_callback just ensure that the + // test is working as we expect it to. We expect the get_and_pin to + // cause a partial eviction of f1's PAIR, reducing its size from 8 to 1 + // and we expect that to be enough so that the unpin does not invoke a partial eviction + // This is just to ensure that the bug is being exercised + // + check_pe_callback = true; + r = toku_cachetable_get_and_pin(f2, make_blocknum(1), 1, &v1, &s1, wc, def_fetch, def_pf_req_callback, def_pf_callback, true, NULL); + assert(r == 0); + ct->ev.signal_eviction_thread(); + usleep(1*1024*1024); + if (enable_partial_eviction) + assert(pe_callback_called); + else + assert(!pe_callback_called); + pe_callback_called = false; + r = toku_test_cachetable_unpin(f2, make_blocknum(1), 1, CACHETABLE_CLEAN, make_pair_attr(8)); + check_pe_callback = false; + assert(!pe_callback_called); + assert(r == 0); + + return arg; +} + +static void +cachetable_test (void) { + const int test_limit = 12; + int r; + check_flush = false; + dirty_flush_called = false; + + toku_cachetable_create(&ct, test_limit, ZERO_LSN, nullptr); + evictor_test_helpers::disable_ev_thread(&ct->ev); // disable eviction thread + + toku_set_enable_partial_eviction(ct, enable_partial_eviction); + + toku_os_recursive_delete(TOKU_TEST_FILENAME); + r = toku_os_mkdir(TOKU_TEST_FILENAME, S_IRWXU); + assert_zero(r); + char fname1[TOKU_PATH_MAX + 1]; + unlink(toku_path_join(fname1, 2, TOKU_TEST_FILENAME, "test1.dat")); + char fname2[TOKU_PATH_MAX + 1]; + unlink(toku_path_join(fname2, 2, TOKU_TEST_FILENAME, "test2.dat")); + + r = toku_cachetable_openf(&f1, ct, fname1, O_RDWR|O_CREAT, S_IRWXU|S_IRWXG|S_IRWXO); + assert(r == 0); + r = toku_cachetable_openf(&f2, ct, fname2, O_RDWR|O_CREAT, S_IRWXU|S_IRWXG|S_IRWXO); + assert(r == 0); + + void* v1; + long s1; + CACHETABLE_WRITE_CALLBACK wc = def_write_callback(NULL); + wc.pe_callback = pe_callback; + wc.flush_callback = flush; + // pin and unpin a node 20 times, just to get clock count up + for (int i = 0; i < 20; i++) { + r = toku_cachetable_get_and_pin(f1, make_blocknum(1), 1, &v1, &s1, wc, def_fetch, def_pf_req_callback, def_pf_callback, true, NULL); + assert(r == 0); + r = toku_test_cachetable_unpin(f1, make_blocknum(1), 1, CACHETABLE_DIRTY, make_pair_attr(8)); + assert(r == 0); + } + + // at this point, we have a dirty PAIR in the cachetable associated with cachefile f1 + // launch a thread that will put another PAIR in the cachetable, and get partial eviction started + toku_pthread_t tid; + r = toku_pthread_create(&tid, NULL, f2_pin, NULL); + assert_zero(r); + + usleep(2*1024*1024); + check_flush = true; + toku_cachefile_close(&f1, false, ZERO_LSN); + if (enable_partial_eviction) + assert(dirty_flush_called); + else + assert(!dirty_flush_called); + check_flush = false; + + void *ret; + r = toku_pthread_join(tid, &ret); + assert_zero(r); + + + toku_cachetable_verify(ct); + toku_cachefile_close(&f2, false, ZERO_LSN); + toku_cachetable_close(&ct); +} + +int +test_main(int argc, const char *argv[]) { + if (argc!=2) { printf("argcount should be 2.\n"); exit(1); } + if (strcmp(argv[1], "enable_pe") == 0) { + enable_partial_eviction = true; + } else { + enable_partial_eviction = false; + } + cachetable_test(); + return 0; +} |