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
|
#ifndef THREAD_CONTEXT_H
#define THREAD_CONTEXT_H
#include <thread>
#include "wiredtiger.h"
namespace test_harness {
/* Define the different thread operations. */
enum class thread_operation { INSERT, UPDATE, READ, REMOVE, CHECKPOINT, TIMESTAMP, MONITOR };
/* Container class for a thread and any data types it may need to interact with the database. */
class thread_context {
public:
thread_context(
WT_SESSION *session, std::vector<std::string> collection_names, thread_operation type)
: _collection_names(collection_names), _running(false), _session(session), _thread(nullptr),
_type(type)
{
}
thread_context(thread_operation type)
: _running(false), _session(nullptr), _thread(nullptr), _type(type)
{
}
~thread_context()
{
if (_session != nullptr)
testutil_die(DEBUG_ABORT,
"Session should've been cleaned up already. Did you forget to"
"call finish on the thread_manager?");
delete _thread;
_thread = nullptr;
}
/* Cleanup state. */
void
finish()
{
_running = false;
if (_thread != nullptr && _thread->joinable())
_thread->join();
if (_session != nullptr) {
if (_session->close(_session, NULL) != 0)
debug_info("Failed to close session for current thread", _trace_level, DEBUG_ERROR);
_session = nullptr;
}
}
const std::vector<std::string> &
get_collection_names() const
{
return _collection_names;
}
WT_SESSION *
get_session() const
{
return _session;
}
thread_operation
get_thread_operation() const
{
return _type;
}
bool
is_running() const
{
return _running;
}
void
set_thread(std::thread *thread)
{
_thread = thread;
}
void
set_running(bool running)
{
_running = running;
}
private:
const std::vector<std::string> _collection_names;
bool _running;
WT_SESSION *_session;
std::thread *_thread;
const thread_operation _type;
};
} // namespace test_harness
#endif
|
