persistence/src/vespa/persistence/spi/resource_usage_listener.h


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

// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#pragma once

#include "i_resource_usage_listener.h"
#include "resource_usage.h"
#include <memory>

namespace vespalib { class IDestructorCallback; }

namespace storage::spi {

/*
 * Class for listening to resource usage updates.
 */
class ResourceUsageListener : public IResourceUsageListener
{
    ResourceUsage _usage;
    std::unique_ptr<vespalib::IDestructorCallback> _register_guard;
public:
    ResourceUsageListener();
    ~ResourceUsageListener() override;
    void update_resource_usage(const ResourceUsage& resource_usage) override;
    const ResourceUsage& get_usage() const noexcept { return _usage; }
    void set_register_guard(std::unique_ptr<vespalib::IDestructorCallback> register_guard);
    void reset();
};

}