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 // The intent of this test is to measure how fast a single thread can
53 // commit and create transactions when there exist N transactions.
54 
55 DB_TXN** txns;
56 int num_txns;
57 
commit_and_create_txn(DB_TXN * UU (txn),ARG arg,void * UU (operation_extra),void * UU (stats_extra))58 static int commit_and_create_txn(
59     DB_TXN* UU(txn),
60     ARG arg,
61     void* UU(operation_extra),
62     void* UU(stats_extra)
63     )
64 {
65     int rand_txn_id = random() % num_txns;
66     int r = txns[rand_txn_id]->commit(txns[rand_txn_id], 0);
67     CKERR(r);
68     r = arg->env->txn_begin(arg->env, 0, &txns[rand_txn_id], arg->txn_flags);
69     CKERR(r);
70     return 0;
71 }
72 
73 static void
stress_table(DB_ENV * env,DB ** dbp,struct cli_args * cli_args)74 stress_table(DB_ENV* env, DB** dbp, struct cli_args *cli_args) {
75     if (verbose) printf("starting running of stress\n");
76 
77     num_txns = cli_args->txn_size;
78     XCALLOC_N(num_txns, txns);
79     for (int i = 0; i < num_txns; i++) {
80         int r = env->txn_begin(env, 0, &txns[i], DB_TXN_SNAPSHOT);
81         CKERR(r);
82     }
83 
84     struct arg myarg;
85     arg_init(&myarg, dbp, env, cli_args);
86     myarg.operation = commit_and_create_txn;
87 
88     run_workers(&myarg, 1, cli_args->num_seconds, false, cli_args);
89 
90     for (int i = 0; i < num_txns; i++) {
91         int chk_r = txns[i]->commit(txns[i], 0);
92         CKERR(chk_r);
93     }
94     toku_free(txns);
95     num_txns = 0;
96 }
97 
98 int
test_main(int argc,char * const argv[])99 test_main(int argc, char *const argv[]) {
100     num_txns = 0;
101     txns = NULL;
102     struct cli_args args = get_default_args_for_perf();
103     parse_stress_test_args(argc, argv, &args);
104     args.single_txn = true;
105     // this test is all about transactions, make the DB small
106     args.num_elements = 1;
107     args.num_DBs= 1;
108     perf_test_main(&args);
109     return 0;
110 }
111