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
|
/*-
* Public Domain 2014-present MongoDB, Inc.
* Public Domain 2008-2014 WiredTiger, Inc.
*
* This is free and unencumbered software released into the public domain.
*
* Anyone is free to copy, modify, publish, use, compile, sell, or
* distribute this software, either in source code form or as a compiled
* binary, for any purpose, commercial or non-commercial, and by any
* means.
*
* In jurisdictions that recognize copyright laws, the author or authors
* of this software dedicate any and all copyright interest in the
* software to the public domain. We make this dedication for the benefit
* of the public at large and to the detriment of our heirs and
* successors. We intend this dedication to be an overt act of
* relinquishment in perpetuity of all present and future rights to this
* software under copyright law.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
* IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR
* OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
* ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
* OTHER DEALINGS IN THE SOFTWARE.
*/
#ifndef DATABASE_MODEL_H
#define DATABASE_MODEL_H
#include <atomic>
#include <string>
#include <map>
#include "workload_tracking.h"
/* Forward declarations for classes to reduce compilation time and modules coupling. */
class timestamp_manager;
namespace test_harness {
/* Key/Value type. */
typedef std::string key_value_t;
/* A collection is made of mapped key value objects. */
class collection {
public:
explicit collection(const uint64_t id, const uint64_t key_count, const std::string &name);
/* Copies aren't allowed. */
collection(const collection &) = delete;
collection &operator=(const collection &) = delete;
uint64_t get_key_count() const;
/*
* Adding new keys should generally be singly threaded per collection. If two threads both
* attempt to add keys using the incrementing id pattern they'd frequently conflict.
*
* The usage pattern is:
* 1. Call get_key_count to get the number of keys already existing. Add keys with id's equal
* to and greater than this value.
* 2. Once the transaction has successfully committed then call increase_key_count() with the
* number of added keys.
*
* The set of keys should always be contiguous such that other threads calling get_key_count
* will always know that the keys in existence are 0 -> _key_count - 1.
*/
void increase_key_count(uint64_t increment);
public:
const std::string name;
const uint64_t id;
private:
std::atomic<uint64_t> _key_count{0};
};
/* Representation of the collections in memory. */
class database {
public:
static std::string build_collection_name(const uint64_t id);
public:
/*
* Add a new collection, this will create the underlying collection in the database.
*/
void add_collection(uint64_t key_count = 0);
/* Get a collection using the id of the collection. */
collection &get_collection(uint64_t id);
/* Get a random collection. */
collection &get_random_collection();
/*
* Retrieve the current collection count, collection names are indexed from 0 so when using this
* take care to avoid an off by one error.
*/
uint64_t get_collection_count();
/* FIX-ME-Test-Framework: Replace usages of this with get_collection_ids. */
std::vector<std::string> get_collection_names();
std::vector<uint64_t> get_collection_ids();
void set_timestamp_manager(timestamp_manager *tsm);
void set_workload_tracking(workload_tracking *tracking);
void set_create_config(bool use_compression);
private:
std::string _collection_create_config = "";
scoped_session _session;
timestamp_manager *_tsm = nullptr;
workload_tracking *_tracking = nullptr;
uint64_t _next_collection_id = 0;
std::map<uint64_t, collection> _collections;
std::mutex _mtx;
};
} // namespace test_harness
#endif
|
