home *** CD-ROM | disk | FTP | other *** search
/ Chip 2001 January / Chip_2001-01_cd1.bin / tema / mysql / mysql-3.23.28g-win-source.exe / isam / test3.c < prev    next >
C/C++ Source or Header  |  2000-08-31  |  11KB  |  480 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 "nisam.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.ISM";
  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(N_INFO *,int),test_write(N_INFO *,int,int),
  47.     test_update(N_INFO *,int,int),test_rrnd(N_INFO *,int);
  48.  
  49. struct record {
  50.   char id[8];
  51.   uint32 nr;
  52.   char text[10];
  53. } record;
  54.  
  55.  
  56. int main(int argc,char **argv)
  57. {
  58.   int status,wait_ret;
  59.   uint i;
  60.   N_KEYDEF keyinfo[10];
  61.   N_RECINFO recinfo[10];
  62.   MY_INIT(argv[0]);
  63.  
  64.   get_options(argc,argv);
  65.  
  66.   keyinfo[0].seg[0].base.start=0;
  67.   keyinfo[0].seg[0].base.length=8;
  68.   keyinfo[0].seg[0].base.type=HA_KEYTYPE_TEXT;
  69.   keyinfo[0].seg[0].base.flag=HA_SPACE_PACK;
  70.   keyinfo[0].seg[1].base.type=0;
  71.   keyinfo[0].base.flag = (uint8) HA_PACK_KEY;
  72.   keyinfo[1].seg[0].base.start=8;
  73.   keyinfo[1].seg[0].base.length=sizeof(uint32);
  74.   keyinfo[1].seg[0].base.type=HA_KEYTYPE_LONG_INT;
  75.   keyinfo[1].seg[0].base.flag=0;
  76.   keyinfo[1].seg[1].base.type=0;
  77.   keyinfo[1].base.flag =HA_NOSAME;
  78.  
  79.   recinfo[0].base.type=0;
  80.   recinfo[0].base.length=sizeof(record.id);
  81.   recinfo[1].base.type=0;
  82.   recinfo[1].base.length=sizeof(record.nr);
  83.   recinfo[2].base.type=0;
  84.   recinfo[2].base.length=sizeof(record.text);
  85.   recinfo[3].base.type=FIELD_LAST;
  86.  
  87.   puts("- Creating isam-file");
  88.   my_delete(filename,MYF(0));        /* Remove old locks under gdb */
  89.   if (nisam_create(filename,2,&keyinfo[0],&recinfo[0],10000,0,0,0,0L))
  90.     exit(1);
  91.  
  92.   rnd_init(0);
  93.   printf("- Starting %d processes\n",forks); fflush(stdout);
  94.   for (i=0 ; i < forks; i++)
  95.   {
  96.     if (!fork())
  97.     {
  98.       start_test(i+1);
  99.       sleep(1);
  100.       return 0;
  101.     }
  102.     VOID(rnd(1));
  103.   }
  104.  
  105.   for (i=0 ; i < forks ; i++)
  106.     while ((wait_ret=wait(&status)) && wait_ret == -1);
  107.   return 0;
  108. }
  109.  
  110.  
  111. static void get_options(argc,argv)
  112. int argc;
  113. char *argv[];
  114. {
  115.   char *pos,*progname;
  116.   DEBUGGER_OFF;
  117.  
  118.   progname= argv[0];
  119.  
  120.   while (--argc >0 && *(pos = *(++argv)) == '-' ) {
  121.     switch(*++pos) {
  122.     case 'l':
  123.       use_log=1;
  124.       break;
  125.     case 'f':
  126.       forks=atoi(++pos);
  127.       break;
  128.     case 't':
  129.       tests=atoi(++pos);
  130.       break;
  131.     case 'K':                /* Use key cacheing */
  132.       key_cacheing=1;
  133.       break;
  134.     case 'A':                /* All flags */
  135.       use_log=key_cacheing=1;
  136.       break;
  137.    case '?':
  138.     case 'I':
  139.     case 'V':
  140.       printf("%s  Ver 1.0 for %s at %s\n",progname,SYSTEM_TYPE,MACHINE_TYPE);
  141.       puts("TCX Datakonsult AB, by Monty, for your professional use\n");
  142.       puts("Test av locking with threads\n");
  143.       printf("Usage: %s [-?lKA] [-f#] [-t#]\n",progname);
  144.       exit(0);
  145.     case '#':
  146.       DEBUGGER_ON;
  147.       DBUG_PUSH (++pos);
  148.       break;
  149.     default:
  150.       printf("Illegal option: '%c'\n",*pos);
  151.       break;
  152.     }
  153.   }
  154.   return;
  155. }
  156.  
  157.  
  158. void start_test(int id)
  159. {
  160.   uint i;
  161.   int error,lock_type;
  162.   N_ISAMINFO isam_info;
  163.   N_INFO *file,*file1,*file2,*lock;
  164.  
  165.   if (use_log)
  166.     nisam_log(1);
  167.   if (!(file1=nisam_open(filename,O_RDWR,HA_OPEN_WAIT_IF_LOCKED)) ||
  168.       !(file2=nisam_open(filename,O_RDWR,HA_OPEN_WAIT_IF_LOCKED)))
  169.   {
  170.     fprintf(stderr,"Can't open isam-file: %s\n",filename);
  171.     exit(1);
  172.   }
  173.   if (key_cacheing && rnd(2) == 0)
  174.     init_key_cache(65536L,(uint) IO_SIZE*4*10);
  175.   printf("Process %d, pid: %d\n",id,getpid()); fflush(stdout);
  176.  
  177.   for (error=i=0 ; i < tests && !error; i++)
  178.   {
  179.     file= (rnd(2) == 1) ? file1 : file2;
  180.     lock=0 ; lock_type=0;
  181.     if (rnd(10) == 0)
  182.     {
  183.       if (nisam_lock_database(lock=(rnd(2) ? file1 : file2),
  184.                lock_type=(rnd(2) == 0 ? F_RDLCK : F_WRLCK)))
  185.       {
  186.     fprintf(stderr,"%2d: start: Can't lock table %d\n",id,my_errno);
  187.     error=1;
  188.     break;
  189.       }
  190.     }
  191.     switch (rnd(4)) {
  192.     case 0: error=test_read(file,id); break;
  193.     case 1: error=test_rrnd(file,id); break;
  194.     case 2: error=test_write(file,id,lock_type); break;
  195.     case 3: error=test_update(file,id,lock_type); break;
  196.     }
  197.     if (lock)
  198.       nisam_lock_database(lock,F_UNLCK);
  199.   }
  200.   if (!error)
  201.   {
  202.     nisam_info(file1,&isam_info,0);
  203.     printf("%2d: End of test.  Records:  %ld  Deleted:  %ld\n",
  204.        id,isam_info.records,isam_info.deleted);
  205.     fflush(stdout);
  206.   }
  207.  
  208.   nisam_close(file1);
  209.   nisam_close(file2);
  210.   if (use_log)
  211.     nisam_log(0);
  212.   if (error)
  213.   {
  214.     printf("%2d: Aborted\n",id); fflush(stdout);
  215.     exit(1);
  216.   }
  217. }
  218.  
  219.  
  220. int test_read(N_INFO *file,int id)
  221. {
  222.   uint i,lock,found,next,prev;
  223.   ulong find;
  224.  
  225.   lock=0;
  226.   if (rnd(2) == 0)
  227.   {
  228.     lock=1;
  229.     if (nisam_lock_database(file,F_RDLCK))
  230.     {
  231.       fprintf(stderr,"%2d: Can't lock table %d\n",id,my_errno);
  232.       return 1;
  233.     }
  234.   }
  235.  
  236.   found=next=prev=0;
  237.   for (i=0 ; i < 100 ; i++)
  238.   {
  239.     find=rnd(100000);
  240.     if (!nisam_rkey(file,record.id,1,(byte*) &find,
  241.          sizeof(find),HA_READ_KEY_EXACT))
  242.       found++;
  243.     else
  244.     {
  245.       if (my_errno != HA_ERR_KEY_NOT_FOUND)
  246.       {
  247.     fprintf(stderr,"%2d: Got error %d from read in read\n",id,my_errno);
  248.     return 1;
  249.       }
  250.       else if (!nisam_rnext(file,record.id,1))
  251.     next++;
  252.       else
  253.       {
  254.     if (my_errno != HA_ERR_END_OF_FILE)
  255.     {
  256.       fprintf(stderr,"%2d: Got error %d from rnext in read\n",id,my_errno);
  257.       return 1;
  258.     }
  259.     else if (!nisam_rprev(file,record.id,1))
  260.       prev++;
  261.     else
  262.     {
  263.       if (my_errno != HA_ERR_END_OF_FILE)
  264.       {
  265.         fprintf(stderr,"%2d: Got error %d from rnext in read\n",
  266.             id,my_errno);
  267.         return 1;
  268.       }
  269.     }
  270.       }
  271.     }
  272.   }
  273.   if (lock)
  274.   {
  275.     if (nisam_lock_database(file,F_UNLCK))
  276.     {
  277.       fprintf(stderr,"%2d: Can't unlock table\n",id);
  278.       return 1;
  279.     }
  280.   }
  281.   printf("%2d: read:   found: %5d  next: %5d   prev: %5d\n",
  282.      id,found,next,prev);
  283.   fflush(stdout);
  284.   return 0;
  285. }
  286.  
  287.  
  288. int test_rrnd(N_INFO *file,int id)
  289. {
  290.   uint count,lock;
  291.  
  292.   lock=0;
  293.   if (rnd(2) == 0)
  294.   {
  295.     lock=1;
  296.     if (nisam_lock_database(file,F_RDLCK))
  297.     {
  298.       fprintf(stderr,"%2d: Can't lock table (%d)\n",id,my_errno);
  299.       nisam_close(file);
  300.       return 1;
  301.     }
  302.     if (rnd(2) == 0)
  303.       nisam_extra(file,HA_EXTRA_CACHE);
  304.   }
  305.  
  306.   count=0;
  307.   if (nisam_rrnd(file,record.id,0L))
  308.   {
  309.     if (my_errno == HA_ERR_END_OF_FILE)
  310.       goto end;
  311.     fprintf(stderr,"%2d: Can't read first record (%d)\n",id,my_errno);
  312.     return 1;
  313.   }
  314.   for (count=1 ; !nisam_rrnd(file,record.id,NI_POS_ERROR) ;count++) ;
  315.   if (my_errno != HA_ERR_END_OF_FILE)
  316.   {
  317.     fprintf(stderr,"%2d: Got error %d from rrnd\n",id,my_errno);
  318.     return 1;
  319.   }
  320.  
  321. end:
  322.   if (lock)
  323.   {
  324.     nisam_extra(file,HA_EXTRA_NO_CACHE);
  325.     if (nisam_lock_database(file,F_UNLCK))
  326.     {
  327.       fprintf(stderr,"%2d: Can't unlock table\n",id);
  328.       exit(0);
  329.     }
  330.   }
  331.   printf("%2d: rrnd:   %5d\n",id,count); fflush(stdout);
  332.   return 0;
  333. }
  334.  
  335.  
  336. int test_write(N_INFO *file,int id,int lock_type)
  337. {
  338.   uint i,tries,count,lock;
  339.  
  340.   lock=0;
  341.   if (rnd(2) == 0 || lock_type == F_RDLCK)
  342.   {
  343.     lock=1;
  344.     if (nisam_lock_database(file,F_WRLCK))
  345.     {
  346.       if (lock_type == F_RDLCK && my_errno == EDEADLK)
  347.       {
  348.     printf("%2d: write:  deadlock\n",id); fflush(stdout);
  349.     return 0;
  350.       }
  351.       fprintf(stderr,"%2d: Can't lock table (%d)\n",id,my_errno);
  352.       nisam_close(file);
  353.       return 1;
  354.     }
  355.     if (rnd(2) == 0)
  356.       nisam_extra(file,HA_EXTRA_WRITE_CACHE);
  357.   }
  358.  
  359.   sprintf(record.id,"%7d",getpid());
  360.   strmov(record.text,"Testing...");
  361.  
  362.   tries=(uint) rnd(100)+10;
  363.   for (i=count=0 ; i < tries ; i++)
  364.   {
  365.     record.nr=rnd(80000)+20000;
  366.     if (!nisam_write(file,record.id))
  367.       count++;
  368.     else
  369.     {
  370.       if (my_errno != HA_ERR_FOUND_DUPP_KEY)
  371.       {
  372.     fprintf(stderr,"%2d: Got error %d (errno %d) from write\n",id,my_errno,
  373.         errno);
  374.     return 1;
  375.       }
  376.     }
  377.   }
  378.   if (lock)
  379.   {
  380.     nisam_extra(file,HA_EXTRA_NO_CACHE);
  381.     if (nisam_lock_database(file,F_UNLCK))
  382.     {
  383.       fprintf(stderr,"%2d: Can't unlock table\n",id);
  384.       exit(0);
  385.     }
  386.   }
  387.   printf("%2d: write:  %5d\n",id,count); fflush(stdout);
  388.   return 0;
  389. }
  390.  
  391.  
  392. int test_update(N_INFO *file,int id,int lock_type)
  393. {
  394.   uint i,lock,found,next,prev,update;
  395.   ulong find;
  396.   struct record new_record;
  397.  
  398.   lock=0;
  399.   if (rnd(2) == 0 || lock_type == F_RDLCK)
  400.   {
  401.     lock=1;
  402.     if (nisam_lock_database(file,F_WRLCK))
  403.     {
  404.       if (lock_type == F_RDLCK && my_errno == EDEADLK)
  405.       {
  406.     printf("%2d: write:  deadlock\n",id); fflush(stdout);
  407.     return 0;
  408.       }
  409.       fprintf(stderr,"%2d: Can't lock table (%d)\n",id,my_errno);
  410.       return 1;
  411.     }
  412.   }
  413.   bzero((char*) &new_record,sizeof(new_record));
  414.   strmov(new_record.text,"Updated");
  415.  
  416.   found=next=prev=update=0;
  417.   for (i=0 ; i < 100 ; i++)
  418.   {
  419.     find=rnd(100000);
  420.     if (!nisam_rkey(file,record.id,1,(byte*) &find,
  421.          sizeof(find),HA_READ_KEY_EXACT))
  422.       found++;
  423.     else
  424.     {
  425.       if (my_errno != HA_ERR_KEY_NOT_FOUND)
  426.       {
  427.     fprintf(stderr,"%2d: Got error %d from read in update\n",id,my_errno);
  428.     return 1;
  429.       }
  430.       else if (!nisam_rnext(file,record.id,1))
  431.     next++;
  432.       else
  433.       {
  434.     if (my_errno != HA_ERR_END_OF_FILE)
  435.     {
  436.       fprintf(stderr,"%2d: Got error %d from rnext in update\n",
  437.           id,my_errno);
  438.       return 1;
  439.     }
  440.     else if (!nisam_rprev(file,record.id,1))
  441.       prev++;
  442.     else
  443.     {
  444.       if (my_errno != HA_ERR_END_OF_FILE)
  445.       {
  446.         fprintf(stderr,"%2d: Got error %d from rnext in update\n",
  447.             id,my_errno);
  448.         return 1;
  449.       }
  450.       continue;
  451.     }
  452.       }
  453.     }
  454.     memcpy_fixed(new_record.id,record.id,sizeof(record.id));
  455.     new_record.nr=rnd(20000)+40000;
  456.     if (!nisam_update(file,record.id,new_record.id))
  457.       update++;
  458.     else
  459.     {
  460.       if (my_errno != HA_ERR_RECORD_CHANGED &&
  461.       my_errno != HA_ERR_RECORD_DELETED &&
  462.       my_errno != HA_ERR_FOUND_DUPP_KEY)
  463.       {
  464.     fprintf(stderr,"%2d: Got error %d from update\n",id,my_errno);
  465.     return 1;
  466.       }
  467.     }
  468.   }
  469.   if (lock)
  470.   {
  471.     if (nisam_lock_database(file,F_UNLCK))
  472.     {
  473.       fprintf(stderr,"Can't unlock table,id, error%d\n",my_errno);
  474.       return 1;
  475.     }
  476.   }
  477.   printf("%2d: update: %5d\n",id,update); fflush(stdout);
  478.   return 0;
  479. }
  480.