mi_test3.c
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:11k
- /* Copyright (C) 2000 MySQL AB & MySQL Finland AB & TCX DataKonsult AB
- This program is free software; you can redistribute it and/or modify
- it under the terms of the GNU General Public License as published by
- the Free Software Foundation; either version 2 of the License, or
- (at your option) any later version.
- This program is distributed in the hope that it will be useful,
- but WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- GNU General Public License for more details.
- You should have received a copy of the GNU General Public License
- along with this program; if not, write to the Free Software
- Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */
- /* Test av locking */
- #ifndef __NETWARE__
- #include "myisam.h"
- #include <sys/types.h>
- #ifdef HAVE_SYS_WAIT_H
- # include <sys/wait.h>
- #endif
- #ifndef WEXITSTATUS
- # define WEXITSTATUS(stat_val) ((unsigned)(stat_val) >> 8)
- #endif
- #ifndef WIFEXITED
- # define WIFEXITED(stat_val) (((stat_val) & 255) == 0)
- #endif
- #if defined(HAVE_LRAND48)
- #define rnd(X) (lrand48() % X)
- #define rnd_init(X) srand48(X)
- #else
- #define rnd(X) (random() % X)
- #define rnd_init(X) srandom(X)
- #endif
- const char *filename= "test3";
- uint tests=10,forks=10,key_cacheing=0,use_log=0;
- static void get_options(int argc, char *argv[]);
- void start_test(int id);
- int test_read(MI_INFO *,int),test_write(MI_INFO *,int,int),
- test_update(MI_INFO *,int,int),test_rrnd(MI_INFO *,int);
- struct record {
- char id[8];
- char nr[4];
- char text[10];
- } record;
- int main(int argc,char **argv)
- {
- int status,wait_ret;
- uint i=0;
- MI_KEYDEF keyinfo[10];
- MI_COLUMNDEF recinfo[10];
- HA_KEYSEG keyseg[10][2];
- MY_INIT(argv[0]);
- get_options(argc,argv);
- bzero((char*) keyinfo,sizeof(keyinfo));
- bzero((char*) recinfo,sizeof(recinfo));
- keyinfo[0].seg= &keyseg[0][0];
- keyinfo[0].seg[0].start=0;
- keyinfo[0].seg[0].length=8;
- keyinfo[0].seg[0].type=HA_KEYTYPE_TEXT;
- keyinfo[0].seg[0].flag=HA_SPACE_PACK;
- keyinfo[0].key_alg=HA_KEY_ALG_BTREE;
- keyinfo[0].keysegs=1;
- keyinfo[0].flag = (uint8) HA_PACK_KEY;
- keyinfo[1].seg= &keyseg[1][0];
- keyinfo[1].seg[0].start=8;
- keyinfo[1].seg[0].length=4; /* Long is always 4 in myisam */
- keyinfo[1].seg[0].type=HA_KEYTYPE_LONG_INT;
- keyinfo[1].seg[0].flag=0;
- keyinfo[1].key_alg=HA_KEY_ALG_BTREE;
- keyinfo[1].keysegs=1;
- keyinfo[1].flag =HA_NOSAME;
- recinfo[0].type=0;
- recinfo[0].length=sizeof(record.id);
- recinfo[1].type=0;
- recinfo[1].length=sizeof(record.nr);
- recinfo[2].type=0;
- recinfo[2].length=sizeof(record.text);
- puts("- Creating myisam-file");
- my_delete(filename,MYF(0)); /* Remove old locks under gdb */
- if (mi_create(filename,2,&keyinfo[0],2,&recinfo[0],0,(MI_UNIQUEDEF*) 0,
- (MI_CREATE_INFO*) 0,0))
- exit(1);
- rnd_init(0);
- printf("- Starting %d processesn",forks); fflush(stdout);
- for (i=0 ; i < forks; i++)
- {
- if (!fork())
- {
- start_test(i+1);
- sleep(1);
- return 0;
- }
- VOID(rnd(1));
- }
- for (i=0 ; i < forks ; i++)
- while ((wait_ret=wait(&status)) && wait_ret == -1);
- return 0;
- }
- static void get_options(int argc, char **argv)
- {
- char *pos,*progname;
- DEBUGGER_OFF;
- progname= argv[0];
- while (--argc >0 && *(pos = *(++argv)) == '-' ) {
- switch(*++pos) {
- case 'l':
- use_log=1;
- break;
- case 'f':
- forks=atoi(++pos);
- break;
- case 't':
- tests=atoi(++pos);
- break;
- case 'K': /* Use key cacheing */
- key_cacheing=1;
- break;
- case 'A': /* All flags */
- use_log=key_cacheing=1;
- break;
- case '?':
- case 'I':
- case 'V':
- printf("%s Ver 1.0 for %s at %sn",progname,SYSTEM_TYPE,MACHINE_TYPE);
- puts("By Monty, for your professional usen");
- puts("Test av locking with threadsn");
- printf("Usage: %s [-?lKA] [-f#] [-t#]n",progname);
- exit(0);
- case '#':
- DEBUGGER_ON;
- DBUG_PUSH (++pos);
- break;
- default:
- printf("Illegal option: '%c'n",*pos);
- break;
- }
- }
- return;
- }
- void start_test(int id)
- {
- uint i;
- int error,lock_type;
- MI_ISAMINFO isam_info;
- MI_INFO *file,*file1,*file2=0,*lock;
- if (use_log)
- mi_log(1);
- if (!(file1=mi_open(filename,O_RDWR,HA_OPEN_WAIT_IF_LOCKED)) ||
- !(file2=mi_open(filename,O_RDWR,HA_OPEN_WAIT_IF_LOCKED)))
- {
- fprintf(stderr,"Can't open isam-file: %sn",filename);
- exit(1);
- }
- if (key_cacheing && rnd(2) == 0)
- init_key_cache(dflt_key_cache, KEY_CACHE_BLOCK_SIZE, 65536L, 0, 0);
- printf("Process %d, pid: %dn",id,getpid()); fflush(stdout);
- for (error=i=0 ; i < tests && !error; i++)
- {
- file= (rnd(2) == 1) ? file1 : file2;
- lock=0 ; lock_type=0;
- if (rnd(10) == 0)
- {
- if (mi_lock_database(lock=(rnd(2) ? file1 : file2),
- lock_type=(rnd(2) == 0 ? F_RDLCK : F_WRLCK)))
- {
- fprintf(stderr,"%2d: start: Can't lock table %dn",id,my_errno);
- error=1;
- break;
- }
- }
- switch (rnd(4)) {
- case 0: error=test_read(file,id); break;
- case 1: error=test_rrnd(file,id); break;
- case 2: error=test_write(file,id,lock_type); break;
- case 3: error=test_update(file,id,lock_type); break;
- }
- if (lock)
- mi_lock_database(lock,F_UNLCK);
- }
- if (!error)
- {
- mi_status(file1,&isam_info,HA_STATUS_VARIABLE);
- printf("%2d: End of test. Records: %ld Deleted: %ldn",
- id,(long) isam_info.records, (long) isam_info.deleted);
- fflush(stdout);
- }
- mi_close(file1);
- mi_close(file2);
- if (use_log)
- mi_log(0);
- if (error)
- {
- printf("%2d: Abortedn",id); fflush(stdout);
- exit(1);
- }
- }
- int test_read(MI_INFO *file,int id)
- {
- uint i,lock,found,next,prev;
- ulong find;
- lock=0;
- if (rnd(2) == 0)
- {
- lock=1;
- if (mi_lock_database(file,F_RDLCK))
- {
- fprintf(stderr,"%2d: Can't lock table %dn",id,my_errno);
- return 1;
- }
- }
- found=next=prev=0;
- for (i=0 ; i < 100 ; i++)
- {
- find=rnd(100000);
- if (!mi_rkey(file,record.id,1,(byte*) &find,
- sizeof(find),HA_READ_KEY_EXACT))
- found++;
- else
- {
- if (my_errno != HA_ERR_KEY_NOT_FOUND)
- {
- fprintf(stderr,"%2d: Got error %d from read in readn",id,my_errno);
- return 1;
- }
- else if (!mi_rnext(file,record.id,1))
- next++;
- else
- {
- if (my_errno != HA_ERR_END_OF_FILE)
- {
- fprintf(stderr,"%2d: Got error %d from rnext in readn",id,my_errno);
- return 1;
- }
- else if (!mi_rprev(file,record.id,1))
- prev++;
- else
- {
- if (my_errno != HA_ERR_END_OF_FILE)
- {
- fprintf(stderr,"%2d: Got error %d from rnext in readn",
- id,my_errno);
- return 1;
- }
- }
- }
- }
- }
- if (lock)
- {
- if (mi_lock_database(file,F_UNLCK))
- {
- fprintf(stderr,"%2d: Can't unlock tablen",id);
- return 1;
- }
- }
- printf("%2d: read: found: %5d next: %5d prev: %5dn",
- id,found,next,prev);
- fflush(stdout);
- return 0;
- }
- int test_rrnd(MI_INFO *file,int id)
- {
- uint count,lock;
- lock=0;
- if (rnd(2) == 0)
- {
- lock=1;
- if (mi_lock_database(file,F_RDLCK))
- {
- fprintf(stderr,"%2d: Can't lock table (%d)n",id,my_errno);
- mi_close(file);
- return 1;
- }
- if (rnd(2) == 0)
- mi_extra(file,HA_EXTRA_CACHE,0);
- }
- count=0;
- if (mi_rrnd(file,record.id,0L))
- {
- if (my_errno == HA_ERR_END_OF_FILE)
- goto end;
- fprintf(stderr,"%2d: Can't read first record (%d)n",id,my_errno);
- return 1;
- }
- for (count=1 ; !mi_rrnd(file,record.id,HA_OFFSET_ERROR) ;count++) ;
- if (my_errno != HA_ERR_END_OF_FILE)
- {
- fprintf(stderr,"%2d: Got error %d from rrndn",id,my_errno);
- return 1;
- }
- end:
- if (lock)
- {
- mi_extra(file,HA_EXTRA_NO_CACHE,0);
- if (mi_lock_database(file,F_UNLCK))
- {
- fprintf(stderr,"%2d: Can't unlock tablen",id);
- exit(0);
- }
- }
- printf("%2d: rrnd: %5dn",id,count); fflush(stdout);
- return 0;
- }
- int test_write(MI_INFO *file,int id,int lock_type)
- {
- uint i,tries,count,lock;
- lock=0;
- if (rnd(2) == 0 || lock_type == F_RDLCK)
- {
- lock=1;
- if (mi_lock_database(file,F_WRLCK))
- {
- if (lock_type == F_RDLCK && my_errno == EDEADLK)
- {
- printf("%2d: write: deadlockn",id); fflush(stdout);
- return 0;
- }
- fprintf(stderr,"%2d: Can't lock table (%d)n",id,my_errno);
- mi_close(file);
- return 1;
- }
- if (rnd(2) == 0)
- mi_extra(file,HA_EXTRA_WRITE_CACHE,0);
- }
- sprintf(record.id,"%7d",getpid());
- strnmov(record.text,"Testing...", sizeof(record.text));
- tries=(uint) rnd(100)+10;
- for (i=count=0 ; i < tries ; i++)
- {
- uint32 tmp=rnd(80000)+20000;
- int4store(record.nr,tmp);
- if (!mi_write(file,record.id))
- count++;
- else
- {
- if (my_errno != HA_ERR_FOUND_DUPP_KEY)
- {
- fprintf(stderr,"%2d: Got error %d (errno %d) from writen",id,my_errno,
- errno);
- return 1;
- }
- }
- }
- if (lock)
- {
- mi_extra(file,HA_EXTRA_NO_CACHE,0);
- if (mi_lock_database(file,F_UNLCK))
- {
- fprintf(stderr,"%2d: Can't unlock tablen",id);
- exit(0);
- }
- }
- printf("%2d: write: %5dn",id,count); fflush(stdout);
- return 0;
- }
- int test_update(MI_INFO *file,int id,int lock_type)
- {
- uint i,lock,found,next,prev,update;
- uint32 tmp;
- char find[4];
- struct record new_record;
- lock=0;
- if (rnd(2) == 0 || lock_type == F_RDLCK)
- {
- lock=1;
- if (mi_lock_database(file,F_WRLCK))
- {
- if (lock_type == F_RDLCK && my_errno == EDEADLK)
- {
- printf("%2d: write: deadlockn",id); fflush(stdout);
- return 0;
- }
- fprintf(stderr,"%2d: Can't lock table (%d)n",id,my_errno);
- return 1;
- }
- }
- bzero((char*) &new_record,sizeof(new_record));
- strmov(new_record.text,"Updated");
- found=next=prev=update=0;
- for (i=0 ; i < 100 ; i++)
- {
- tmp=rnd(100000);
- int4store(find,tmp);
- if (!mi_rkey(file,record.id,1,(byte*) find,
- sizeof(find),HA_READ_KEY_EXACT))
- found++;
- else
- {
- if (my_errno != HA_ERR_KEY_NOT_FOUND)
- {
- fprintf(stderr,"%2d: Got error %d from read in updaten",id,my_errno);
- return 1;
- }
- else if (!mi_rnext(file,record.id,1))
- next++;
- else
- {
- if (my_errno != HA_ERR_END_OF_FILE)
- {
- fprintf(stderr,"%2d: Got error %d from rnext in updaten",
- id,my_errno);
- return 1;
- }
- else if (!mi_rprev(file,record.id,1))
- prev++;
- else
- {
- if (my_errno != HA_ERR_END_OF_FILE)
- {
- fprintf(stderr,"%2d: Got error %d from rnext in updaten",
- id,my_errno);
- return 1;
- }
- continue;
- }
- }
- }
- memcpy_fixed(new_record.id,record.id,sizeof(record.id));
- tmp=rnd(20000)+40000;
- int4store(new_record.nr,tmp);
- if (!mi_update(file,record.id,new_record.id))
- update++;
- else
- {
- if (my_errno != HA_ERR_RECORD_CHANGED &&
- my_errno != HA_ERR_RECORD_DELETED &&
- my_errno != HA_ERR_FOUND_DUPP_KEY)
- {
- fprintf(stderr,"%2d: Got error %d from updaten",id,my_errno);
- return 1;
- }
- }
- }
- if (lock)
- {
- if (mi_lock_database(file,F_UNLCK))
- {
- fprintf(stderr,"Can't unlock table,id, error%dn",my_errno);
- return 1;
- }
- }
- printf("%2d: update: %5dn",id,update); fflush(stdout);
- return 0;
- }
- #else /* __NETWARE__ */
- #include <stdio.h>
- main()
- {
- fprintf(stderr,"this test has not been ported to NetWaren");
- return 0;
- }
- #endif /* __NETWARE__ */