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
|
#include "include/rados/librados.h"
#include "include/rados/librados.hpp"
#include "include/utime.h"
#include "common/Thread.h"
#include "common/Clock.h"
#include "test/librados/test_cxx.h"
#include "gtest/gtest.h"
#include <semaphore.h>
#include <errno.h>
#include <map>
#include <sstream>
#include <iostream>
#include <string>
#include <atomic>
#include "test/librados/testcase_cxx.h"
using namespace librados;
using std::map;
using std::ostringstream;
using std::string;
static sem_t *sem;
static std::atomic<bool> stop_flag = { false };
class WatchNotifyTestCtx : public WatchCtx
{
public:
void notify(uint8_t opcode, uint64_t ver, bufferlist& bl) override
{
sem_post(sem);
}
};
#pragma GCC diagnostic ignored "-Wpragmas"
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wdeprecated-declarations"
struct WatcherUnwatcher : public Thread {
string pool;
explicit WatcherUnwatcher(string& _pool) : pool(_pool) {}
void *entry() override {
Rados cluster;
connect_cluster_pp(cluster);
while (!stop_flag) {
IoCtx ioctx;
cluster.ioctx_create(pool.c_str(), ioctx);
uint64_t handle;
WatchNotifyTestCtx watch_ctx;
int r = ioctx.watch("foo", 0, &handle, &watch_ctx);
if (r == 0)
ioctx.unwatch("foo", handle);
ioctx.close();
}
return NULL;
}
};
typedef RadosTestParamPP WatchStress;
INSTANTIATE_TEST_CASE_P(WatchStressTests, WatchStress,
::testing::Values("", "cache"));
TEST_P(WatchStress, Stress1) {
ASSERT_NE(SEM_FAILED, (sem = sem_open("test_stress_watch", O_CREAT, 0644, 0)));
Rados ncluster;
std::string pool_name = get_temp_pool_name();
ASSERT_EQ("", create_one_pool_pp(pool_name, ncluster));
IoCtx nioctx;
ncluster.ioctx_create(pool_name.c_str(), nioctx);
WatchNotifyTestCtx ctx;
WatcherUnwatcher *thr = new WatcherUnwatcher(pool_name);
thr->create("watcher_unwatch");
ASSERT_EQ(0, nioctx.create("foo", false));
for (unsigned i = 0; i < 75; ++i) {
std::cerr << "Iteration " << i << std::endl;
uint64_t handle;
Rados cluster;
IoCtx ioctx;
WatchNotifyTestCtx ctx;
connect_cluster_pp(cluster);
cluster.ioctx_create(pool_name.c_str(), ioctx);
ASSERT_EQ(0, ioctx.watch("foo", 0, &handle, &ctx));
bool do_blacklist = i % 2;
if (do_blacklist) {
cluster.test_blacklist_self(true);
std::cerr << "blacklisted" << std::endl;
sleep(1);
}
bufferlist bl2;
ASSERT_EQ(0, nioctx.notify("foo", 0, bl2));
if (do_blacklist) {
sleep(1); // Give a change to see an incorrect notify
} else {
TestAlarm alarm;
sem_wait(sem);
}
if (do_blacklist) {
cluster.test_blacklist_self(false);
}
ioctx.unwatch("foo", handle);
ioctx.close();
}
stop_flag = true;
thr->join();
nioctx.close();
ASSERT_EQ(0, destroy_one_pool_pp(pool_name, ncluster));
sem_close(sem);
}
#pragma GCC diagnostic pop
#pragma GCC diagnostic warning "-Wpragmas"
|
