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 #include "test.h"
40 #include "cachetable-test.h"
41
42 bool clone_called;
43 bool check_flush;
44 bool flush_expected;
45 bool flush_called;
46
47 static void
clone_callback(void * UU (value_data),void ** cloned_value_data,long * clone_size,PAIR_ATTR * new_attr,bool UU (for_checkpoint),void * UU (write_extraargs))48 clone_callback(void* UU(value_data), void** cloned_value_data, long* clone_size, PAIR_ATTR* new_attr, bool UU(for_checkpoint), void* UU(write_extraargs))
49 {
50 *cloned_value_data = (void *)1;
51 new_attr->is_valid = false;
52 clone_called = true;
53 *clone_size = 8;
54 }
55
56 static void
flush(CACHEFILE f,int UU (fd),CACHEKEY k,void * v,void ** UU (dd),void * e,PAIR_ATTR s,PAIR_ATTR * new_size,bool w,bool keep,bool c,bool UU (is_clone))57 flush (
58 CACHEFILE f __attribute__((__unused__)),
59 int UU(fd),
60 CACHEKEY k __attribute__((__unused__)),
61 void *v __attribute__((__unused__)),
62 void** UU(dd),
63 void *e __attribute__((__unused__)),
64 PAIR_ATTR s __attribute__((__unused__)),
65 PAIR_ATTR* new_size __attribute__((__unused__)),
66 bool w __attribute__((__unused__)),
67 bool keep __attribute__((__unused__)),
68 bool c __attribute__((__unused__)),
69 bool UU(is_clone)
70 )
71 {
72 if (w && check_flush) {
73 assert(flush_expected);
74 flush_called = true;
75 }
76 }
77
78 //
79 // test the following things for simple cloning:
80 // - verifies that after the checkpoint ends, the PAIR is properly
81 // dirty or clean based on the second unpin
82 //
83 static void
test_clean(enum cachetable_dirty dirty,bool cloneable)84 test_clean (enum cachetable_dirty dirty, bool cloneable) {
85 const int test_limit = 200;
86 int r;
87 CACHETABLE ct;
88 toku_cachetable_create(&ct, test_limit, ZERO_LSN, nullptr);
89 const char *fname1 = TOKU_TEST_FILENAME;
90 unlink(fname1);
91 CACHEFILE f1;
92 r = toku_cachetable_openf(&f1, ct, fname1, O_RDWR|O_CREAT, S_IRWXU|S_IRWXG|S_IRWXO); assert(r == 0);
93 create_dummy_functions(f1);
94 check_flush = false;
95
96 void* v1;
97 CACHETABLE_WRITE_CALLBACK wc = def_write_callback(NULL);
98 wc.clone_callback = cloneable ? clone_callback : NULL;
99 wc.flush_callback = flush;
100 r = toku_cachetable_get_and_pin(f1, make_blocknum(1), 1, &v1, wc, def_fetch, def_pf_req_callback, def_pf_callback, true, NULL);
101 r = toku_test_cachetable_unpin(f1, make_blocknum(1), 1, CACHETABLE_DIRTY, make_pair_attr(8));
102
103 // begin checkpoint, since pair is clean, we should not
104 // have the clone called
105 CHECKPOINTER cp = toku_cachetable_get_checkpointer(ct);
106 toku_cachetable_begin_checkpoint(cp, NULL);
107 assert_zero(r);
108 r = toku_cachetable_get_and_pin(f1, make_blocknum(1), 1, &v1, wc, def_fetch, def_pf_req_callback, def_pf_callback, true, NULL);
109
110 // at this point, there should be no more dirty writes
111 r = toku_test_cachetable_unpin(f1, make_blocknum(1), 1, dirty, make_pair_attr(8));
112 usleep(2*1024*1024);
113 toku_cachetable_end_checkpoint(
114 cp,
115 NULL,
116 NULL,
117 NULL
118 );
119
120 check_flush = true;
121 flush_expected = (dirty == CACHETABLE_DIRTY) ? true : false;
122 flush_called = false;
123
124 toku_cachetable_verify(ct);
125 toku_cachefile_close(&f1, false, ZERO_LSN);
126 toku_cachetable_close(&ct);
127 if (flush_expected) assert(flush_called);
128 }
129
130 int
test_main(int argc,const char * argv[])131 test_main(int argc, const char *argv[]) {
132 default_parse_args(argc, argv);
133 test_clean(CACHETABLE_CLEAN, true);
134 test_clean(CACHETABLE_DIRTY, true);
135 test_clean(CACHETABLE_CLEAN, false);
136 test_clean(CACHETABLE_DIRTY, false);
137 return 0;
138 }
139