mirror/mariadb
1
0
Fork 0
mirror of https://github.com/MariaDB/server.git synced 2025-01-22 23:04:20 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fixup command line parsing

Browse source 
git-svn-id: file:///svn/tokudb@4473 c7de825b-a66e-492c-adef-691d508d4ae1
This commit is contained in:
Bradley C. Kuszmaul 2008-06-11 18:02:43 +00:00
parent 62fa995705
commit 8102ccf804
1 changed files with 46 additions and 42 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

88
db-benchmark-test/scanscan.c
View file

@ -9,23 +9,39 @@
#include <unistd.h> #include <unistd.h>
const char *pname; const char *pname;
int verify_lwc=0, lwc=0, hwc=1, prelock=0, prelockflag=0; enum run_mode { RUN_HWC, RUN_LWC, RUN_VERIFY } run_mode = RUN_HWC;
u_int32_t lock_flag = DB_PRELOCKED; int do_txns=1, prelock=0, prelockflag=0;
u_int32_t lock_flag = 0;
void parse_args (int argc, const char *argv[]) { void parse_args (int argc, const char *argv[]) {
pname=argv[0]; pname=argv[0];
argc--; argc--;
argv++; argv++;
int specified_run_mode=0;
while (argc>0) { while (argc>0) {
if (strcmp(*argv,"--verify-lwc")==0) verify_lwc=1; if (strcmp(*argv,"--verify-lwc")==0) {
else if (strcmp(*argv, "--lwc")==0) lwc=1; if (specified_run_mode && run_mode!=RUN_VERIFY) { two_modes: fprintf(stderr, "You specified two run modes\n"); exit(1); }
else if (strcmp(*argv, "--nohwc")==0) hwc=0; run_mode = RUN_VERIFY;
else if (strcmp(*argv, "--prelock")==0) prelock=1; } else if (strcmp(*argv, "--lwc")==0) {
else if (strcmp(*argv, "--prelockflag")==0) prelockflag=1; if (specified_run_mode && run_mode!=RUN_LWC) goto two_modes;
run_mode = RUN_LWC;
} else if (strcmp(*argv, "--hwc")==0) {
if (specified_run_mode && run_mode!=RUN_VERIFY) goto two_modes;
run_mode = RUN_HWC;
} else if (strcmp(*argv, "--prelock")==0) prelock=1;
else if (strcmp(*argv, "--prelockflag")==0) { prelockflag=1; lock_flag = DB_PRELOCKED; }
else if (strcmp(*argv, "--prelockwriteflag")==0) { prelockflag=1; lock_flag = DB_PRELOCKED_WRITE; } else if (strcmp(*argv, "--prelockwriteflag")==0) { prelockflag=1; lock_flag = DB_PRELOCKED_WRITE; }
else if (strcmp(*argv, "--nox")==0) { do_txns=0; }
else { else {
printf("Usage:\n%s [--verify-lwc] [--lwc] [--nohwc] [--prelock] [--prelockflag] [--prelockwriteflag]\n", pname); fprintf(stderr, "Usage:\n%s [--verify-lwc | --lwc | --nohwc] [--prelock] [--prelockflag] [--prelockwriteflag]\n", pname);
fprintf(stderr, " --hwc run heavy weight cursors (this is the default)\n");
fprintf(stderr, " --verify-lwc means to run the light weight cursor and the heavyweight cursor to verify that they get the same answer.\n");
fprintf(stderr, " --lwc run light weight cursors instead of heavy weight cursors\n");
fprintf(stderr, " --prelock acquire a read lock on the entire table before running\n");
fprintf(stderr, " --prelockflag pass DB_PRELOCKED to the the cursor get operation whenever the locks have been acquired\n");
fprintf(stderr, " --prelockwriteflag pass DB_PRELOCKED_WRITE to the cursor get operation\n");
fprintf(stderr, " --nox no transactions\n");
exit(1); exit(1);
} }
argc--; argc--;
@ -41,24 +57,20 @@ DB_TXN *tid=0;
#define STRINGIFY2(s) #s #define STRINGIFY2(s) #s
#define STRINGIFY(s) STRINGIFY2(s) #define STRINGIFY(s) STRINGIFY2(s)
const char *dbdir = "./bench." STRINGIFY(DIRSUF) "/"; /* DIRSUF is passed in as a -D argument to the compiler. */; const char *dbdir = "./bench." STRINGIFY(DIRSUF) "/"; /* DIRSUF is passed in as a -D argument to the compiler. */;
#define TXNS int env_open_flags_yesx = DB_CREATE|DB_PRIVATE|DB_INIT_MPOOL|DB_INIT_TXN|DB_INIT_LOG|DB_INIT_LOCK;
#ifdef TXNS int env_open_flags_nox = DB_CREATE|DB_PRIVATE|DB_INIT_MPOOL;
int env_open_flags = DB_CREATE|DB_PRIVATE|DB_INIT_MPOOL|DB_INIT_TXN|DB_INIT_LOG|DB_INIT_LOCK;
#else
int env_open_flags = DB_CREATE|DB_PRIVATE|DB_INIT_MPOOL;
#endif
char *dbfilename = "bench.db"; char *dbfilename = "bench.db";
void setup (void) { void setup (void) {
int r; int r;
r = db_env_create(&env, 0); assert(r==0); r = db_env_create(&env, 0); assert(r==0);
r = env->set_cachesize(env, 0, 127*1024*1024, 1); assert(r==0); r = env->set_cachesize(env, 0, 127*1024*1024, 1); assert(r==0);
r = env->open(env, dbdir, env_open_flags, 0644); assert(r==0); r = env->open(env, dbdir, do_txns? env_open_flags_yesx : env_open_flags_nox, 0644); assert(r==0);
r = db_create(&db, env, 0); assert(r==0); r = db_create(&db, env, 0); assert(r==0);
#ifdef TXNS if (do_txns) {
r = env->txn_begin(env, 0, &tid, 0); assert(r==0); r = env->txn_begin(env, 0, &tid, 0); assert(r==0);
#endif }
r = db->open(db, tid, dbfilename, NULL, DB_BTREE, 0, 0644); assert(r==0); r = db->open(db, tid, dbfilename, NULL, DB_BTREE, 0, 0644); assert(r==0);
if (prelock) { if (prelock) {
r = db->pre_acquire_read_lock(db, r = db->pre_acquire_read_lock(db,
tid, tid,
@ -71,9 +83,9 @@ void setup (void) {
void shutdown (void) { void shutdown (void) {
int r; int r;
r = db->close(db, 0); assert(r==0); r = db->close(db, 0); assert(r==0);
#ifdef TXNS if (do_txns) {
r = tid->commit(tid, 0); assert(r==0); r = tid->commit(tid, 0); assert(r==0);
#endif }
r = env->close(env, 0); assert(r==0); r = env->close(env, 0); assert(r==0);
{ {
extern unsigned long toku_get_maxrss(void); extern unsigned long toku_get_maxrss(void);
@ -88,7 +100,7 @@ double gettime (void) {
return tv.tv_sec + 1e-6*tv.tv_usec; return tv.tv_sec + 1e-6*tv.tv_usec;
} }
void scanscan (void) { void scanscan_hwc (void) {
int r; int r;
int counter=0; int counter=0;
for (counter=0; counter<2; counter++) { for (counter=0; counter<2; counter++) {
@ -202,23 +214,15 @@ int main (int argc, const char *argv[]) {
parse_args(argc,argv); parse_args(argc,argv);
if (hwc) { setup();
setup(); switch (run_mode) {
scanscan(); case RUN_HWC: scanscan_hwc(); goto ok;
shutdown(); case RUN_LWC: scanscan_lwc(); goto ok;
} case RUN_VERIFY: scanscan_verify(); goto ok;
if (lwc) {
setup();
scanscan_lwc();
shutdown();
}
if (verify_lwc) {
setup();
scanscan_verify();
shutdown();
} }
assert(0);
ok:
shutdown();
return 0; return 0;
} }