aboutsummaryrefslogtreecommitdiffstats
path: root/searchcore/src/vespa/searchcore/proton/metrics/documentdb_job_trackers.cpp
blob: f7605e8b1c6389d2ec30c8cc52db1a0312ad9407 (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
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
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#include "documentdb_job_trackers.h"
#include "job_tracked_flush_target.h"

#include <vespa/log/log.h>
LOG_SETUP(".proton.metrics.documentdb_job_trackers");

using searchcorespi::IFlushTarget;
using FTT = IFlushTarget::Type;
using FTC = IFlushTarget::Component;

using time_point = std::chrono::time_point<std::chrono::steady_clock>;

namespace proton {

DocumentDBJobTrackers::DocumentDBJobTrackers()
    : _lock(),
      _now(std::chrono::steady_clock::now()),
      _attributeFlush(std::make_shared<JobTracker>(_now, _lock)),
      _memoryIndexFlush(std::make_shared<JobTracker>(_now, _lock)),
      _diskIndexFusion(std::make_shared<JobTracker>(_now, _lock)),
      _documentStoreFlush(std::make_shared<JobTracker>(_now, _lock)),
      _documentStoreCompact(std::make_shared<JobTracker>(_now, _lock)),
      _bucketMove(std::make_shared<JobTracker>(_now, _lock)),
      _lidSpaceCompact(std::make_shared<JobTracker>(_now, _lock)),
      _removedDocumentsPrune(std::make_shared<JobTracker>(_now, _lock))
{
}

DocumentDBJobTrackers::~DocumentDBJobTrackers() = default;

namespace {

IFlushTarget::SP
trackFlushTarget(std::shared_ptr<IJobTracker> tracker,
                 std::shared_ptr<IFlushTarget> target)
{
    return std::make_shared<JobTrackedFlushTarget>(std::move(tracker), std::move(target));
}

}

IFlushTarget::List
DocumentDBJobTrackers::trackFlushTargets(const IFlushTarget::List &flushTargets)
{
    IFlushTarget::List retval;
    for (const auto &ft : flushTargets) {
        if (ft->getComponent() == FTC::ATTRIBUTE && ft->getType() == FTT::SYNC) {
            retval.push_back(trackFlushTarget(_attributeFlush, ft));
        } else if (ft->getComponent() == FTC::ATTRIBUTE && ft->getType() == FTT::GC) {
            retval.push_back(trackFlushTarget(_attributeFlush, ft));
        } else if (ft->getComponent() == FTC::INDEX && ft->getType() == FTT::FLUSH) {
            retval.push_back(trackFlushTarget(_memoryIndexFlush, ft));
        } else if (ft->getComponent() == FTC::INDEX && ft->getType() == FTT::GC) {
            retval.push_back(trackFlushTarget(_diskIndexFusion, ft));
        } else if (ft->getComponent() == FTC::DOCUMENT_STORE && ft->getType() == FTT::SYNC) {
            retval.push_back(trackFlushTarget(_documentStoreFlush, ft));
        } else if (ft->getComponent() == FTC::DOCUMENT_STORE && ft->getType() == FTT::GC) {
            retval.push_back(trackFlushTarget(_documentStoreCompact, ft));
        } else {
            LOG(warning, "trackFlushTargets(): Flush target '%s' with type '%d' and component '%d' "
                    "is not known and will not be tracked",
                    ft->getName().c_str(), static_cast<int>(ft->getType()),
                    static_cast<int>(ft->getComponent()));
            retval.push_back(ft);
        }
    }
    return retval;
}

namespace {

double
updateMetric(metrics::DoubleAverageMetric &metric,
             JobTracker &tracker,
             time_point now,
             const std::lock_guard<std::mutex> &guard)
{
    double load = tracker.sampleLoad(now, guard);
    metric.addValue(load);
    return load;
}

}

void
DocumentDBJobTrackers::updateMetrics(DocumentDBTaggedMetrics::JobMetrics &metrics)
{
    std::lock_guard<std::mutex> guard(_lock);
    _now = std::chrono::steady_clock::now();
    double load = 0.0;
    load += updateMetric(metrics.attributeFlush, *_attributeFlush, _now, guard);
    load += updateMetric(metrics.memoryIndexFlush, *_memoryIndexFlush, _now, guard);
    load += updateMetric(metrics.diskIndexFusion, *_diskIndexFusion, _now, guard);
    load += updateMetric(metrics.documentStoreFlush, *_documentStoreFlush, _now, guard);
    load += updateMetric(metrics.documentStoreCompact, *_documentStoreCompact, _now, guard);
    load += updateMetric(metrics.bucketMove, *_bucketMove, _now, guard);
    load += updateMetric(metrics.lidSpaceCompact, *_lidSpaceCompact, _now, guard);
    load += updateMetric(metrics.removedDocumentsPrune, *_removedDocumentsPrune, _now, guard);
    metrics.total.addValue(load);
}

} // namespace proton