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 #define ISSPACE(X) isspace((unsigned char)(X)) 33 34 #include "test_osinst.c" 35 36 /* 37 ** Prepare and run a single statement of SQL. 38 */ 39 static void prepareAndRun(sqlite3_vfs *pInstVfs, sqlite3 *db, const char *zSql){ 40 sqlite3_stmt *pStmt; 41 const char *stmtTail; 42 int rc; 43 char zMessage[1024]; 44 zMessage[1023] = '\0'; 45 46 sqlite3_uint64 iTime; 47 48 sqlite3_snprintf(1023, zMessage, "sqlite3_prepare_v2: %s", zSql); 49 sqlite3_instvfs_binarylog_marker(pInstVfs, zMessage); 50 51 iTime = sqlite3Hwtime(); 52 rc = sqlite3_prepare_v2(db, zSql, -1, &pStmt, &stmtTail); 53 iTime = sqlite3Hwtime() - iTime; 54 sqlite3_instvfs_binarylog_call(pInstVfs,BINARYLOG_PREPARE_V2,iTime,rc,zSql); 55 56 if( rc==SQLITE_OK ){ 57 int nRow = 0; 58 59 sqlite3_snprintf(1023, zMessage, "sqlite3_step loop: %s", zSql); 60 sqlite3_instvfs_binarylog_marker(pInstVfs, zMessage); 61 iTime = sqlite3Hwtime(); 62 while( (rc=sqlite3_step(pStmt))==SQLITE_ROW ){ nRow++; } 63 iTime = sqlite3Hwtime() - iTime; 64 sqlite3_instvfs_binarylog_call(pInstVfs, BINARYLOG_STEP, iTime, rc, zSql); 65 66 sqlite3_snprintf(1023, zMessage, "sqlite3_finalize: %s", zSql); 67 sqlite3_instvfs_binarylog_marker(pInstVfs, zMessage); 68 iTime = sqlite3Hwtime(); 69 rc = sqlite3_finalize(pStmt); 70 iTime = sqlite3Hwtime() - iTime; 71 sqlite3_instvfs_binarylog_call(pInstVfs, BINARYLOG_FINALIZE, iTime, rc, zSql); 72 } 73 } 74 75 static int stringcompare(const char *zLeft, const char *zRight){ 76 int ii; 77 for(ii=0; zLeft[ii] && zRight[ii]; ii++){ 78 if( zLeft[ii]!=zRight[ii] ) return 0; 79 } 80 return( zLeft[ii]==zRight[ii] ); 81 } 82 83 static char *readScriptFile(const char *zFile, int *pnScript){ 84 sqlite3_vfs *pVfs = sqlite3_vfs_find(0); 85 sqlite3_file *p; 86 int rc; 87 sqlite3_int64 nByte; 88 char *zData = 0; 89 int flags = SQLITE_OPEN_READONLY|SQLITE_OPEN_MAIN_DB; 90 91 p = (sqlite3_file *)malloc(pVfs->szOsFile); 92 rc = pVfs->xOpen(pVfs, zFile, p, flags, &flags); 93 if( rc!=SQLITE_OK ){ 94 goto error_out; 95 } 96 97 rc = p->pMethods->xFileSize(p, &nByte); 98 if( rc!=SQLITE_OK ){ 99 goto close_out; 100 } 101 102 zData = (char *)malloc(nByte+1); 103 rc = p->pMethods->xRead(p, zData, nByte, 0); 104 if( rc!=SQLITE_OK ){ 105 goto close_out; 106 } 107 zData[nByte] = '\0'; 108 109 p->pMethods->xClose(p); 110 free(p); 111 *pnScript = nByte; 112 return zData; 113 114 close_out: 115 p->pMethods->xClose(p); 116 117 error_out: 118 free(p); 119 free(zData); 120 return 0; 121 } 122 123 int main(int argc, char **argv){ 124 125 const char zUsageMsg[] = 126 "Usage: %s options...\n" 127 " where available options are:\n" 128 "\n" 129 " -db DATABASE-FILE (database file to operate on)\n" 130 " -script SCRIPT-FILE (script file to read sql from)\n" 131 " -log LOG-FILE (log file to create)\n" 132 " -logdata (log all data to log file)\n" 133 "\n" 134 " Options -db, -script and -log are compulsory\n" 135 "\n" 136 ; 137 138 const char *zDb = 0; 139 const char *zScript = 0; 140 const char *zLog = 0; 141 int logdata = 0; 142 143 int ii; 144 int i, j; 145 int rc; 146 147 sqlite3_vfs *pInstVfs; /* Instrumentation VFS */ 148 149 char *zSql = 0; 150 int nSql; 151 152 sqlite3 *db; 153 154 for(ii=1; ii<argc; ii++){ 155 if( stringcompare("-db", argv[ii]) && (ii+1)<argc ){ 156 zDb = argv[++ii]; 157 } 158 159 else if( stringcompare("-script", argv[ii]) && (ii+1)<argc ){ 160 zScript = argv[++ii]; 161 } 162 163 else if( stringcompare("-log", argv[ii]) && (ii+1)<argc ){ 164 zLog = argv[++ii]; 165 } 166 167 else if( stringcompare("-logdata", argv[ii]) ){ 168 logdata = 1; 169 } 170 171 else { 172 goto usage; 173 } 174 } 175 if( !zDb || !zScript || !zLog ) goto usage; 176 177 zSql = readScriptFile(zScript, &nSql); 178 if( !zSql ){ 179 fprintf(stderr, "Failed to read script file\n"); 180 return -1; 181 } 182 183 pInstVfs = sqlite3_instvfs_binarylog("logging", 0, zLog, logdata); 184 185 rc = sqlite3_open_v2( 186 zDb, &db, SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE, "logging" 187 ); 188 if( rc!=SQLITE_OK ){ 189 fprintf(stderr, "Failed to open db: %s\n", sqlite3_errmsg(db)); 190 return -2; 191 } 192 193 for(i=j=0; j<nSql; j++){ 194 if( zSql[j]==';' ){ 195 int isComplete; 196 char c = zSql[j+1]; 197 zSql[j+1] = 0; 198 isComplete = sqlite3_complete(&zSql[i]); 199 zSql[j+1] = c; 200 if( isComplete ){ 201 zSql[j] = 0; 202 while( i<j && ISSPACE(zSql[i]) ){ i++; } 203 if( i<j ){ 204 prepareAndRun(pInstVfs, db, &zSql[i]); 205 } 206 zSql[j] = ';'; 207 i = j+1; 208 } 209 } 210 } 211 212 sqlite3_instvfs_destroy(pInstVfs); 213 return 0; 214 215 usage: 216 fprintf(stderr, zUsageMsg, argv[0]); 217 return -3; 218 } 219