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