blob: 46909c2dd4e9cdce00b4bf9dc10097d045f33b30 (
plain) (
blame)
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
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#pragma once
#include <vespa/searchcore/proton/server/i_disk_mem_usage_notifier.h>
#include <vespa/searchcore/proton/server/i_disk_mem_usage_listener.h>
namespace proton
{
namespace test
{
/**
* Test notifier for disk/mem usage.
*/
class DiskMemUsageNotifier : public IDiskMemUsageNotifier
{
std::vector<IDiskMemUsageListener *> _listeners;
DiskMemUsageState _state;
public:
DiskMemUsageNotifier(DiskMemUsageState state)
: IDiskMemUsageNotifier(),
_listeners(),
_state(state)
{
}
DiskMemUsageNotifier()
: DiskMemUsageNotifier(DiskMemUsageState())
{
}
virtual ~DiskMemUsageNotifier() { }
virtual void addDiskMemUsageListener(IDiskMemUsageListener *listener) override {
_listeners.push_back(listener);
listener->notifyDiskMemUsage(_state);
}
virtual void removeDiskMemUsageListener(IDiskMemUsageListener *listener) override {
for (auto itr = _listeners.begin(); itr != _listeners.end(); ++itr) {
if (*itr == listener) {
_listeners.erase(itr);
break;
}
}
}
void notify(DiskMemUsageState state) {
if (_state != state) {
_state = state;
for (const auto &listener : _listeners) {
listener->notifyDiskMemUsage(state);
}
}
}
};
} // namespace proton::test
} // namespace proton
|