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 */
prepareAndRun(sqlite3_vfs * pInstVfs,sqlite3 * db,const char * zSql)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
stringcompare(const char * zLeft,const char * zRight)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
readScriptFile(const char * zFile,int * pnScript)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
main(int argc,char ** argv)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