clean up sqlite test

1 files changed, 82 insertions, 0 deletions

diff --git a/tests/sqlite/benchmark.c b/tests/sqlite/benchmark.c
new file mode 100644
index 00000000..de8a9a82
--- /dev/null
+++ b/tests/sqlite/benchmark.c
@@ -0,0 +1,82 @@
+#include <time.h>
+#include <stdio.h>
+#include <sqlite3.h>
+
+static int callback(void *NotUsed, int argc, char **argv, char **azColName){
+  int i;
+  for(i=0; i<argc; i++){
+    printf("%s = %s\n", azColName[i], argv[i] ? argv[i] : "NULL");
+  }
+  printf("\n");
+  return 0;
+}
+
+int main(){
+  sqlite3 *db;
+  char *zErrMsg = 0;
+  int rc, i;
+  clock_t t;
+
+  rc = sqlite3_open(":memory:", &db);
+  if( rc ){
+    fprintf(stderr, "Can't open database: %s\n", sqlite3_errmsg(db));
+    sqlite3_close(db);
+    exit(1);
+  }
+
+  #define RUN(cmd) \
+    { \
+      rc = sqlite3_exec(db, cmd, callback, 0, &zErrMsg); \
+      if( rc!=SQLITE_OK ){ \
+        fprintf(stderr, "SQL error on %d: %s\n", i, zErrMsg); \
+        sqlite3_free(zErrMsg); \
+        exit(1); \
+      } \
+    }
+
+  #define TIME(msg) \
+    { \
+      printf(msg " : took %d ms\n", (clock()-t)/1000); \
+      t = clock(); \
+    }
+
+  t = clock();
+
+  RUN("CREATE TABLE t1(a INTEGER, b INTEGER, c VARCHAR(100));");
+  TIME("create table");
+
+  RUN("BEGIN;");
+
+  // 25000 INSERTs in a transaction
+  for (i = 0; i < 5000; i++) {
+    RUN("INSERT INTO t1 VALUES(1,12345,'one 1 one 1 one 1');");
+    RUN("INSERT INTO t1 VALUES(2,23422,'two two two two');");
+    RUN("INSERT INTO t1 VALUES(3,31233,'three three 33333333333 three');");
+    RUN("INSERT INTO t1 VALUES(4,41414,'FOUR four 4 phor FOUR 44444');");
+    RUN("INSERT INTO t1 VALUES(5,52555,'five 5 FIVE Five phayve 55 5 5 5 5 55  5');");
+  }
+  TIME("25,000 inserts");
+
+  RUN("COMMIT;");
+  TIME("commit");
+
+  // Counts
+  RUN("SELECT count(*) FROM t1;");
+  RUN("SELECT count(*) FROM t1 WHERE a == 4");
+  RUN("SELECT count(*) FROM t1 WHERE b > 20000 AND b < 50000;");
+  RUN("SELECT count(*) FROM t1 WHERE c like '%three%';");
+  TIME("selects");
+
+  // Index
+  RUN("CREATE INDEX iiaa ON t1(a);");
+  RUN("CREATE INDEX iibb ON t1(b);");
+  TIME("create indexes");
+
+  RUN("SELECT count(*) FROM t1 WHERE a == 4");
+  RUN("SELECT count(*) FROM t1 WHERE b > 20000 AND b < 50000;");
+  TIME("selects with indexes");
+
+  sqlite3_close(db);
+  return 0;
+}
+