insert_and_repair.pl
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:5k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. #!/usr/bin/perl -w
  2. #
  3. # This is a test of insert and repair/check.
  4. #
  5. $opt_loop_count=100000; # Change this to make test harder/easier
  6. ##################### Standard benchmark inits ##############################
  7. use DBI;
  8. use Getopt::Long;
  9. use Benchmark;
  10. package main;
  11. $opt_skip_create=$opt_skip_in=$opt_verbose=$opt_fast_insert=
  12.   $opt_lock_tables=$opt_debug=$opt_skip_delete=$opt_fast=$opt_force=0;
  13. $opt_host=$opt_user=$opt_password=""; $opt_db="test";
  14. GetOptions("host=s","db=s","loop-count=i","skip-create","skip-in",
  15.    "skip-delete","verbose","fast-insert","lock-tables","debug","fast",
  16.    "force","user=s","password=s") || die "Aborted";
  17. $opt_verbose=$opt_debug=$opt_lock_tables=$opt_fast_insert=$opt_fast=$opt_skip_in=$opt_force=undef;  # Ignore warnings from these
  18. $firsttable  = "bench_f1";
  19. $secondtable = "bench_f2";
  20. ####  
  21. ####  Start timeing and start test
  22. ####
  23. $start_time=new Benchmark;
  24. if (!$opt_skip_create)
  25. {
  26.   $dbh = DBI->connect("DBI:mysql:$opt_db:$opt_host",
  27.       $opt_user, $opt_password,
  28.     { PrintError => 0}) || die $DBI::errstr;
  29.   $dbh->do("drop table if exists $firsttable, $secondtable");
  30.   print "Creating tables $firsttable and $secondtable in database $opt_dbn";
  31.   $dbh->do("create table $firsttable (id int(7) not null, thread tinyint not null, info varchar(32), marker char(1), primary key(id,thread))") or die $DBI::errstr;
  32.   $dbh->do("create table $secondtable (id int(7) not null, thread tinyint not null, row int(3) not null,value double, primary key(id,thread,row)) delay_key_write=1") or die $DBI::errstr;
  33.   $dbh->disconnect; $dbh=0; # Close handler
  34. }
  35. $|= 1; # Autoflush
  36. ####
  37. #### Start the tests
  38. ####
  39. insert_in_bench1() if (($pid=fork()) == 0); $work{$pid}="insert in bench1";
  40. insert_in_bench2() if (($pid=fork()) == 0); $work{$pid}="insert in bench2";
  41. repair_and_check() if (($pid=fork()) == 0); $work{$pid}="repair/check";
  42. $errors=0;
  43. while (($pid=wait()) != -1)
  44. {
  45.   $ret=$?/256;
  46.   print "thread '" . $work{$pid} . "' finished with exit code $retn";
  47.   $errors++ if ($ret != 0);
  48. }
  49. if (!$opt_skip_delete && !$errors)
  50. {
  51.   $dbh = DBI->connect("DBI:mysql:$opt_db:$opt_host",
  52.       $opt_user, $opt_password,
  53.     { PrintError => 0}) || die $DBI::errstr;
  54.   $dbh->do("drop table $firsttable,$secondtable");
  55. }
  56. print ($errors ? "Test failedn" :"Test okn");
  57. $end_time=new Benchmark;
  58. print "Total time: " .
  59.   timestr(timediff($end_time, $start_time),"noc") . "n";
  60. exit(0);
  61. #
  62. # Insert records in the two tables
  63. sub insert_in_bench1
  64. {
  65.   my ($dbh,$rows,$found,$i);
  66.   $dbh = DBI->connect("DBI:mysql:$opt_db:$opt_host",
  67.       $opt_user, $opt_password,
  68.     { PrintError => 0}) || die $DBI::errstr;
  69.   $rows=$found=0;
  70.   for ($i=0 ; $i < $opt_loop_count; $i++)
  71.   {
  72.     $sth=$dbh->do("insert into $firsttable values ($i,0,'This is entry $i','')") || die "Got error on insert: $DBI::errstrn";
  73.     $row_count=($i % 7)+1;
  74.     $rows+=1+$row_count;
  75.     for ($j=0 ; $j < $row_count; $j++)
  76.     {
  77.       $sth=$dbh->do("insert into $secondtable values ($i,0,$j,0)") || die "Got error on insert: $DBI::errstrn";
  78.     }
  79.   }
  80.   $dbh->disconnect; $dbh=0;
  81.   print "insert_in_bench1: Inserted $rows rowsn";
  82.   exit(0);
  83. }
  84. sub insert_in_bench2
  85. {
  86.   my ($dbh,$rows,$found,$i);
  87.   $dbh = DBI->connect("DBI:mysql:$opt_db:$opt_host",
  88.       $opt_user, $opt_password,
  89.     { PrintError => 0}) || die $DBI::errstr;
  90.   $rows=$found=0;
  91.   for ($i=0 ; $i < $opt_loop_count; $i++)
  92.   {
  93.     $sth=$dbh->do("insert into $firsttable values ($i,1,'This is entry $i','')") || die "Got error on insert: $DBI::errstrn";
  94.     $row_count=((7-$i) % 7)+1;
  95.     $rows+=1+$row_count;
  96.     for ($j=0 ; $j < $row_count; $j++)
  97.     {
  98.       $sth=$dbh->do("insert into $secondtable values ($i,1,$j,0)") || die "Got error on insert: $DBI::errstrn";
  99.     }
  100.   }
  101.   $dbh->disconnect; $dbh=0;
  102.   print "insert_in_bench2: Inserted $rows rowsn";
  103.   exit(0);
  104. }
  105. sub repair_and_check
  106. {
  107.   my ($dbh,$row,@row,$found1,$found2,$last_found1,$last_found2,$i,$type,
  108.       $table);
  109.   $found1=$found2=0; $last_found1=$last_found2= -1;
  110.   $dbh = DBI->connect("DBI:mysql:$opt_db:$opt_host",
  111.       $opt_user, $opt_password,
  112.     { PrintError => 0}) || die $DBI::errstr;
  113.   for ($i=0; $found1 != $last_found1 && $found2 != $last_found1 ; $i++)
  114.   {
  115.     $type=($i & 2) ? "repair" : "check";
  116.     if ($i & 1)
  117.     {
  118.       $table=$firsttable;
  119.       $last_found1=$found1;
  120.     }
  121.     else
  122.     {
  123.       $table=$secondtable;
  124.       $last_found2=$found2;
  125.     }
  126.     $sth=$dbh->prepare("$type table $table") || die "Got error on prepare: $dbh->errstrn";
  127.     $sth->execute || die $dbh->errstr;    
  128.     while (($row=$sth->fetchrow_arrayref))
  129.     {
  130.       if ($row->[3] ne "OK")
  131.       {
  132. print "Got error " . $row->[3] . " when doing $type on $tablen";
  133. exit(1);
  134.       }
  135.     }
  136.     $sth=$dbh->prepare("select count(*) from $table") || die "Got error on prepare: $dbh->errstrn";
  137.     $sth->execute || die $dbh->errstr;
  138.     @row = $sth->fetchrow_array();
  139.     if ($i & 1)
  140.     {
  141.       $found1= $row[0];
  142.     }
  143.     else
  144.     {
  145.       $found2= $row[0];
  146.     }
  147.     $sth->finish;
  148.     sleep(2);
  149.   }
  150.   $dbh->disconnect; $dbh=0;
  151.   print "check/repair: Did $i repair/checksn";
  152.   exit(0);
  153. }