single_sample_metrics_factory_impl.cc 3.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495
  1. // Copyright 2017 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "components/metrics/single_sample_metrics_factory_impl.h"
  5. #include <memory>
  6. #include "base/threading/thread_checker.h"
  7. #include "mojo/public/cpp/bindings/pending_remote.h"
  8. #include "mojo/public/cpp/bindings/remote.h"
  9. namespace metrics {
  10. namespace {
  11. class SingleSampleMetricImpl : public base::SingleSampleMetric {
  12. public:
  13. SingleSampleMetricImpl(mojo::PendingRemote<mojom::SingleSampleMetric> metric)
  14. : metric_(std::move(metric)) {}
  15. SingleSampleMetricImpl(const SingleSampleMetricImpl&) = delete;
  16. SingleSampleMetricImpl& operator=(const SingleSampleMetricImpl&) = delete;
  17. ~SingleSampleMetricImpl() override {
  18. DCHECK(thread_checker_.CalledOnValidThread());
  19. }
  20. void SetSample(base::HistogramBase::Sample sample) override {
  21. DCHECK(thread_checker_.CalledOnValidThread());
  22. metric_->SetSample(sample);
  23. }
  24. private:
  25. base::ThreadChecker thread_checker_;
  26. mojo::Remote<mojom::SingleSampleMetric> metric_;
  27. };
  28. } // namespace
  29. SingleSampleMetricsFactoryImpl::SingleSampleMetricsFactoryImpl(
  30. CreateProviderCB create_provider_cb)
  31. : create_provider_cb_(std::move(create_provider_cb)) {}
  32. SingleSampleMetricsFactoryImpl::~SingleSampleMetricsFactoryImpl() {}
  33. std::unique_ptr<base::SingleSampleMetric>
  34. SingleSampleMetricsFactoryImpl::CreateCustomCountsMetric(
  35. const std::string& histogram_name,
  36. base::HistogramBase::Sample min,
  37. base::HistogramBase::Sample max,
  38. uint32_t bucket_count) {
  39. return CreateMetric(histogram_name, min, max, bucket_count,
  40. base::HistogramBase::kUmaTargetedHistogramFlag);
  41. }
  42. void SingleSampleMetricsFactoryImpl::DestroyProviderForTesting() {
  43. if (auto* provider = provider_tls_.Get())
  44. delete provider;
  45. provider_tls_.Set(nullptr);
  46. }
  47. std::unique_ptr<base::SingleSampleMetric>
  48. SingleSampleMetricsFactoryImpl::CreateMetric(const std::string& histogram_name,
  49. base::HistogramBase::Sample min,
  50. base::HistogramBase::Sample max,
  51. uint32_t bucket_count,
  52. int32_t flags) {
  53. mojo::PendingRemote<mojom::SingleSampleMetric> metric;
  54. GetProvider()->AcquireSingleSampleMetric(
  55. histogram_name, min, max, bucket_count, flags,
  56. metric.InitWithNewPipeAndPassReceiver());
  57. return std::make_unique<SingleSampleMetricImpl>(std::move(metric));
  58. }
  59. mojom::SingleSampleMetricsProvider*
  60. SingleSampleMetricsFactoryImpl::GetProvider() {
  61. // Check the current TLS slot to see if we have created a provider already for
  62. // this thread.
  63. if (auto* provider = provider_tls_.Get())
  64. return provider->get();
  65. // If not, create a new one which will persist until process shutdown and put
  66. // it in the TLS slot for the current thread.
  67. mojo::Remote<mojom::SingleSampleMetricsProvider>* provider =
  68. new mojo::Remote<mojom::SingleSampleMetricsProvider>();
  69. provider_tls_.Set(provider);
  70. // Start the provider connection and return it; it won't be fully connected
  71. // until later, but mojo will buffer all calls prior to completion.
  72. create_provider_cb_.Run(provider->BindNewPipeAndPassReceiver());
  73. return provider->get();
  74. }
  75. } // namespace metrics