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

base / metrics / single_sample_metrics.cc [blame]

// 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 "base/metrics/single_sample_metrics.h"

#include "base/memory/ptr_util.h"
#include "base/metrics/histogram.h"

namespace base {

static SingleSampleMetricsFactory* g_factory = nullptr;

// static
SingleSampleMetricsFactory* SingleSampleMetricsFactory::Get() {
  if (!g_factory)
    g_factory = new DefaultSingleSampleMetricsFactory();

  return g_factory;
}

// static
void SingleSampleMetricsFactory::SetFactory(
    std::unique_ptr<SingleSampleMetricsFactory> factory) {
  DCHECK(!g_factory);
  g_factory = factory.release();
}

// static
void SingleSampleMetricsFactory::DeleteFactoryForTesting() {
  DCHECK(g_factory);
  delete g_factory;
  g_factory = nullptr;
}

std::unique_ptr<SingleSampleMetric>
DefaultSingleSampleMetricsFactory::CreateCustomCountsMetric(
    const std::string& histogram_name,
    HistogramBase::Sample min,
    HistogramBase::Sample max,
    uint32_t bucket_count) {
  return std::make_unique<DefaultSingleSampleMetric>(
      histogram_name, min, max, bucket_count,
      HistogramBase::kUmaTargetedHistogramFlag);
}

DefaultSingleSampleMetric::DefaultSingleSampleMetric(
    const std::string& histogram_name,
    HistogramBase::Sample min,
    HistogramBase::Sample max,
    uint32_t bucket_count,
    int32_t flags)
    : histogram_(Histogram::FactoryGet(histogram_name,
                                       min,
                                       max,
                                       bucket_count,
                                       flags)) {
  // Bad construction parameters may lead to |histogram_| being null; DCHECK to
  // find accidental errors in production. We must still handle the nullptr in
  // destruction though since this construction may come from another untrusted
  // process.
  DCHECK(histogram_);
}

DefaultSingleSampleMetric::~DefaultSingleSampleMetric() {
  // |histogram_| may be nullptr if bad construction parameters are given.
  if (sample_ < 0 || !histogram_)
    return;
  histogram_->Add(sample_);
}

void DefaultSingleSampleMetric::SetSample(HistogramBase::Sample sample) {
  DCHECK_GE(sample, 0);
  sample_ = sample;
}

}  // namespace base