// sqltctest.c
//
// SQLiteとTokyoCabinetの簡易性能テスト
//
// gcc -std=c99 -I/usr/local/include -D_GNU_SOURCE=1 -Wall -O3 -o sqltcttest sqltcttest.c -L/usr/local/lib -lsqlite3 -ltokyocabinet -lpthread -ldl -lc


#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/time.h>
#include <assert.h>
#include <sqlite3.h>
#include <tctdb.h>

#define SQLFILE    "casket.sql"
#define TCTFILE    "casket.tct"
#define SQLBUFSIZ   1024
#define SAMPLENUM   10
#define RECORDNUM   1000000

int main(int argc, char **argv);
static void test_sqlite(void);
static void test_tctdb(void);
static double gettime(void);
static int myrand(void);
static int callback(void *opq, int argc, char **argv, char **colname);


// メインルーチン
int main(int argc, char **argv){
  test_sqlite();
  test_tctdb();
  return 0;
}


// SQLiteのテスト
static void test_sqlite(void){

  // レコード格納のテスト
  double sum = 0.0;
  for(int i = 1; i <= SAMPLENUM; i++){
    unlink(SQLFILE);
    sync(); sync();
    printf("SQLite write sample:%d ... ", i);
    fflush(stdout);
    double stime = gettime();

    // データベースを開く
    sqlite3 *db;
    if(sqlite3_open(SQLFILE, &db) != 0) assert(!__LINE__);

    // スキーマとインデックスを設定する
    char sql[SQLBUFSIZ], *errmsg;
    sprintf(sql, "CREATE TABLE tbl ( k TEXT PRIMARY KEY, a TEXT, b INTEGER,"
            " c TEXT, d INTEGER );");
    if(sqlite3_exec(db, sql, callback, 0, &errmsg) != SQLITE_OK) assert(!__LINE__);
    sprintf(sql, "CREATE INDEX tbl_s ON tbl ( a );");
    if(sqlite3_exec(db, sql, callback, 0, &errmsg) != SQLITE_OK) assert(!__LINE__);
    sprintf(sql, "CREATE INDEX tbl_n ON tbl ( b );");
    if(sqlite3_exec(db, sql, callback, 0, &errmsg) != SQLITE_OK) assert(!__LINE__);

    // キャッシュサイズを設定する
    sprintf(sql, "PRAGMA cache_size = %d;", RECORDNUM);
    if(sqlite3_exec(db, sql, callback, 0, &errmsg) != SQLITE_OK) assert(!__LINE__);

    // トランザクションを開始する
    sprintf(sql, "BEGIN TRANSACTION;");
    if(sqlite3_exec(db, sql, callback, 0, &errmsg) != SQLITE_OK) assert(!__LINE__);

    // レコードを格納する
    for(int j = 1; j <= RECORDNUM; j++){
      sprintf(sql, "INSERT INTO tbl VALUES ( '%08d', '%08d', %d, '%08d', %d );",
	    j, j, myrand() % RECORDNUM, myrand() % RECORDNUM, j);
      if(sqlite3_exec(db, sql, callback, 0, &errmsg) != SQLITE_OK) assert(!__LINE__);
    }

    // トランザクションを終了する
    sprintf(sql, "END TRANSACTION;");
    if(sqlite3_exec(db, sql, callback, 0, &errmsg) != SQLITE_OK) assert(!__LINE__);

    // データベースを閉じる
    sqlite3_close(db);

    double etime = gettime() - stime;
    printf("%.6f\n", etime);
    sum += etime;
  }
  printf("SQLite write average: %.6f\n", sum / SAMPLENUM);

  // レコード検索のテスト
  sum = 0.0;
  for(int i = 1; i <= SAMPLENUM; i++){
    sync(); sync();
    printf("SQLite read sample:%d ... ", i);
    fflush(stdout);
    double stime = gettime();

    // データベースを開く
    sqlite3 *db;
    if(sqlite3_open(SQLFILE, &db) != 0) assert(!__LINE__);

    // キャッシュサイズを設定する
    char sql[SQLBUFSIZ], *errmsg;
    sprintf(sql, "PRAGMA cache_size = %d;", RECORDNUM);
    if(sqlite3_exec(db, sql, callback, 0, &errmsg) != SQLITE_OK) assert(!__LINE__);

    // レコードを検索する
    for(int j = 1; j <= RECORDNUM; j++){
      sprintf(sql, "SELECT * FROM tbl WHERE a = '%08d';", myrand() % RECORDNUM + 1);
      if(sqlite3_exec(db, sql, callback, 0, &errmsg) != SQLITE_OK) assert(!__LINE__);
    }

    // データベースを閉じる
    sqlite3_close(db);

    double etime = gettime() - stime;
    printf("%.6f\n", etime);
    sum += etime;
  }
  printf("SQLite read average: %.6f\n", sum / SAMPLENUM);
}


