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

MySQL数据库

开发平台:

Visual C++

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