| 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 #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 } | |
| OLD | NEW |