aboutsummaryrefslogtreecommitdiffstats
path: root/searchcore/src/vespa/searchcore/proton/server/disk_mem_usage_forwarder.h
blob: da04b18e8c12cd3a8f93e6d040fff5d93ea710d0 (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
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#pragma once

#include "i_disk_mem_usage_notifier.h"
#include "i_disk_mem_usage_listener.h"
#include <vespa/searchcorespi/index/i_thread_service.h>
#include <vector>
#include <mutex>

namespace proton {

/**
 * Forwarder for disk/memory usage state changes. Notification is forwarded
 * as a task run by the supplied executor.
 */
class DiskMemUsageForwarder : public IDiskMemUsageNotifier,
                              public IDiskMemUsageListener
{
    searchcorespi::index::IThreadService &_executor;
    std::vector<IDiskMemUsageListener *> _listeners;
    std::mutex        _lock;
    DiskMemUsageState _state;
    void forward(DiskMemUsageState state);
public:
    DiskMemUsageForwarder(searchcorespi::index::IThreadService &executor);
    ~DiskMemUsageForwarder() override;
    void addDiskMemUsageListener(IDiskMemUsageListener *listener) override;
    void removeDiskMemUsageListener(IDiskMemUsageListener *listener) override;
    void notifyDiskMemUsage(DiskMemUsageState state) override;
};

} // namespace proton