Return to threadtest2.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / sqlite3 / test

Annotation of embedaddon/sqlite3/test/threadtest2.c, revision 1.1.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: }