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
|
// Copyright 2017 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/metrics/single_sample_metrics_factory_impl.h"
#include <memory>
#include "base/threading/thread_checker.h"
#include "mojo/public/cpp/bindings/pending_remote.h"
#include "mojo/public/cpp/bindings/remote.h"
namespace metrics {
namespace {
class SingleSampleMetricImpl : public base::SingleSampleMetric {
public:
SingleSampleMetricImpl(mojo::PendingRemote<mojom::SingleSampleMetric> metric)
: metric_(std::move(metric)) {}
SingleSampleMetricImpl(const SingleSampleMetricImpl&) = delete;
SingleSampleMetricImpl& operator=(const SingleSampleMetricImpl&) = delete;
~SingleSampleMetricImpl() override {
DCHECK(thread_checker_.CalledOnValidThread());
}
void SetSample(base::HistogramBase::Sample32 sample) override {
DCHECK(thread_checker_.CalledOnValidThread());
metric_->SetSample(sample);
}
private:
base::ThreadChecker thread_checker_;
mojo::Remote<mojom::SingleSampleMetric> metric_;
};
constinit thread_local mojo::Remote<mojom::SingleSampleMetricsProvider>*
provider = nullptr;
} // namespace
SingleSampleMetricsFactoryImpl::SingleSampleMetricsFactoryImpl(
CreateProviderCB create_provider_cb)
: create_provider_cb_(std::move(create_provider_cb)) {}
SingleSampleMetricsFactoryImpl::~SingleSampleMetricsFactoryImpl() = default;
std::unique_ptr<base::SingleSampleMetric>
SingleSampleMetricsFactoryImpl::CreateCustomCountsMetric(
const std::string& histogram_name,
base::HistogramBase::Sample32 min,
base::HistogramBase::Sample32 max,
uint32_t bucket_count) {
return CreateMetric(histogram_name, min, max, bucket_count,
base::HistogramBase::kUmaTargetedHistogramFlag);
}
void SingleSampleMetricsFactoryImpl::DestroyProviderForTesting() {
delete provider;
provider = nullptr;
}
std::unique_ptr<base::SingleSampleMetric>
SingleSampleMetricsFactoryImpl::CreateMetric(const std::string& histogram_name,
base::HistogramBase::Sample32 min,
base::HistogramBase::Sample32 max,
uint32_t bucket_count,
int32_t flags) {
mojo::PendingRemote<mojom::SingleSampleMetric> metric;
GetProvider()->AcquireSingleSampleMetric(
histogram_name, min, max, bucket_count, flags,
metric.InitWithNewPipeAndPassReceiver());
return std::make_unique<SingleSampleMetricImpl>(std::move(metric));
}
mojom::SingleSampleMetricsProvider*
SingleSampleMetricsFactoryImpl::GetProvider() {
// Check the current TLS slot to see if we have created a provider already for
// this thread.
if (!provider) {
// If not, create a new one which will persist until process shutdown and
// put it in the TLS slot for the current thread.
provider = new mojo::Remote<mojom::SingleSampleMetricsProvider>();
// Start the provider connection and return it; it won't be fully connected
// until later, but mojo will buffer all calls prior to completion.
create_provider_cb_.Run(provider->BindNewPipeAndPassReceiver());
}
return provider->get();
}
} // namespace metrics
|
