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
105
106
|
// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/segmentation_platform/internal/signals/histogram_signal_handler.h"
#include <cstdint>
#include "base/containers/contains.h"
#include "base/functional/bind.h"
#include "base/functional/callback_helpers.h"
#include "base/metrics/metrics_hashes.h"
#include "base/observer_list.h"
#include "components/segmentation_platform/internal/database/signal_database.h"
#include "components/segmentation_platform/internal/database/ukm_types.h"
namespace segmentation_platform {
HistogramSignalHandler::HistogramSignalHandler(const std::string& profile_id,
SignalDatabase* signal_database,
UkmDatabase* ukm_db)
: profile_id_(profile_id),
db_(signal_database),
ukm_db_(ukm_db),
metrics_enabled_(false) {}
HistogramSignalHandler::~HistogramSignalHandler() {
DCHECK(observers_.empty());
}
void HistogramSignalHandler::SetRelevantHistograms(
const RelevantHistograms& histograms) {
auto it = histogram_observers_.begin();
while (it != histogram_observers_.end()) {
if (!base::Contains(histograms, it->first)) {
it = histogram_observers_.erase(it);
} else {
++it;
}
}
for (const auto& pair : histograms) {
if (base::Contains(histogram_observers_, pair)) {
continue;
}
const auto& histogram_name = pair.first;
proto::SignalType signal_type = pair.second;
auto histogram_observer = std::make_unique<
base::StatisticsRecorder::ScopedHistogramSampleObserver>(
histogram_name,
base::BindRepeating(&HistogramSignalHandler::OnHistogramSample,
weak_ptr_factory_.GetWeakPtr(), signal_type));
histogram_observers_[pair] = std::move(histogram_observer);
}
}
void HistogramSignalHandler::EnableMetrics(bool enable_metrics) {
if (metrics_enabled_ == enable_metrics) {
return;
}
metrics_enabled_ = enable_metrics;
}
void HistogramSignalHandler::OnHistogramSample(
proto::SignalType signal_type,
std::string_view histogram_name,
uint64_t name_hash,
base::HistogramBase::Sample32 sample) {
if (!metrics_enabled_) {
return;
}
db_->WriteSample(signal_type, name_hash, sample,
base::BindOnce(&HistogramSignalHandler::OnSampleWritten,
weak_ptr_factory_.GetWeakPtr(),
std::string(histogram_name), sample));
if (ukm_db_) {
ukm_db_->AddUmaMetric(profile_id_, UmaMetricEntry{.type = signal_type,
.name_hash = name_hash,
.time = base::Time::Now(),
.value = sample});
}
}
void HistogramSignalHandler::AddObserver(Observer* observer) {
observers_.AddObserver(observer);
}
void HistogramSignalHandler::RemoveObserver(Observer* observer) {
observers_.RemoveObserver(observer);
}
void HistogramSignalHandler::OnSampleWritten(
std::string_view histogram_name,
base::HistogramBase::Sample32 sample,
bool success) {
if (!success) {
return;
}
for (Observer& ob : observers_) {
ob.OnHistogramSignalUpdated(histogram_name, sample);
}
}
} // namespace segmentation_platform
|
