Annotation of embedaddon/sqlite3/test/threadtest2.c, revision 1.1

1.1     ! misho       1: /*
        !             2: ** 2004 January 13
        !             3: **
        !             4: ** The author disclaims copyright to this source code.  In place of
        !             5: ** a legal notice, here is a blessing:
        !             6: **
        !             7: **    May you do good and not evil.
        !             8: **    May you find forgiveness for yourself and forgive others.
        !             9: **    May you share freely, never taking more than you give.
        !            10: **
        !            11: *************************************************************************
        !            12: ** This file implements a simple standalone program used to test whether
        !            13: ** or not the SQLite library is threadsafe.
        !            14: **
        !            15: ** This file is NOT part of the standard SQLite library.  It is used for
        !            16: ** testing only.
        !            17: */
        !            18: #include <stdio.h>
        !            19: #include <unistd.h>
        !            20: #include <pthread.h>
        !            21: #include <string.h>
        !            22: #include <stdlib.h>
        !            23: #include "sqlite.h"
        !            24: 
        !            25: /*
        !            26: ** Name of the database
        !            27: */
        !            28: #define DB_FILE "test.db"
        !            29: 
        !            30: /* 
        !            31: ** When this variable becomes non-zero, all threads stop
        !            32: ** what they are doing.
        !            33: */
        !            34: volatile int all_stop = 0;
        !            35: 
        !            36: /* 
        !            37: ** Callback from the integrity check.  If the result is anything other
        !            38: ** than "ok" it means the integrity check has failed.  Set the "all_stop"
        !            39: ** global variable to stop all other activity.  Print the error message
        !            40: ** or print OK if the string "ok" is seen.
        !            41: */
        !            42: int check_callback(void *pid, int argc, char **argv, char **notUsed2){
        !            43:   int id = (int)pid;
        !            44:   if( strcmp(argv[0],"ok") ){
        !            45:     all_stop = 1;
        !            46:     fprintf(stderr,"id: %s\n", id, argv[0]);
        !            47:   }else{
        !            48:     /* fprintf(stderr,"%d: OK\n", id); */
        !            49:   }
        !            50:   return 0;
        !            51: }
        !            52: 
        !            53: /*
        !            54: ** Do an integrity check on the database.  If the first integrity check
        !            55: ** fails, try it a second time.
        !            56: */
        !            57: int integrity_check(sqlite *db, int id){
        !            58:   int rc;
        !            59:   if( all_stop ) return 0;
        !            60:   /* fprintf(stderr,"%d: CHECK\n", id); */
        !            61:   rc = sqlite3_exec(db, "pragma integrity_check", check_callback, 0, 0);
        !            62:   if( rc!=SQLITE_OK && rc!=SQLITE_BUSY ){
        !            63:     fprintf(stderr,"%d, Integrity check returns %d\n", id, rc);
        !            64:   }
        !            65:   if( all_stop ){
        !            66:     sqlite3_exec(db, "pragma integrity_check", check_callback, 0, 0);
        !            67:   }
        !            68:   return 0;
        !            69: }
        !            70: 
        !            71: /*
        !            72: ** This is the worker thread
        !            73: */
        !            74: void *worker(void *workerArg){
        !            75:   sqlite *db;
        !            76:   int id = (int)workerArg;
        !            77:   int rc;
        !            78:   int cnt = 0;
        !            79:   fprintf(stderr, "Starting worker %d\n", id);
        !            80:   while( !all_stop && cnt++<10000 ){
        !            81:     if( cnt%100==0 ) printf("%d: %d\n", id, cnt);
        !            82:     while( (sqlite3_open(DB_FILE, &db))!=SQLITE_OK ) sched_yield();
        !            83:     sqlite3_exec(db, "PRAGMA synchronous=OFF", 0, 0, 0);
        !            84:     /* integrity_check(db, id); */
        !            85:     if( all_stop ){ sqlite3_close(db); break; }
        !            86:     /* fprintf(stderr, "%d: BEGIN\n", id); */
        !            87:     rc = sqlite3_exec(db, "INSERT INTO t1 VALUES('bogus data')", 0, 0, 0);
        !            88:     /* fprintf(stderr, "%d: END rc=%d\n", id, rc); */
        !            89:     sqlite3_close(db);
        !            90:   }
        !            91:   fprintf(stderr, "Worker %d finished\n", id);
        !            92:   return 0;
        !            93: }
        !            94: 
        !            95: /*
        !            96: ** Initialize the database and start the threads
        !            97: */
        !            98: int main(int argc, char **argv){
        !            99:   sqlite *db;
        !           100:   int i, rc;
        !           101:   pthread_t aThread[5];
        !           102: 
        !           103:   if( strcmp(DB_FILE,":memory:") ){
        !           104:     char *zJournal = sqlite3_mprintf("%s-journal", DB_FILE);
        !           105:     unlink(DB_FILE);
        !           106:     unlink(zJournal);
        !           107:     sqlite3_free(zJournal);
        !           108:   }  
        !           109:   sqlite3_open(DB_FILE, &db);
        !           110:   if( db==0 ){
        !           111:     fprintf(stderr,"unable to initialize database\n");
        !           112:     exit(1);
        !           113:   }
        !           114:   rc = sqlite3_exec(db, "CREATE TABLE t1(x);", 0,0,0);
        !           115:   if( rc ){
        !           116:     fprintf(stderr,"cannot create table t1: %d\n", rc);
        !           117:     exit(1);
        !           118:   }
        !           119:   sqlite3_close(db);
        !           120:   for(i=0; i<sizeof(aThread)/sizeof(aThread[0]); i++){
        !           121:     pthread_create(&aThread[i], 0, worker, (void*)i);
        !           122:   }
        !           123:   for(i=0; i<sizeof(aThread)/sizeof(aThread[i]); i++){
        !           124:     pthread_join(aThread[i], 0);
        !           125:   }
        !           126:   if( !all_stop ){
        !           127:     printf("Everything seems ok.\n");
        !           128:     return 0;
        !           129:   }else{
        !           130:     printf("We hit an error.\n");
        !           131:     return 1;
        !           132:   }
        !           133: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>