searchcore/src/vespa/searchcore/proton/server/remove_operations_rate_tracker.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
30
31
32
33
34
35

// 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/common/operation_rate_tracker.h>
#include <vespa/searchcore/proton/documentmetastore/operation_listener.h>

namespace proton {

/**
 * Class that tracks the rate of remove operations handled by the document meta store.
 *
 * For each operation we can tell if it is above or below a given rate threshold.
 */
class RemoveOperationsRateTracker : public documentmetastore::OperationListener {
private:
    OperationRateTracker _remove_batch_tracker;
    OperationRateTracker _remove_tracker;

public:
    RemoveOperationsRateTracker(double remove_batch_rate_threshold,
                                double remove_rate_threshold);

    void notify_remove_batch() override;
    void notify_remove() override;

    bool remove_batch_above_threshold() const;
    bool remove_above_threshold() const;

    // Should only be used for testing
    OperationRateTracker& get_remove_batch_tracker() { return _remove_batch_tracker; }
    OperationRateTracker& get_remove_tracker() { return _remove_tracker; }
};

}