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
|
#include "catch.hpp"
#include "duckdb/common/file_system.hpp"
#include "duckdb.hpp"
#include "test_helpers.hpp"
#include <signal.h>
#include <sys/mman.h>
#include <unistd.h>
using namespace duckdb;
using namespace std;
#define BOOL_COUNT 3
TEST_CASE("Test write lock with multiple processes", "[persistence][.]") {
uint64_t *count =
(uint64_t *)mmap(NULL, sizeof(uint64_t), PROT_READ | PROT_WRITE, MAP_ANONYMOUS | MAP_SHARED, 0, 0);
*count = 0;
string dbdir = TestCreatePath("writelocktest");
DeleteDatabase(dbdir);
// test write lock
// fork away a child
pid_t pid = fork();
if (pid == 0) {
// child process
// open db for writing
DuckDB db(dbdir);
Connection con(db);
// opened db for writing
// insert some values
(*count)++;
con.Query("CREATE TABLE a(i INTEGER)");
con.Query("INSERT INTO a VALUES(42)");
while (true) {
con.Query("SELECT * FROM a");
usleep(100);
}
} else if (pid > 0) {
duckdb::unique_ptr<DuckDB> db;
// parent process
// sleep a bit to wait for child process
while (*count == 0) {
usleep(100);
}
// try to open db for writing, this should fail
REQUIRE_THROWS(db = make_uniq<DuckDB>(dbdir));
// kill the child
if (kill(pid, SIGKILL) != 0) {
FAIL();
}
}
}
TEST_CASE("Test read lock with multiple processes", "[persistence][.]") {
uint64_t *count =
(uint64_t *)mmap(NULL, sizeof(uint64_t), PROT_READ | PROT_WRITE, MAP_ANONYMOUS | MAP_SHARED, 0, 0);
*count = 0;
string dbdir = TestCreatePath("readlocktest");
DeleteDatabase(dbdir);
// create the database
{
DuckDB db(dbdir);
Connection con(db);
REQUIRE_NO_FAIL(con.Query("CREATE TABLE a(i INTEGER)"));
REQUIRE_NO_FAIL(con.Query("INSERT INTO a VALUES (42)"));
}
// test read lock
pid_t pid = fork();
DBConfig config;
config.options.access_mode = AccessMode::READ_ONLY;
if (pid == 0) {
// child process
// open db for reading
DuckDB db(dbdir, &config);
Connection con(db);
(*count)++;
// query some values
con.Query("SELECT i+2 FROM a");
while (true) {
usleep(100);
con.Query("SELECT * FROM a");
}
} else if (pid > 0) {
duckdb::unique_ptr<DuckDB> db;
// parent process
// sleep a bit to wait for child process
while (*count == 0) {
usleep(100);
}
// try to open db for writing, this should fail
REQUIRE_THROWS(db = make_uniq<DuckDB>(dbdir));
// but opening db for reading should work
REQUIRE_NOTHROW(db = make_uniq<DuckDB>(dbdir, &config));
// we can query the database
Connection con(*db);
REQUIRE_NO_FAIL(con.Query("SELECT * FROM a"));
// kill the child
if (kill(pid, SIGKILL) != 0) {
FAIL();
}
}
}
|
