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
128
|
// Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include <vespa/vespalib/util/reusable_set_pool.h>
#include <vespa/vespalib/gtest/gtest.h>
using namespace vespalib;
using Mark = ReusableSet::Mark;
void verify_set(const ReusableSet &set, size_t sz, Mark val, size_t marked) {
EXPECT_EQ(sz, set.capacity());
EXPECT_EQ(val, set.generation());
size_t count = 0;
for (size_t i = 0; i < set.capacity(); ++i) {
if (set.is_marked(i)) ++count;
}
EXPECT_EQ(marked, count);
}
void verify_handle(const ReusableSetHandle &set, size_t sz, Mark val, size_t marked) {
EXPECT_EQ(sz, set.capacity());
EXPECT_EQ(val, set.generation());
size_t count = 0;
for (size_t i = 0; i < set.capacity(); ++i) {
if (set.is_marked(i)) ++count;
}
EXPECT_EQ(marked, count);
}
class Pool : public ::testing::Test {
public:
ReusableSetPool pool;
Pool() : pool() {}
~Pool() {}
void exercise(ReusableSetHandle &set) {
size_t sz = set.capacity();
size_t count = 0;
for (size_t i = 0; i < sz; ++i) {
if (set.is_marked(i)) ++count;
}
EXPECT_EQ(0, count);
for (int i = 0; i < 17; ++i) {
set.mark((i * 711) % sz);
}
count = 0;
for (size_t i = 0; i < sz; ++i) {
if (set.is_marked(i)) ++count;
}
EXPECT_EQ(17, count);
for (int i = 0; i < 17; ++i) {
set.mark((i * 711) % sz);
}
count = 0;
for (size_t i = 0; i < sz; ++i) {
if (set.is_marked(i)) ++count;
}
EXPECT_EQ(17, count);
}
};
TEST(ReusableSetTest, simple_usage)
{
ReusableSet visited(7);
verify_set(visited, 7, 1, 0);
visited.mark(1);
visited.mark(2);
visited.mark(4);
EXPECT_EQ(false, visited.is_marked(0));
EXPECT_EQ(true, visited.is_marked(1));
EXPECT_EQ(true, visited.is_marked(2));
EXPECT_EQ(false, visited.is_marked(3));
verify_set(visited, 7, 1, 3);
visited.mark(4);
visited.mark(1);
visited.mark(2);
verify_set(visited, 7, 1, 3);
EXPECT_EQ(false, visited.is_marked(0));
EXPECT_EQ(true, visited.is_marked(1));
EXPECT_EQ(true, visited.is_marked(2));
EXPECT_EQ(false, visited.is_marked(3));
visited.clear();
verify_set(visited, 7, 2, 0);
visited.clear();
verify_set(visited, 7, 3, 0);
}
TEST_F(Pool, reuse_works)
{
for (int i = 0; i < 65535; ++i) {
auto handle = pool.get(7);
EXPECT_EQ(i, pool.reuse_count());
EXPECT_EQ(1, pool.create_count());
verify_handle(handle, 248, i+1, 0);
exercise(handle);
}
EXPECT_TRUE(500 < pool.memory_usage());
EXPECT_TRUE(1000 > pool.memory_usage());
for (int i = 0; i < 5; ++i) {
auto handle = pool.get(7);
EXPECT_EQ(65535+i, pool.reuse_count());
EXPECT_EQ(1, pool.create_count());
verify_handle(handle, 248, i+1, 0);
exercise(handle);
}
auto handle3 = pool.get(260);
EXPECT_EQ(2, pool.create_count());
verify_handle(handle3, 297, 1, 0);
exercise(handle3);
{
auto handle4 = pool.get(400);
EXPECT_EQ(3, pool.create_count());
verify_handle(handle4, 400, 1, 0);
exercise(handle4);
}
auto handle7 = pool.get(401);
EXPECT_EQ(4, pool.create_count());
verify_handle(handle7, 480, 1, 0);
exercise(handle7);
EXPECT_TRUE(1000 < pool.memory_usage());
EXPECT_TRUE(3000 > pool.memory_usage());
auto handle8 = pool.get(2500);
auto handle9 = pool.get(2500);
EXPECT_TRUE(11000 < pool.memory_usage());
EXPECT_TRUE(13000 > pool.memory_usage());
}
GTEST_MAIN_RUN_ALL_TESTS()
|