Implement UMA log throttling for cellular connections

components/data_use_measurement/content/data_use_measurement.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/data_use_measurement/content/data_use_measurement.h"
 
 #include "base/metrics/histogram.h"
 #include "base/metrics/sparse_histogram.h"
 #include "base/strings/stringprintf.h"
 #include "build/build_config.h"
+#include "components/metrics/data_use_tracker.h"
 #include "content/public/browser/resource_request_info.h"
 #include "net/base/network_change_notifier.h"
 #include "net/base/upload_data_stream.h"
 #include "net/http/http_response_headers.h"
 #include "net/url_request/url_request.h"
 
 namespace data_use_measurement {
 
 namespace {
 
@@ skipping 21 matching lines @@
   histogram->AddCount(sample, value);
 }
 
 }  // namespace
 
 DataUseMeasurement::DataUseMeasurement()
 #if defined(OS_ANDROID)
     : app_state_(base::android::APPLICATION_STATE_HAS_RUNNING_ACTIVITIES),
       app_listener_(new base::android::ApplicationStatusListener(
           base::Bind(&DataUseMeasurement::OnApplicationStateChange,
-                     base::Unretained(this))))
+                     base::Unretained(this)))),
+      weak_ptr_factory_(this)
 #endif
-{
+    : weak_ptr_factory_(this) {
 }
 
 DataUseMeasurement::~DataUseMeasurement(){};
 
 void DataUseMeasurement::ReportDataUseUMA(
     const net::URLRequest* request) const {
   const content::ResourceRequestInfo* info =
       content::ResourceRequestInfo::ForRequest(request);
   // Having |info| is the sign of a request for a web content from user. For now
   // we could add a condition to check ProcessType in info is
@@ skipping 16 matching lines @@
                        UPSTREAM),
       total_upload_bytes);
   RecordUMAHistogramCount(
       GetHistogramName(is_user_traffic ? "DataUse.TrafficSize.User"
                                        : "DataUse.TrafficSize.System",
                        DOWNSTREAM),
       total_received_bytes);
 
   DataUseUserData* attached_service_data = reinterpret_cast<DataUseUserData*>(
       request->GetUserData(DataUseUserData::kUserDataKey));
-
+  DataUseUserData::ServiceName service_name =
+      attached_service_data ? attached_service_data->service_name()
+                            : DataUseUserData::NOT_TAGGED;
   if (!is_user_traffic) {
-    DataUseUserData::ServiceName service_name =
-        attached_service_data ? attached_service_data->service_name()
-                              : DataUseUserData::NOT_TAGGED;
     ReportDataUsageServices(service_name, UPSTREAM, total_upload_bytes);
     ReportDataUsageServices(service_name, DOWNSTREAM, total_received_bytes);
   }
+  bool is_cellular = net::NetworkChangeNotifier::IsConnectionCellular(
+      net::NetworkChangeNotifier::GetConnectionType());
+  if (is_cellular)

[Alexei Svitkine (slow), 2016/03/29 16:29:10]

Nit: Combine this with if below.

[gayane -on leave until 09-2017, 2016/03/31 01:38:24]

Done.

+    if (!metrics_data_use_forwarder_.is_null()) {
+      metrics::DataUseTracker::RunOnUI(
+          metrics_data_use_forwarder_,
+          attached_service_data->GetServiceNameAsString(service_name),
+          total_upload_bytes + total_received_bytes);
+    }
 }
 
 #if defined(OS_ANDROID)
 void DataUseMeasurement::OnApplicationStateChangeForTesting(
     base::android::ApplicationState application_state) {
   app_state_ = application_state;
 }
 #endif
 
 DataUseMeasurement::AppState DataUseMeasurement::CurrentAppState() const {
@@ skipping 30 matching lines @@
   RecordUMAHistogramCount(
       "DataUse.MessageSize." + DataUseUserData::GetServiceNameAsString(service),
       message_size);
   if (message_size > 0) {
     IncreaseSparseHistogramByValue(
         GetHistogramName("DataUse.MessageSize.AllServices", dir), service,
         message_size);
   }
 }
 
+void DataUseMeasurement::SetMetricsDataUseForwarder(
+    base::Callback<void(const std::string&, int)> metrics_data_use_forwarder) {
+  metrics_data_use_forwarder_ = metrics_data_use_forwarder;
+}
+
+base::WeakPtr<DataUseMeasurement> DataUseMeasurement::GetWeakPtr() {
+  return weak_ptr_factory_.GetWeakPtr();
+}
 }  // namespace data_use_measurement