pthreads version of fts5 bug code

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <pthread.h>
#include <unistd.h>
#include "sqlite3.h"

char *orgName[] = {
  "yun j ji yunji yunj yji yj z zh zha zhao q qi qia qian d du dua duan zhaoqian zhaoq zhqian zhq zqian zq qianduan qiand qduan qd zhaoqianduan zhaoqiand zhaoqduan zhaoqd zhqianduan zhqiand zhqduan zhqd zqianduan zqiand zqduan zqd",
  "murren m mu l lu lun mulun mul mlun ml",
  "l lu m ma man luman lum lman lm",
  "d   di   din   ding   d   di   din   ding   f   fu   w   wu   dingding   dingd   dding   dd   dingfu   dingf   dfu   df   fuwu   fuw   fwu   fw   dingdingfu   dingdingf   dingdfu   dingdf   ddingfu   ddingf   ddfu   ddf   dingfuwu   dingfuw   dingfwu   dingfw   dfuwu   dfuw   dfwu   dfw   dingdingfuwu   dingdingfuw   dingdingfwu   dingdingfw   dingdfuwu   dingdfuw   dingdfwu   dingdfw   ddingfuwu   ddingfuw   ddingfwu   ddingfw   ddfuwu   ddfuw   ddfwu   ddfw   l   li   x   xi   xia   lixia   lix   lxia   lx",
  "lucy y ya yan x xi yanxi yanx yxi yx",
  "p pe pen peng l le lei penglei pengl plei pl lucy"
};
char *nickName[] = {
  "ab",
  "bc",
  "cd",
  "de",
  "ef",
  "fg"
};
int id[] = { 21078,21218,21125,53234,40824,164873 };

void prepareSchema_prefix(sqlite3 *db) {
  char *schemas[] = {
    "PRAGMA journal_mode=WAL;",
    "drop table if exists tbl;",
    "drop table if exists tbl_fts;",
    "create table if not exists tbl(name text, uid int primary key, nick text);",
    "create virtual table if not exists tbl_fts USING fts5(name, uid);",
    "create trigger if not exists trigger_insert_tbl after insert on tbl \
      begin \
      replace into tbl_fts(rowid, name, uid) values(new.rowid,new.name,new.uid); \
      end;",
    "create trigger if not exists trigger_update_tbl after update on tbl \
      begin \
      replace into tbl_fts(rowid, name, uid) values(new.rowid,new.name,new.uid);\
      end;"
  };
  int i, j, rc, cnt;
  cnt = sizeof(schemas) / sizeof(char*);
  char *errMsg;
  for (i = 0; i < cnt; i++) {
    rc = sqlite3_exec(db, schemas[i], NULL, NULL, &errMsg);
    if (rc != SQLITE_OK&&rc != SQLITE_DONE) {
      printf("prepareSchema error: %s\n", errMsg);
      return;
    }
  }
  printf("prepareSchema over\n");
}

/*
** Return a pseudo-random integer between 0 and (nMax-1), inclusive.
*/
int randomint(int nMax){
  unsigned int x;
  sqlite3_randomness(sizeof(unsigned int), (void*)&x);
  return (int)(x % nMax);
}

