OLD | NEW |
(Empty) | |
| 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 } |
OLD | NEW |