ramcloud-sqlite3/test2.c at fc6d17bd5ead82f10ebaa24b187dc1f121dbf28a · jblomer/ramcloud-sqlite3 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
#include "sqlite3.h"
#include "vfs-ramcloud.h"

#include <inttypes.h>
#include <assert.h>
#include <stdio.h>
#include <pthread.h>

#define ROWS 2
#define THREADS 10


char *dbname;

static void *mainThread(void *data) {
  uint64_t threadno = (uint64_t)(data);
  int retval;
  sqlite3 *db;
  retval = sqlite3_open_v2(dbname, &db, SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE, NULL);
  assert(retval == 0);

  unsigned i;
  sqlite3_stmt *sql_insert;
  do {
    retval = sqlite3_prepare(db, "INSERT INTO test (value) VALUES (:v)", -1, &sql_insert, NULL);
  } while (retval == SQLITE_BUSY);
  assert(retval == 0);
  for (i = 0; i < ROWS; ++i) {
    do {
      retval = sqlite3_exec(db, "BEGIN;", NULL, NULL, NULL);
    } while (retval == SQLITE_BUSY);
    printf("[%lu] begin transaction (%u)\n", threadno, i);
    //printf("retval is %d (error %s)\n", retval, sqlite3_errmsg(db));
    assert(retval == 0);
    printf("[%lu] prepare stmt (%u)\n", threadno, i);
    retval = sqlite3_bind_int64(sql_insert, 1, i);
    assert(retval == 0);
    do {
      retval = sqlite3_step(sql_insert);
    } while (retval == SQLITE_BUSY);
    assert(retval == SQLITE_DONE);
    printf("[%lu] step (%u)\n", threadno, i);
    retval = sqlite3_reset(sql_insert);
    assert(retval == 0);
    do {
      retval = sqlite3_exec(db, "COMMIT;", NULL, NULL, NULL);
    } while (retval == SQLITE_BUSY);
    printf("[%lu] COMMIT retval is %d (error %s) (i is %d)\n", threadno, retval, sqlite3_errmsg(db), i);
    assert(retval == 0);
  }
  retval = sqlite3_finalize(sql_insert);
  assert(retval == 0);

  retval = sqlite3_close(db);
  printf("[%lu] CLOSE retval is %d (error %s) (i is %d)\n", threadno, retval, sqlite3_errmsg(db), i);
  assert(retval == 0);

  return NULL;
}


int main(int argc, char **argv) {
  int retval;

  if (argc < 3) {
    printf("usage: %s <ramcloud locator> <db name>\n", argv[0]);
    return 1;
  }

  char *locator = argv[1];
  dbname = argv[2];
  //sqlite3_vfs *vfs = sqlite3_vfs_find("unix");
  //assert(vfs != NULL);
  //retval = sqlite3_vfs_register(vfs, 1);
  SQLITE_RCVFS_CONNECTION *conn =
    sqlite3_rcvfs_connect(locator, "main", "sqlite3");
  assert(conn != NULL);
  retval = sqlite3_vfs_register(sqlite3_rcvfs("ramcloud", conn), 1);
  assert(retval == 0);

  sqlite3 *db;
  retval = sqlite3_open_v2(dbname, &db, SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE, NULL);
  assert(retval == 0);
  printf("file opened\n");

  do {
    retval = sqlite3_exec(db, "CREATE TABLE IF NOT EXISTS test (value INTEGER);", NULL, NULL, NULL);
  } while (retval == SQLITE_BUSY);
  printf("retval is %d\n", retval);
  assert(retval == 0);

  pthread_t threads[THREADS];
  uint64_t i;
  for (i = 0; i < THREADS; ++i) {
    retval = pthread_create(&threads[i], NULL, mainThread, (void *)i);
    assert(retval == 0);
  }
  for (i = 0; i < THREADS; ++i) {
    retval = pthread_join(threads[i], NULL);
    assert(retval == 0);
  }

  retval = sqlite3_close(db);
  assert(retval == 0);

  retval = sqlite3_open_v2(dbname, &db, SQLITE_OPEN_READONLY, NULL);
  assert(retval == 0);

  sqlite3_stmt *sql_sum;
  retval = sqlite3_prepare(db, "SELECT SUM(value) FROM test;", -1, &sql_sum, NULL);
  assert(retval == 0);
  retval = sqlite3_step(sql_sum);
  assert(retval = SQLITE_ROW);
  uint64_t sum = sqlite3_column_int64(sql_sum, 0);
  retval = sqlite3_reset(sql_sum);
  assert(retval == 0);
  retval = sqlite3_finalize(sql_sum);
  assert(retval == 0);

  retval = sqlite3_close(db);
  assert(retval == 0);

  sqlite3_rcvfs_disconnect(conn);

  printf("sum is %lu\n", sum);
  return 0;
}