void fts5_prefix_match_test() {
  sqlite3 *db;
  int i, j, k, rc;
  char *errMsg;
  char *append = "abcdefghijklmnopqrstuvwxyz";
  sqlite3_open("a.db", &db);
  prepareSchema_prefix(db);
  int test_cnt = 300, rc_copy = 0;
  char zSql[2048];
  char *tmp;
  const int len_zSql = 2047;
  int len_left;
  int nRow = 0, nCol = 0;
  char **pazResult;
  const int trx_cnt = 10000;
  int trx_idx;
  for (trx_idx = 0; trx_idx < 10000; trx_idx++) {
    for (i = 0; i < test_cnt; i++) {
      j = randomint(6);
      len_left = len_zSql;
      tmp = zSql;
      if (j < 4) {
        int uid = id[j] + randomint(300);
        rc_copy = snprintf(tmp, len_left, "insert into tbl values(\"%s%c\",%d,\"%s%c\");",
            orgName[j], append[randomint(26)], uid, nickName[j], append[randomint(26)]);
        //printf("%s\n", zSql);
        rc = sqlite3_exec(db, zSql, NULL, NULL, &errMsg);
        if (rc != SQLITE_OK&&rc != SQLITE_DONE) {
          if (rc != SQLITE_CONSTRAINT) {
            printf("error %s: %s\n", zSql, errMsg);
            continue;
          }
          else {
            rc_copy = snprintf(tmp, len_left, "update tbl set name=\"%s%c\" where uid=%d\n",
                orgName[j], append[randomint(26)], uid);
            rc = sqlite3_exec(db, zSql, NULL, NULL, &errMsg);
            if (rc != SQLITE_OK&&rc != SQLITE_DONE) {
              printf("error %s: %s\n", zSql, errMsg);
              continue;
            }
          }
        }
      }
      else {
        rc_copy = snprintf(zSql, len_zSql, "select * from tbl where uid=%d;", id[j]);
        rc = sqlite3_get_table(db, zSql, &pazResult, &nRow, &nCol, &errMsg);
        if (nRow == 0) {
          rc_copy = snprintf(tmp, len_left, "insert into tbl(name, uid, nick) values(\"%s\",%d,\"%s\");",
              orgName[j], id[j], nickName[j]);
        }
        else {
          rc_copy = snprintf(zSql, len_zSql, "update tbl set name=\"%s\" where uid=%d;",
              orgName[j], id[j]);
        }
        rc = sqlite3_exec(db, zSql, NULL, NULL, &errMsg);
        if (rc != SQLITE_OK&&rc != SQLITE_DONE) {
          printf("error %s: %s\n", zSql, errMsg);
          continue;
        }
      }
    }
    printf("insert 300 records\n");
    sqlite3_sleep(200);
  }
}

static void *writeFunc(void *pIn) {
  fts5_prefix_match_test();
  printf("write over\n");
  return 0;
}

static void *readFunc(void *pIn) {
  sqlite3 *db;
  int i, j, rc;
  char *errMsg;
  char **pazResult;
  int nRow, nCol;
  sqlite3_sleep(2000);
  rc = sqlite3_open("a.db", &db);
  if (rc != SQLITE_OK) {
    printf("open db error\n");
    return 0;
  }
  char *zSql = "select rowid, * from tbl_fts where tbl_fts match \'lucy*\';";
  while (1) {
    rc = sqlite3_get_table(db, zSql, &pazResult, &nRow, &nCol, &errMsg);
    if (nRow > 2) {
      printf("mis-match\n");
      for (i = 0; i <= nRow; i++) {
        for (j = 0; j < nCol; j++) {
          printf("%s\t", pazResult[i*nCol + j]);
        }
        printf("\n");
      }
      return 0;
    }
    else {
      printf("match cnt = %d\n", nRow);
    }
    sqlite3_free_table(pazResult);
    sqlite3_sleep(3000);
  }
  return 0;
}
#define THREAD_NUM 2

typedef void*(*ThreadFuncType)(void*);

int thread(ThreadFuncType writeFunc, ThreadFuncType readFunc) {
  pthread_t handle[THREAD_NUM];
  memset(handle, 0, sizeof(handle));

  if (writeFunc) {
    pthread_create(&handle[0], 0, writeFunc, 0);
  }
  if (readFunc) {
    for (int i = 1; i < THREAD_NUM; i++) {
      pthread_create(&handle[1], 0, readFunc, 0);
      sleep(5);
    }
    if (writeFunc) {
      pthread_join(handle[0], 0);
    }
    else {
      pthread_join(handle[1], 0);
    }
    printf("multithread are all over now.\n");
  }
  else {
    pthread_join(handle[0], 0);
    printf("multithread are all over now.\n");
  }
  return 0;
}

int main() {
  thread(writeFunc, readFunc);
  return 0;
}