home *** CD-ROM | disk | FTP | other *** search
/ Chip 2001 January / Chip_2001-01_cd1.bin / tema / mysql / mysql-3.23.28g-win-source.exe / myisam / mi_test3.c < prev    next >
C/C++ Source or Header  |  2000-08-31  |  11KB  |  486 lines
  1. /* Copyright (C) 2000 MySQL AB & MySQL Finland AB & TCX DataKonsult AB
  2.    
  3.    This program is free software; you can redistribute it and/or modify
  4.    it under the terms of the GNU General Public License as published by
  5.    the Free Software Foundation; either version 2 of the License, or
  6.    (at your option) any later version.
  7.    
  8.    This program is distributed in the hope that it will be useful,
  9.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  10.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11.    GNU General Public License for more details.
  12.    
  13.    You should have received a copy of the GNU General Public License
  14.    along with this program; if not, write to the Free Software
  15.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  16.  
  17. /* Test av locking */
  18.  
  19. #include "myisam.h"
  20. #include <sys/types.h>
  21. #ifdef HAVE_SYS_WAIT_H
  22. # include <sys/wait.h>
  23. #endif
  24. #ifndef WEXITSTATUS
  25. # define WEXITSTATUS(stat_val) ((unsigned)(stat_val) >> 8)
  26. #endif
  27. #ifndef WIFEXITED
  28. # define WIFEXITED(stat_val) (((stat_val) & 255) == 0)
  29. #endif
  30.  
  31.  
  32. #if defined(HAVE_LRAND48)
  33. #define rnd(X) (lrand48() % X)
  34. #define rnd_init(X) srand48(X)
  35. #else
  36. #define rnd(X) (random() % X)
  37. #define rnd_init(X) srandom(X)
  38. #endif
  39.  
  40.  
  41. const char *filename= "test3.MSI";
  42. uint tests=10,forks=10,key_cacheing=0,use_log=0;
  43.  
  44. static void get_options(int argc, char *argv[]);
  45. void start_test(int id);
  46. int test_read(MI_INFO *,int),test_write(MI_INFO *,int,int),
  47.     test_update(MI_INFO *,int,int),test_rrnd(MI_INFO *,int);
  48.  
  49. struct record {
  50.   char id[8];
  51.   char nr[4];
  52.   char text[10];
  53. } record;
  54.  
  55.  
  56. int main(int argc,char **argv)
  57. {
  58.   int status,wait_ret;
  59.   uint i=0;
  60.   MI_KEYDEF keyinfo[10];
  61.   MI_COLUMNDEF recinfo[10];
  62.   MI_KEYSEG keyseg[10][2];
  63.   MY_INIT(argv[0]);
  64.   get_options(argc,argv);
  65.  
  66.   bzero((char*) keyinfo,sizeof(keyinfo));
  67.   bzero((char*) recinfo,sizeof(recinfo));
  68.   keyinfo[0].seg= &keyseg[0][0];
  69.   keyinfo[0].seg[0].start=0;
  70.   keyinfo[0].seg[0].length=8;
  71.   keyinfo[0].seg[0].type=HA_KEYTYPE_TEXT;
  72.   keyinfo[0].seg[0].flag=HA_SPACE_PACK;
  73.   keyinfo[0].keysegs=1;
  74.   keyinfo[0].flag = (uint8) HA_PACK_KEY;
  75.   keyinfo[1].seg= &keyseg[1][0];
  76.   keyinfo[1].seg[0].start=8;
  77.   keyinfo[1].seg[0].length=4;        /* Long is always 4 in myisam */
  78.   keyinfo[1].seg[0].type=HA_KEYTYPE_LONG_INT;
  79.   keyinfo[1].seg[0].flag=0;
  80.   keyinfo[1].keysegs=1;
  81.   keyinfo[1].flag =HA_NOSAME;
  82.  
  83.   recinfo[0].type=0;
  84.   recinfo[0].length=sizeof(record.id);
  85.   recinfo[1].type=0;
  86.   recinfo[1].length=sizeof(record.nr);
  87.   recinfo[2].type=0;
  88.   recinfo[2].length=sizeof(record.text);
  89.  
  90.   puts("- Creating myisam-file");
  91.   my_delete(filename,MYF(0));        /* Remove old locks under gdb */
  92.   if (mi_create(filename,2,&keyinfo[0],2,&recinfo[0],0,(MI_UNIQUEDEF*) 0,
  93.         (MI_CREATE_INFO*) 0,0))
  94.     exit(1);
  95.  
  96.   rnd_init(0);
  97.   printf("- Starting %d processes\n",forks); fflush(stdout);
  98.   for (i=0 ; i < forks; i++)
  99.   {
  100.     if (!fork())
  101.     {
  102.       start_test(i+1);
  103.       sleep(1);
  104.       return 0;
  105.     }
  106.     VOID(rnd(1));
  107.   }
  108.  
  109.   for (i=0 ; i < forks ; i++)
  110.     while ((wait_ret=wait(&status)) && wait_ret == -1);
  111.   return 0;
  112. }
  113.  
  114.  
  115. static void get_options(int argc, char **argv)
  116. {
  117.   char *pos,*progname;
  118.   DEBUGGER_OFF;
  119.  
  120.   progname= argv[0];
  121.  
  122.   while (--argc >0 && *(pos = *(++argv)) == '-' ) {
  123.     switch(*++pos) {
  124.     case 'l':
  125.       use_log=1;
  126.       break;
  127.     case 'f':
  128.       forks=atoi(++pos);
  129.       break;
  130.     case 't':
  131.       tests=atoi(++pos);
  132.       break;
  133.     case 'K':                /* Use key cacheing */
  134.       key_cacheing=1;
  135.       break;
  136.     case 'A':                /* All flags */
  137.       use_log=key_cacheing=1;
  138.       break;
  139.    case '?':
  140.     case 'I':
  141.     case 'V':
  142.       printf("%s  Ver 1.0 for %s at %s\n",progname,SYSTEM_TYPE,MACHINE_TYPE);
  143.       puts("By Monty, for your professional use\n");
  144.       puts("Test av locking with threads\n");
  145.       printf("Usage: %s [-?lKA] [-f#] [-t#]\n",progname);
  146.       exit(0);
  147.     case '#':
  148.       DEBUGGER_ON;
  149.       DBUG_PUSH (++pos);
  150.       break;
  151.     default:
  152.       printf("Illegal option: '%c'\n",*pos);
  153.       break;
  154.     }
  155.   }
  156.   return;
  157. }
  158.  
  159.  
  160. void start_test(int id)
  161. {
  162.   uint i;
  163.   int error,lock_type;
  164.   MI_ISAMINFO isam_info;
  165.   MI_INFO *file,*file1,*file2=0,*lock;
  166.  
  167.   if (use_log)
  168.     mi_log(1);
  169.   if (!(file1=mi_open(filename,O_RDWR,HA_OPEN_WAIT_IF_LOCKED)) ||
  170.       !(file2=mi_open(filename,O_RDWR,HA_OPEN_WAIT_IF_LOCKED)))
  171.   {
  172.     fprintf(stderr,"Can't open isam-file: %s\n",filename);
  173.     exit(1);
  174.   }
  175.   if (key_cacheing && rnd(2) == 0)
  176.     init_key_cache(65536L,(uint) IO_SIZE*4*10);
  177.   printf("Process %d, pid: %d\n",id,getpid()); fflush(stdout);
  178.  
  179.   for (error=i=0 ; i < tests && !error; i++)
  180.   {
  181.     file= (rnd(2) == 1) ? file1 : file2;
  182.     lock=0 ; lock_type=0;
  183.     if (rnd(10) == 0)
  184.     {
  185.       if (mi_lock_database(lock=(rnd(2) ? file1 : file2),
  186.                lock_type=(rnd(2) == 0 ? F_RDLCK : F_WRLCK)))
  187.       {
  188.     fprintf(stderr,"%2d: start: Can't lock table %d\n",id,my_errno);
  189.     error=1;
  190.     break;
  191.       }
  192.     }
  193.     switch (rnd(4)) {
  194.     case 0: error=test_read(file,id); break;
  195.     case 1: error=test_rrnd(file,id); break;
  196.     case 2: error=test_write(file,id,lock_type); break;
  197.     case 3: error=test_update(file,id,lock_type); break;
  198.     }
  199.     if (lock)
  200.       mi_lock_database(lock,F_UNLCK);
  201.   }
  202.   if (!error)
  203.   {
  204.     mi_status(file1,&isam_info,HA_STATUS_VARIABLE);
  205.     printf("%2d: End of test.  Records:  %ld  Deleted:  %ld\n",
  206.        id,isam_info.records,isam_info.deleted);
  207.     fflush(stdout);
  208.   }
  209.  
  210.   mi_close(file1);
  211.   mi_close(file2);
  212.   if (use_log)
  213.     mi_log(0);
  214.   if (error)
  215.   {
  216.     printf("%2d: Aborted\n",id); fflush(stdout);
  217.     exit(1);
  218.   }
  219. }
  220.  
  221.  
  222. int test_read(MI_INFO *file,int id)
  223. {
  224.   uint i,lock,found,next,prev;
  225.   ulong find;
  226.  
  227.   lock=0;
  228.   if (rnd(2) == 0)
  229.   {
  230.     lock=1;
  231.     if (mi_lock_database(file,F_RDLCK))
  232.     {
  233.       fprintf(stderr,"%2d: Can't lock table %d\n",id,my_errno);
  234.       return 1;
  235.     }
  236.   }
  237.  
  238.   found=next=prev=0;
  239.   for (i=0 ; i < 100 ; i++)
  240.   {
  241.     find=rnd(100000);
  242.     if (!mi_rkey(file,record.id,1,(byte*) &find,
  243.          sizeof(find),HA_READ_KEY_EXACT))
  244.       found++;
  245.     else
  246.     {
  247.       if (my_errno != HA_ERR_KEY_NOT_FOUND)
  248.       {
  249.     fprintf(stderr,"%2d: Got error %d from read in read\n",id,my_errno);
  250.     return 1;
  251.       }
  252.       else if (!mi_rnext(file,record.id,1))
  253.     next++;
  254.       else
  255.       {
  256.     if (my_errno != HA_ERR_END_OF_FILE)
  257.     {
  258.       fprintf(stderr,"%2d: Got error %d from rnext in read\n",id,my_errno);
  259.       return 1;
  260.     }
  261.     else if (!mi_rprev(file,record.id,1))
  262.       prev++;
  263.     else
  264.     {
  265.       if (my_errno != HA_ERR_END_OF_FILE)
  266.       {
  267.         fprintf(stderr,"%2d: Got error %d from rnext in read\n",
  268.             id,my_errno);
  269.         return 1;
  270.       }
  271.     }
  272.       }
  273.     }
  274.   }
  275.   if (lock)
  276.   {
  277.     if (mi_lock_database(file,F_UNLCK))
  278.     {
  279.       fprintf(stderr,"%2d: Can't unlock table\n",id);
  280.       return 1;
  281.     }
  282.   }
  283.   printf("%2d: read:   found: %5d  next: %5d   prev: %5d\n",
  284.      id,found,next,prev);
  285.   fflush(stdout);
  286.   return 0;
  287. }
  288.  
  289.  
  290. int test_rrnd(MI_INFO *file,int id)
  291. {
  292.   uint count,lock;
  293.  
  294.   lock=0;
  295.   if (rnd(2) == 0)
  296.   {
  297.     lock=1;
  298.     if (mi_lock_database(file,F_RDLCK))
  299.     {
  300.       fprintf(stderr,"%2d: Can't lock table (%d)\n",id,my_errno);
  301.       mi_close(file);
  302.       return 1;
  303.     }
  304.     if (rnd(2) == 0)
  305.       mi_extra(file,HA_EXTRA_CACHE);
  306.   }
  307.  
  308.   count=0;
  309.   if (mi_rrnd(file,record.id,0L))
  310.   {
  311.     if (my_errno == HA_ERR_END_OF_FILE)
  312.       goto end;
  313.     fprintf(stderr,"%2d: Can't read first record (%d)\n",id,my_errno);
  314.     return 1;
  315.   }
  316.   for (count=1 ; !mi_rrnd(file,record.id,HA_OFFSET_ERROR) ;count++) ;
  317.   if (my_errno != HA_ERR_END_OF_FILE)
  318.   {
  319.     fprintf(stderr,"%2d: Got error %d from rrnd\n",id,my_errno);
  320.     return 1;
  321.   }
  322.  
  323. end:
  324.   if (lock)
  325.   {
  326.     mi_extra(file,HA_EXTRA_NO_CACHE);
  327.     if (mi_lock_database(file,F_UNLCK))
  328.     {
  329.       fprintf(stderr,"%2d: Can't unlock table\n",id);
  330.       exit(0);
  331.     }
  332.   }
  333.   printf("%2d: rrnd:   %5d\n",id,count); fflush(stdout);
  334.   return 0;
  335. }
  336.  
  337.  
  338. int test_write(MI_INFO *file,int id,int lock_type)
  339. {
  340.   uint i,tries,count,lock;
  341.  
  342.   lock=0;
  343.   if (rnd(2) == 0 || lock_type == F_RDLCK)
  344.   {
  345.     lock=1;
  346.     if (mi_lock_database(file,F_WRLCK))
  347.     {
  348.       if (lock_type == F_RDLCK && my_errno == EDEADLK)
  349.       {
  350.     printf("%2d: write:  deadlock\n",id); fflush(stdout);
  351.     return 0;
  352.       }
  353.       fprintf(stderr,"%2d: Can't lock table (%d)\n",id,my_errno);
  354.       mi_close(file);
  355.       return 1;
  356.     }
  357.     if (rnd(2) == 0)
  358.       mi_extra(file,HA_EXTRA_WRITE_CACHE);
  359.   }
  360.  
  361.   sprintf(record.id,"%7d",getpid());
  362.   strmov(record.text,"Testing...");
  363.  
  364.   tries=(uint) rnd(100)+10;
  365.   for (i=count=0 ; i < tries ; i++)
  366.   {
  367.     uint32 tmp=rnd(80000)+20000;
  368.     int4store(record.nr,tmp);
  369.     if (!mi_write(file,record.id))
  370.       count++;
  371.     else
  372.     {
  373.       if (my_errno != HA_ERR_FOUND_DUPP_KEY)
  374.       {
  375.     fprintf(stderr,"%2d: Got error %d (errno %d) from write\n",id,my_errno,
  376.         errno);
  377.     return 1;
  378.       }
  379.     }
  380.   }
  381.   if (lock)
  382.   {
  383.     mi_extra(file,HA_EXTRA_NO_CACHE);
  384.     if (mi_lock_database(file,F_UNLCK))
  385.     {
  386.       fprintf(stderr,"%2d: Can't unlock table\n",id);
  387.       exit(0);
  388.     }
  389.   }
  390.   printf("%2d: write:  %5d\n",id,count); fflush(stdout);
  391.   return 0;
  392. }
  393.  
  394.  
  395. int test_update(MI_INFO *file,int id,int lock_type)
  396. {
  397.   uint i,lock,found,next,prev,update;
  398.   uint32 tmp;
  399.   char find[4];
  400.   struct record new_record;
  401.  
  402.   lock=0;
  403.   if (rnd(2) == 0 || lock_type == F_RDLCK)
  404.   {
  405.     lock=1;
  406.     if (mi_lock_database(file,F_WRLCK))
  407.     {
  408.       if (lock_type == F_RDLCK && my_errno == EDEADLK)
  409.       {
  410.     printf("%2d: write:  deadlock\n",id); fflush(stdout);
  411.     return 0;
  412.       }
  413.       fprintf(stderr,"%2d: Can't lock table (%d)\n",id,my_errno);
  414.       return 1;
  415.     }
  416.   }
  417.   bzero((char*) &new_record,sizeof(new_record));
  418.   strmov(new_record.text,"Updated");
  419.  
  420.   found=next=prev=update=0;
  421.   for (i=0 ; i < 100 ; i++)
  422.   {
  423.     tmp=rnd(100000);
  424.     int4store(find,tmp);
  425.     if (!mi_rkey(file,record.id,1,(byte*) find,
  426.          sizeof(find),HA_READ_KEY_EXACT))
  427.       found++;
  428.     else
  429.     {
  430.       if (my_errno != HA_ERR_KEY_NOT_FOUND)
  431.       {
  432.     fprintf(stderr,"%2d: Got error %d from read in update\n",id,my_errno);
  433.     return 1;
  434.       }
  435.       else if (!mi_rnext(file,record.id,1))
  436.     next++;
  437.       else
  438.       {
  439.     if (my_errno != HA_ERR_END_OF_FILE)
  440.     {
  441.       fprintf(stderr,"%2d: Got error %d from rnext in update\n",
  442.           id,my_errno);
  443.       return 1;
  444.     }
  445.     else if (!mi_rprev(file,record.id,1))
  446.       prev++;
  447.     else
  448.     {
  449.       if (my_errno != HA_ERR_END_OF_FILE)
  450.       {
  451.         fprintf(stderr,"%2d: Got error %d from rnext in update\n",
  452.             id,my_errno);
  453.         return 1;
  454.       }
  455.       continue;
  456.     }
  457.       }
  458.     }
  459.     memcpy_fixed(new_record.id,record.id,sizeof(record.id));
  460.     tmp=rnd(20000)+40000;
  461.     int4store(new_record.nr,tmp);
  462.     if (!mi_update(file,record.id,new_record.id))
  463.       update++;
  464.     else
  465.     {
  466.       if (my_errno != HA_ERR_RECORD_CHANGED &&
  467.       my_errno != HA_ERR_RECORD_DELETED &&
  468.       my_errno != HA_ERR_FOUND_DUPP_KEY)
  469.       {
  470.     fprintf(stderr,"%2d: Got error %d from update\n",id,my_errno);
  471.     return 1;
  472.       }
  473.     }
  474.   }
  475.   if (lock)
  476.   {
  477.     if (mi_lock_database(file,F_UNLCK))
  478.     {
  479.       fprintf(stderr,"Can't unlock table,id, error%d\n",my_errno);
  480.       return 1;
  481.     }
  482.   }
  483.   printf("%2d: update: %5d\n",id,update); fflush(stdout);
  484.   return 0;
  485. }
  486.