Annotation of embedaddon/sqlite3/tool/speedtest8inst1.c, revision 1.1.1.1

1.1       misho       1: /*
                      2: ** Performance test for SQLite.
                      3: **
                      4: ** This program reads ASCII text from a file named on the command-line
                      5: ** and submits that text  to SQLite for evaluation.  A new database
                      6: ** is created at the beginning of the program.  All statements are
                      7: ** timed using the high-resolution timer built into Intel-class processors.
                      8: **
                      9: ** To compile this program, first compile the SQLite library separately
                     10: ** will full optimizations.  For example:
                     11: **
                     12: **     gcc -c -O6 -DSQLITE_THREADSAFE=0 sqlite3.c
                     13: **
                     14: ** Then link against this program.  But to do optimize this program
                     15: ** because that defeats the hi-res timer.
                     16: **
                     17: **     gcc speedtest8.c sqlite3.o -ldl -I../src
                     18: **
                     19: ** Then run this program with a single argument which is the name of
                     20: ** a file containing SQL script that you want to test:
                     21: **
                     22: **     ./a.out test.db  test.sql
                     23: */
                     24: #include <stdio.h>
                     25: #include <string.h>
                     26: #include <stdlib.h>
                     27: #include <ctype.h>
                     28: #include <unistd.h>
                     29: #include <stdarg.h>
                     30: #include "sqlite3.h"
                     31: 
                     32: #include "test_osinst.c"
                     33: 
                     34: /*
                     35: ** Prepare and run a single statement of SQL.
                     36: */
                     37: static void prepareAndRun(sqlite3_vfs *pInstVfs, sqlite3 *db, const char *zSql){
                     38:   sqlite3_stmt *pStmt;
                     39:   const char *stmtTail;
                     40:   int rc;
                     41:   char zMessage[1024];
                     42:   zMessage[1023] = '\0';
                     43: 
                     44:   sqlite3_uint64 iTime;
                     45: 
                     46:   sqlite3_snprintf(1023, zMessage, "sqlite3_prepare_v2: %s", zSql);
                     47:   sqlite3_instvfs_binarylog_marker(pInstVfs, zMessage);
                     48: 
                     49:   iTime = sqlite3Hwtime();
                     50:   rc = sqlite3_prepare_v2(db, zSql, -1, &pStmt, &stmtTail);
                     51:   iTime = sqlite3Hwtime() - iTime;
                     52:   sqlite3_instvfs_binarylog_call(pInstVfs,BINARYLOG_PREPARE_V2,iTime,rc,zSql);
                     53: 
                     54:   if( rc==SQLITE_OK ){
                     55:     int nRow = 0;
                     56: 
                     57:     sqlite3_snprintf(1023, zMessage, "sqlite3_step loop: %s", zSql);
                     58:     sqlite3_instvfs_binarylog_marker(pInstVfs, zMessage);
                     59:     iTime = sqlite3Hwtime();
                     60:     while( (rc=sqlite3_step(pStmt))==SQLITE_ROW ){ nRow++; }
                     61:     iTime = sqlite3Hwtime() - iTime;
                     62:     sqlite3_instvfs_binarylog_call(pInstVfs, BINARYLOG_STEP, iTime, rc, zSql);
                     63: 
                     64:     sqlite3_snprintf(1023, zMessage, "sqlite3_finalize: %s", zSql);
                     65:     sqlite3_instvfs_binarylog_marker(pInstVfs, zMessage);
                     66:     iTime = sqlite3Hwtime();
                     67:     rc = sqlite3_finalize(pStmt);
                     68:     iTime = sqlite3Hwtime() - iTime;
                     69:     sqlite3_instvfs_binarylog_call(pInstVfs, BINARYLOG_FINALIZE, iTime, rc, zSql);
                     70:   }
                     71: }
                     72: 
                     73: static int stringcompare(const char *zLeft, const char *zRight){
                     74:   int ii;
                     75:   for(ii=0; zLeft[ii] && zRight[ii]; ii++){
                     76:     if( zLeft[ii]!=zRight[ii] ) return 0;
                     77:   }
                     78:   return( zLeft[ii]==zRight[ii] );
                     79: }
                     80: 
                     81: static char *readScriptFile(const char *zFile, int *pnScript){
                     82:   sqlite3_vfs *pVfs = sqlite3_vfs_find(0);
                     83:   sqlite3_file *p;
                     84:   int rc;
                     85:   sqlite3_int64 nByte;
                     86:   char *zData = 0;
                     87:   int flags = SQLITE_OPEN_READONLY|SQLITE_OPEN_MAIN_DB;
                     88: 
                     89:   p = (sqlite3_file *)malloc(pVfs->szOsFile);
                     90:   rc = pVfs->xOpen(pVfs, zFile, p, flags, &flags);
                     91:   if( rc!=SQLITE_OK ){
                     92:     goto error_out;
                     93:   }
                     94: 
                     95:   rc = p->pMethods->xFileSize(p, &nByte);
                     96:   if( rc!=SQLITE_OK ){
                     97:     goto close_out;
                     98:   }
                     99: 
                    100:   zData = (char *)malloc(nByte+1);
                    101:   rc = p->pMethods->xRead(p, zData, nByte, 0);
                    102:   if( rc!=SQLITE_OK ){
                    103:     goto close_out;
                    104:   }
                    105:   zData[nByte] = '\0';
                    106: 
                    107:   p->pMethods->xClose(p);
                    108:   free(p);
                    109:   *pnScript = nByte;
                    110:   return zData;
                    111: 
                    112: close_out:
                    113:   p->pMethods->xClose(p);
                    114: 
                    115: error_out:
                    116:   free(p);
                    117:   free(zData);
                    118:   return 0;
                    119: }
                    120: 
                    121: int main(int argc, char **argv){
                    122: 
                    123:   const char zUsageMsg[] = 
                    124:     "Usage: %s options...\n"
                    125:     "  where available options are:\n"
                    126:     "\n"
                    127:     "    -db      DATABASE-FILE  (database file to operate on)\n"
                    128:     "    -script  SCRIPT-FILE    (script file to read sql from)\n"
                    129:     "    -log     LOG-FILE       (log file to create)\n"
                    130:     "    -logdata                (log all data to log file)\n"
                    131:     "\n"
                    132:     "  Options -db, -script and -log are compulsory\n"
                    133:     "\n"
                    134:   ;
                    135: 
                    136:   const char *zDb = 0;
                    137:   const char *zScript = 0;
                    138:   const char *zLog = 0;
                    139:   int logdata = 0;
                    140: 
                    141:   int ii;
                    142:   int i, j;
                    143:   int rc;
                    144: 
                    145:   sqlite3_vfs *pInstVfs;                 /* Instrumentation VFS */
                    146: 
                    147:   char *zSql = 0;
                    148:   int nSql;
                    149: 
                    150:   sqlite3 *db;
                    151: 
                    152:   for(ii=1; ii<argc; ii++){
                    153:     if( stringcompare("-db", argv[ii]) && (ii+1)<argc ){
                    154:       zDb = argv[++ii];
                    155:     }
                    156: 
                    157:     else if( stringcompare("-script", argv[ii]) && (ii+1)<argc ){
                    158:       zScript = argv[++ii];
                    159:     }
                    160: 
                    161:     else if( stringcompare("-log", argv[ii]) && (ii+1)<argc ){
                    162:       zLog = argv[++ii];
                    163:     }
                    164: 
                    165:     else if( stringcompare("-logdata", argv[ii]) ){
                    166:       logdata = 1;
                    167:     }
                    168: 
                    169:     else {
                    170:       goto usage;
                    171:     }
                    172:   }
                    173:   if( !zDb || !zScript || !zLog ) goto usage;
                    174: 
                    175:   zSql = readScriptFile(zScript, &nSql);
                    176:   if( !zSql ){
                    177:     fprintf(stderr, "Failed to read script file\n");
                    178:     return -1;
                    179:   }
                    180: 
                    181:   pInstVfs = sqlite3_instvfs_binarylog("logging", 0, zLog, logdata);
                    182: 
                    183:   rc = sqlite3_open_v2(
                    184:      zDb, &db, SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE, "logging"
                    185:   );
                    186:   if( rc!=SQLITE_OK ){
                    187:     fprintf(stderr, "Failed to open db: %s\n", sqlite3_errmsg(db));
                    188:     return -2;
                    189:   }
                    190: 
                    191:   for(i=j=0; j<nSql; j++){
                    192:     if( zSql[j]==';' ){
                    193:       int isComplete;
                    194:       char c = zSql[j+1];
                    195:       zSql[j+1] = 0;
                    196:       isComplete = sqlite3_complete(&zSql[i]);
                    197:       zSql[j+1] = c;
                    198:       if( isComplete ){
                    199:         zSql[j] = 0;
                    200:         while( i<j && isspace(zSql[i]) ){ i++; }
                    201:         if( i<j ){
                    202:           prepareAndRun(pInstVfs, db, &zSql[i]);
                    203:         }
                    204:         zSql[j] = ';';
                    205:         i = j+1;
                    206:       }
                    207:     }
                    208:   }
                    209:   
                    210:   sqlite3_instvfs_destroy(pInstVfs);
                    211:   return 0;
                    212:   
                    213: usage:
                    214:   fprintf(stderr, zUsageMsg, argv[0]);
                    215:   return -3;
                    216: }

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