/* -*- mode: C++; c-basic-offset: 4; indent-tabs-mode: nil -*- */ // vim: ft=cpp:expandtab:ts=8:sw=4:softtabstop=4: #ident "$Id$" #ident "Copyright (c) 2007-2012 Tokutek Inc. All rights reserved." #include "includes.h" #include "test.h" bool foo; // // This test verifies that get_and_pin_nonblocking works and returns DB_TRYAGAIN when the PAIR is being used. // static void flush (CACHEFILE f __attribute__((__unused__)), int UU(fd), CACHEKEY k __attribute__((__unused__)), void *v __attribute__((__unused__)), void** UU(dd), 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) ) { /* Do nothing */ if (verbose) { printf("FLUSH: %d\n", (int)k.b); } // this should not be flushed until the bottom of the test, which // verifies that this is called if we have it pending a checkpoint if (w) { assert(c); assert(keep); } //usleep (5*1024*1024); } static bool true_def_pf_req_callback(void* UU(ftnode_pv), void* UU(read_extraargs)) { return true; } static int true_def_pf_callback(void* UU(ftnode_pv), void* UU(dd), void* UU(read_extraargs), int UU(fd), PAIR_ATTR* sizep) { *sizep = make_pair_attr(8); return 0; } static void kibbutz_work(void *fe_v) { CACHEFILE CAST_FROM_VOIDP(f1, fe_v); sleep(2); foo = true; int r = toku_test_cachetable_unpin(f1, make_blocknum(1), 1, CACHETABLE_CLEAN, make_pair_attr(8)); assert(r==0); remove_background_job_from_cf(f1); } static void run_test (void) { const int test_limit = 12; int r; CACHETABLE ct; r = toku_create_cachetable(&ct, test_limit, ZERO_LSN, NULL_LOGGER); assert(r == 0); char fname1[] = __SRCFILE__ "test1.dat"; unlink(fname1); CACHEFILE f1; r = toku_cachetable_openf(&f1, ct, fname1, 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.flush_callback = flush; // // test that if we are getting a PAIR for the first time that TOKUDB_TRY_AGAIN is returned // because the PAIR was not in the cachetable. // r = toku_cachetable_get_and_pin_nonblocking(f1, make_blocknum(1), 1, &v1, &s1, wc, def_fetch, def_pf_req_callback, def_pf_callback, true, NULL, NULL); assert(r==TOKUDB_TRY_AGAIN); // now it should succeed r = toku_cachetable_get_and_pin_nonblocking(f1, make_blocknum(1), 1, &v1, &s1, wc, def_fetch, def_pf_req_callback, def_pf_callback, true, NULL, NULL); assert(r==0); foo = false; cachefile_kibbutz_enq(f1, kibbutz_work, f1); // because node is in use, should return TOKUDB_TRY_AGAIN r = toku_cachetable_get_and_pin_nonblocking(f1, make_blocknum(1), 1, &v1, &s1, wc, def_fetch, def_pf_req_callback, def_pf_callback, true, NULL, NULL); assert(r==TOKUDB_TRY_AGAIN); 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(foo); r = toku_test_cachetable_unpin(f1, make_blocknum(1), 1, CACHETABLE_CLEAN, make_pair_attr(8)); assert(r==0); // now make sure we get TOKUDB_TRY_AGAIN when a partial fetch is involved // first make sure value is there r = toku_cachetable_get_and_pin_nonblocking(f1, make_blocknum(1), 1, &v1, &s1, wc, def_fetch, def_pf_req_callback, def_pf_callback, true, NULL, NULL); assert(r==0); r = toku_test_cachetable_unpin(f1, make_blocknum(1), 1, CACHETABLE_CLEAN, make_pair_attr(8)); assert(r==0); // now make sure that we get TOKUDB_TRY_AGAIN for the partial fetch r = toku_cachetable_get_and_pin_nonblocking(f1, make_blocknum(1), 1, &v1, &s1, wc, def_fetch, true_def_pf_req_callback, true_def_pf_callback, true, NULL, NULL); assert(r==TOKUDB_TRY_AGAIN); // // now test that if there is a checkpoint pending, // first pin and unpin with dirty // r = toku_cachetable_get_and_pin_nonblocking(f1, make_blocknum(1), 1, &v1, &s1, wc, def_fetch, def_pf_req_callback, def_pf_callback, true, NULL, NULL); assert(r==0); r = toku_test_cachetable_unpin(f1, make_blocknum(1), 1, CACHETABLE_DIRTY, make_pair_attr(8)); assert(r==0); // this should mark the PAIR as pending CHECKPOINTER cp = toku_cachetable_get_checkpointer(ct); r = toku_cachetable_begin_checkpoint(cp, NULL); assert(r == 0); r = toku_cachetable_get_and_pin_nonblocking(f1, make_blocknum(1), 1, &v1, &s1, wc, def_fetch, def_pf_req_callback, def_pf_callback, true, NULL, NULL); assert(r==TOKUDB_TRY_AGAIN); r = toku_cachetable_end_checkpoint( cp, NULL, NULL, NULL ); assert(r==0); toku_cachetable_verify(ct); r = toku_cachefile_close(&f1, 0, false, ZERO_LSN); assert(r == 0); r = toku_cachetable_close(&ct); lazy_assert_zero(r); } int test_main(int argc, const char *argv[]) { default_parse_args(argc, argv); run_test(); return 0; }