// TCHDBのテスト
static void test_tctdb(void){

  // レコード格納のテスト
  double sum = 0.0;
  for(int i = 1; i <= SAMPLENUM; i++){
    unlink(TCTFILE);
    sync(); sync();
    printf("TCTDB write sample:%d ... ", i);
    fflush(stdout);
    double stime = gettime();

    // データベースを開く
    TCTDB *tdb = tctdbnew();
    if(!tctdbtune(tdb, RECORDNUM * 2, -1, -1, 0)) assert(!__LINE__);
    if(!tctdbsetcache(tdb, 0, RECORDNUM, RECORDNUM)) assert(!__LINE__);
    if(!tctdbopen(tdb, TCTFILE, TDBOWRITER | TDBOCREAT | TDBOTRUNC)) assert(!__LINE__);

    // スキーマを設定する
    if(!tctdbsetindex(tdb, "a", TDBITLEXICAL)) assert(!__LINE__);
    if(!tctdbsetindex(tdb, "b", TDBITDECIMAL)) assert(!__LINE__);

    // トランザクションを開始する
    if(!tctdbtranbegin(tdb)) assert(!__LINE__);

    // レコードを格納する
    char buf[SQLBUFSIZ];
    int size;
    for(int j = 1; j <= RECORDNUM; j++){
      TCMAP *cols = tcmapnew2(7);
      size = sprintf(buf, "%08d", j);
      tcmapput(cols, "a", 1, buf, size);
      size = sprintf(buf, "%d", myrand() % RECORDNUM);
      tcmapput(cols, "b", 1, buf, size);
      size = sprintf(buf, "%08d", myrand() % RECORDNUM);
      tcmapput(cols, "c", 1, buf, size);
      size = sprintf(buf, "%d", j);
      tcmapput(cols, "d", 1, buf, size);
      size = sprintf(buf, "%08d", j);
      if(!tctdbput(tdb, buf, size, cols)) assert(!__LINE__);
      tcmapdel(cols);
    }

    // トランザクションを終了する
    if(!tctdbtrancommit(tdb)) assert(!__LINE__);

    // データベースを閉じる
    if(!tctdbclose(tdb)) assert(!__LINE__);
    tctdbdel(tdb);

    double etime = gettime() - stime;
    printf("%.6f\n", etime);
    sum += etime;
  }
  printf("TCTDB write average: %.6f\n", sum / SAMPLENUM);

  // レコード検索のテスト
  sum = 0.0;
  for(int i = 1; i <= SAMPLENUM; i++){
    sync(); sync();
    printf("TCTDB read sample:%d ... ", i);
    fflush(stdout);
    double stime = gettime();

    // データベースを開く
    TCTDB *tdb = tctdbnew();
    if(!tctdbsetcache(tdb, 0, RECORDNUM, RECORDNUM)) assert(!__LINE__);
    if(!tctdbopen(tdb, TCTFILE, TDBOREADER)) assert(!__LINE__);

    // レコードを検索する
    char buf[SQLBUFSIZ];
    for(int j = 1; j <= RECORDNUM; j++){
      TDBQRY *qry = tctdbqrynew(tdb);
      sprintf(buf, "%08d", myrand() % RECORDNUM + 1);
      tctdbqryaddcond(qry, "a", TDBQCSTREQ, buf);
      TCLIST *res = tctdbqrysearch(qry);
      for(int k = 0; k < tclistnum(res); k++){
        int ksiz;
        const char *kbuf = tclistval(res, k, &ksiz);
        TCMAP *cols = tctdbget(tdb, kbuf, ksiz);
        if(!cols) assert(!__LINE__);
        tcmapdel(cols);
      }
      tclistdel(res);
      tctdbqrydel(qry);
    }

    // データベースを閉じる
    if(!tctdbclose(tdb)) assert(!__LINE__);

    double etime = gettime() - stime;
    printf("%.6f\n", etime);
    sum += etime;
  }
  printf("TCTDB read average: %.6f\n", sum / SAMPLENUM);
}


// 実時間取得
static double gettime(void){
  struct timeval tv;
  if(gettimeofday(&tv, NULL) == -1) return 0.0;
  return (double)tv.tv_sec + (double)tv.tv_usec / 1000000.0;
}


// 疑似乱数生成
static int myrand(void){
  static int cnt = 0;
  return (lrand48() + cnt++) & 0x7FFFFFFF;
}


// SQLiteの検索結果のイテレータ
static int callback(void *opq, int argc, char **argv, char **colname){
  return 0;
}