1 /*
2 ** Performance test for SQLite.
3 **
4 ** This program reads ASCII text from a file named on the command-line.
5 ** It converts each SQL statement into UTF16 and submits it to SQLite
6 ** for evaluation. A new UTF16 database is created at the beginning of
7 ** the program. All statements are timed using the high-resolution timer
8 ** built into Intel-class processors.
9 **
10 ** To compile this program, first compile the SQLite library separately
11 ** will full optimizations. For example:
12 **
13 ** gcc -c -O6 -DSQLITE_THREADSAFE=0 sqlite3.c
14 **
15 ** Then link against this program. But to do optimize this program
16 ** because that defeats the hi-res timer.
17 **
18 ** gcc speedtest16.c sqlite3.o -ldl -I../src
19 **
20 ** Then run this program with a single argument which is the name of
21 ** a file containing SQL script that you want to test:
22 **
23 ** ./a.out database.db test.sql
24 */
25 #include <stdio.h>
26 #include <string.h>
27 #include <stdlib.h>
28 #include <ctype.h>
29 #include <unistd.h>
30 #include "sqlite3.h"
31
32 #define ISSPACE(X) isspace((unsigned char)(X))
33
34 /*
35 ** hwtime.h contains inline assembler code for implementing
36 ** high-performance timing routines.
37 */
38 #include "hwtime.h"
39
40 /*
41 ** Convert a zero-terminated ASCII string into a zero-terminated
42 ** UTF-16le string. Memory to hold the returned string comes
43 ** from malloc() and should be freed by the caller.
44 */
asciiToUtf16le(const char * z)45 static void *asciiToUtf16le(const char *z){
46 int n = strlen(z);
47 char *z16;
48 int i, j;
49
50 z16 = malloc( n*2 + 2 );
51 for(i=j=0; i<=n; i++){
52 z16[j++] = z[i];
53 z16[j++] = 0;
54 }
55 return (void*)z16;
56 }
57
58 /*
59 ** Timers
60 */
61 static sqlite_uint64 prepTime = 0;
62 static sqlite_uint64 runTime = 0;
63 static sqlite_uint64 finalizeTime = 0;
64
65 /*
66 ** Prepare and run a single statement of SQL.
67 */
prepareAndRun(sqlite3 * db,const char * zSql)68 static void prepareAndRun(sqlite3 *db, const char *zSql){
69 void *utf16;
70 sqlite3_stmt *pStmt;
71 const void *stmtTail;
72 sqlite_uint64 iStart, iElapse;
73 int rc;
74
75 printf("****************************************************************\n");
76 printf("SQL statement: [%s]\n", zSql);
77 utf16 = asciiToUtf16le(zSql);
78 iStart = sqlite3Hwtime();
79 rc = sqlite3_prepare16_v2(db, utf16, -1, &pStmt, &stmtTail);
80 iElapse = sqlite3Hwtime() - iStart;
81 prepTime += iElapse;
82 printf("sqlite3_prepare16_v2() returns %d in %llu cycles\n", rc, iElapse);
83 if( rc==SQLITE_OK ){
84 int nRow = 0;
85 iStart = sqlite3Hwtime();
86 while( (rc=sqlite3_step(pStmt))==SQLITE_ROW ){ nRow++; }
87 iElapse = sqlite3Hwtime() - iStart;
88 runTime += iElapse;
89 printf("sqlite3_step() returns %d after %d rows in %llu cycles\n",
90 rc, nRow, iElapse);
91 iStart = sqlite3Hwtime();
92 rc = sqlite3_finalize(pStmt);
93 iElapse = sqlite3Hwtime() - iStart;
94 finalizeTime += iElapse;
95 printf("sqlite3_finalize() returns %d in %llu cycles\n", rc, iElapse);
96 }
97 free(utf16);
98 }
99
main(int argc,char ** argv)100 int main(int argc, char **argv){
101 void *utf16;
102 sqlite3 *db;
103 int rc;
104 int nSql;
105 char *zSql;
106 int i, j;
107 FILE *in;
108 sqlite_uint64 iStart, iElapse;
109 sqlite_uint64 iSetup = 0;
110 int nStmt = 0;
111 int nByte = 0;
112
113 if( argc!=3 ){
114 fprintf(stderr, "Usage: %s FILENAME SQL-SCRIPT\n"
115 "Runs SQL-SCRIPT as UTF16 against a UTF16 database\n",
116 argv[0]);
117 exit(1);
118 }
119 in = fopen(argv[2], "r");
120 fseek(in, 0L, SEEK_END);
121 nSql = ftell(in);
122 zSql = malloc( nSql+1 );
123 fseek(in, 0L, SEEK_SET);
124 nSql = fread(zSql, 1, nSql, in);
125 zSql[nSql] = 0;
126
127 printf("SQLite version: %d\n", sqlite3_libversion_number());
128 unlink(argv[1]);
129 utf16 = asciiToUtf16le(argv[1]);
130 iStart = sqlite3Hwtime();
131 rc = sqlite3_open16(utf16, &db);
132 iElapse = sqlite3Hwtime() - iStart;
133 iSetup = iElapse;
134 printf("sqlite3_open16() returns %d in %llu cycles\n", rc, iElapse);
135 free(utf16);
136 for(i=j=0; j<nSql; j++){
137 if( zSql[j]==';' ){
138 int isComplete;
139 char c = zSql[j+1];
140 zSql[j+1] = 0;
141 isComplete = sqlite3_complete(&zSql[i]);
142 zSql[j+1] = c;
143 if( isComplete ){
144 zSql[j] = 0;
145 while( i<j && ISSPACE(zSql[i]) ){ i++; }
146 if( i<j ){
147 nStmt++;
148 nByte += j-i;
149 prepareAndRun(db, &zSql[i]);
150 }
151 zSql[j] = ';';
152 i = j+1;
153 }
154 }
155 }
156 iStart = sqlite3Hwtime();
157 sqlite3_close(db);
158 iElapse = sqlite3Hwtime() - iStart;
159 iSetup += iElapse;
160 printf("sqlite3_close() returns in %llu cycles\n", iElapse);
161 printf("\n");
162 printf("Statements run: %15d\n", nStmt);
163 printf("Bytes of SQL text: %15d\n", nByte);
164 printf("Total prepare time: %15llu cycles\n", prepTime);
165 printf("Total run time: %15llu cycles\n", runTime);
166 printf("Total finalize time: %15llu cycles\n", finalizeTime);
167 printf("Open/Close time: %15llu cycles\n", iSetup);
168 printf("Total Time: %15llu cycles\n",
169 prepTime + runTime + finalizeTime + iSetup);
170 return 0;
171 }
172