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 
41 #include <stdio.h>
42 #include <stdlib.h>
43 
44 #include <toku_pthread.h>
45 #include <unistd.h>
46 #include <memory.h>
47 #include <sys/stat.h>
48 #include <db.h>
49 
50 #include "threaded_stress_test_helpers.h"
51 
52 
ptquery_op2(DB_TXN * txn,ARG arg,void * operation_extra,void * stats_extra)53 static int ptquery_op2(DB_TXN *txn, ARG arg, void* operation_extra, void *stats_extra) {
54     int db_index = *(int *)operation_extra;
55     DB* db = arg->dbp[db_index];
56     int r = ptquery_and_maybe_check_op(db, txn, arg, true);
57     increment_counter(stats_extra, PTQUERIES, 1);
58     return r;
59 }
60 
61 
62 //
63 // This test is a form of stress that does operations on a single dictionary:
64 // We create a dictionary bigger than the cachetable (around 4x greater).
65 // Then, we spawn a bunch of pthreads that do the following:
66 //  - scan dictionary forward with bulk fetch
67 //  - scan dictionary forward slowly
68 //  - scan dictionary backward with bulk fetch
69 //  - scan dictionary backward slowly
70 //  - Grow the dictionary with insertions
71 //  - do random point queries into the dictionary
72 // With the small cachetable, this should produce quite a bit of churn in reading in and evicting nodes.
73 // If the test runs to completion without crashing, we consider it a success. It also tests that snapshots
74 // work correctly by verifying that table scans sum their vals to 0.
75 //
76 // This does NOT test:
77 //  - splits and merges
78 //  - multiple DBs
79 //
80 // Variables that are interesting to tweak and run:
81 //  - small cachetable
82 //  - number of elements
83 //
84 
85 static void
stress_table(DB_ENV * env,DB ** dbp,struct cli_args * cli_args)86 stress_table(DB_ENV* env, DB** dbp, struct cli_args *cli_args) {
87     //
88     // the threads that we want:
89     //   - some threads constantly updating random values
90     //   - one thread doing table scan with bulk fetch
91     //   - one thread doing table scan without bulk fetch
92     //   - some threads doing random point queries
93     //
94 
95     if (verbose) printf("starting creation of pthreads\n");
96     const int num_threads = cli_args->num_ptquery_threads;
97     struct arg myargs[num_threads];
98     int thread_ids[num_threads];
99     for (int i = 0; i < num_threads; i++) {
100         arg_init(&myargs[i], dbp, env, cli_args);
101         thread_ids[i] = i % cli_args->num_DBs;
102         myargs[i].operation = ptquery_op2;
103         myargs[i].operation_extra = &thread_ids[i];
104         myargs[i].txn_flags |= DB_TXN_READ_ONLY;
105     }
106     run_workers(myargs, num_threads, cli_args->num_seconds, false, cli_args);
107 }
108 
109 int
test_main(int argc,char * const argv[])110 test_main(int argc, char *const argv[]) {
111     struct cli_args args = get_default_args_for_perf();
112     parse_stress_test_args(argc, argv, &args);
113     perf_test_main(&args);
114     return 0;
115 }
116