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
129
130
131
132
133
134
|
// Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include "service_map_history.h"
#include <vespa/log/log.h>
LOG_SETUP(".slobrok.server.service_map_history");
namespace slobrok {
ServiceMapHistory::UpdateLog::UpdateLog()
: startGeneration(1),
currentGeneration(1),
updates(keep_items + 1)
{}
ServiceMapHistory::UpdateLog::~UpdateLog() = default;
void ServiceMapHistory::UpdateLog::add(const vespalib::string &name) {
currentGeneration.add();
updates.push(name);
while (updates.size() > keep_items) {
startGeneration.add();
updates.pop();
}
}
bool ServiceMapHistory::UpdateLog::isInRange(const Generation &gen) const {
return gen.inRangeInclusive(startGeneration, currentGeneration);
}
std::vector<vespalib::string>
ServiceMapHistory::UpdateLog::updatedSince(const Generation &gen) const {
std::vector<vespalib::string> result;
uint32_t skip = startGeneration.distance(gen);
uint32_t last = startGeneration.distance(currentGeneration);
for (uint32_t idx = skip; idx < last; ++idx) {
result.push_back(updates.peek(idx));
}
return result;
}
//-----------------------------------------------------------------------------
ServiceMapHistory::ServiceMapHistory()
: _lock(),
_map(),
_waitList(),
_log()
{}
ServiceMapHistory::~ServiceMapHistory() {
notify_updated();
}
void ServiceMapHistory::notify_updated() {
WaitList waitList;
{
std::lock_guard guard(_lock);
std::swap(waitList, _waitList);
}
for (auto & [ handler, gen ] : waitList) {
handler->handle(makeDiffFrom(gen));
}
}
void ServiceMapHistory::asyncGenerationDiff(DiffCompletionHandler *handler, const Generation &fromGen) {
{
std::lock_guard guard(_lock);
if (fromGen == myGen()) {
_waitList.emplace_back(handler, fromGen);
return;
}
}
handler->handle(makeDiffFrom(fromGen));
}
bool ServiceMapHistory::cancel(DiffCompletionHandler *handler) {
std::lock_guard guard(_lock);
size_t removed = std::erase_if(_waitList, [=](const Waiter &elem){ return elem.first == handler; });
return (removed > 0);
}
void ServiceMapHistory::remove(const vespalib::string &name) {
{
std::lock_guard guard(_lock);
auto iter = _map.find(name);
if (iter == _map.end()) {
LOG(warning, "already removed: %s", name.c_str());
// already removed
return;
}
_map.erase(iter);
_log.add(name);
}
notify_updated();
}
void ServiceMapHistory::update(const ServiceMapping &mapping) {
{
std::lock_guard guard(_lock);
_map.insert_or_assign(mapping.name, mapping.spec);
_log.add(mapping.name);
}
notify_updated();
}
MapDiff ServiceMapHistory::makeDiffFrom(const Generation &fromGen) const {
std::lock_guard guard(_lock);
if (_log.isInRange(fromGen)) {
std::vector<vespalib::string> removes;
ServiceMappingList updates;
auto changes = _log.updatedSince(fromGen);
for (const vespalib::string & name : changes) {
if (_map.contains(name)) {
updates.emplace_back(name, _map.at(name));
} else {
removes.push_back(name);
}
}
return MapDiff(fromGen, removes, updates, myGen());
} else {
ServiceMappingList mappings;
for (const auto & [ name, spec ] : _map) {
mappings.emplace_back(name, spec);
}
return MapDiff(mappings, myGen());
}
}
//-----------------------------------------------------------------------------
} // namespace slobrok
|
