Support needed to measure user and service traffic in Chrome.

components/data_use_measurement/content/data_use_measurement.h

Index: components/data_use_measurement/content/data_use_measurement.h
diff --git a/components/data_use_measurement/content/data_use_measurement.h b/components/data_use_measurement/content/data_use_measurement.h
new file mode 100644
index 0000000000000000000000000000000000000000..f4712ac9fb748d47f708380e97796a0c0a85313f
--- /dev/null
+++ b/components/data_use_measurement/content/data_use_measurement.h
@@ -0,0 +1,108 @@
+// 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.
+
+#ifndef COMPONENTS_DATA_USE_MEASUREMENT_CONTENT_DATA_USE_MEASUREMENT_H_
+#define COMPONENTS_DATA_USE_MEASUREMENT_CONTENT_DATA_USE_MEASUREMENT_H_
+
+#include <stdint.h>
+
+#include <string>
+
+#include "base/memory/scoped_ptr.h"
+#include "build/build_config.h"
+#include "components/data_use_measurement/core/data_use_user_data.h"
+
+#if defined(OS_ANDROID)
+#include "base/android/application_status_listener.h"
+#endif
+
+namespace net {
+class URLRequest;
+}
+
+namespace data_use_measurement {
+
+// Records the data use of user traffic and various services in UMA histograms.
+// The UMA is broken down by network technology used (Wi-Fi vs cellular). On
+// Android, the UMA is further broken down by whether the application was in the
+// background or foreground during the request.
+class DataUseMeasurement {
+ public:
+  DataUseMeasurement();
+

[Alexei Svitkine (slow), 2015/09/01 20:04:04]

Nit: No empty line.

[amohammadkhan, 2015/09/01 23:02:43]

Done.

+  ~DataUseMeasurement();
+
+  // Records the data use of the |request|, thus |request| must be non-null.
+  void ReportDataUseUMA(const net::URLRequest* request);
+
+#if defined(OS_ANDROID)
+  // This function should just be used for testing purposes. A change in
+  // application state can be simulated by calling this function.
+  void OnApplicationStateChangeForTesting(
+      base::android::ApplicationState application_state);
+#endif
+
+ private:
+  // Specifies that data is received or sent, respectively.
+  enum TrafficDirection { DOWNSTREAM, UPSTREAM };
+
+  // The state of the application. Only available on Android and on other
+  // platforms it is always FOREGROUND.
+  enum AppState { BACKGROUND, FOREGROUND };
+
+  // The request is from a user or from a service.
+  enum RequestInitiator { REQUEST_INITIATOR_USER, REQUEST_INITIATOR_SERVICE };
+
+  // Returns the current application state (Foreground or Background). It always
+  // returns Foreground if Chrome is not running on Android.
+  AppState CurrentAppState();
+
+  // Produces a string representing whether the data use was on the send
+  // ("Upstream") or receive ("Downstream") path, whether the app was in the
+  // "Foreground" or "Background", and whether a "Cellular" or "WiFi" network
+  // was use. For example, "Upstream.Foreground.Cellular" is a possible output.
+  std::string GetSuffixForHistogramName(TrafficDirection dir);
+
+#if defined(OS_ANDROID)
+  // Called whenever the application transitions from foreground to background
+  // and
+  // vice versa.
+  void OnApplicationStateChange(
+      base::android::ApplicationState application_state);
+#endif
+
+  // A helper function used to record data use of services. It gets the size of
+  // exchanged message, its direction (which is upstream or downstream) and
+  // reports to two histogram groups. DataUse.MessageSize.ServiceName and
+  // DataUse.Services.{Dimensions}. In the second one, services are buckets.
+  void ReportDataUsageServices(
+      data_use_measurement::DataUseUserData::ServiceType service,
+      TrafficDirection dir,
+      int64_t message_size);
+
+  // A helper function used for reporting data use of user and all the services
+  // as a single group. Two histogram groups are updated by this function. The
+  // first one is DataUse.User.{Dimensions} and the second one is
+  // DataUse.NotUser.{Dimensions}. Based on |service_type| one of these groups
+  // is selected and updated.
+  void ReportDataUsageGeneral(RequestInitiator service_type,
+                              TrafficDirection dir,
+                              int64_t message_size);
+
+#if defined(OS_ANDROID)
+  // Application listener store the last known state of the application in this
+  // field.
+  base::android::ApplicationState app_state_;
+
+  // ApplicationStatusListener used to monitor whether the application is in the
+  // foreground or in the background. It is owned by DataUseMeasurement
+  scoped_ptr<base::android::ApplicationStatusListener> app_listener_;
+#endif
+
+  DISALLOW_COPY_AND_ASSIGN(DataUseMeasurement);
+};
+
+}  // namespace data_use_measurement
+
+#endif  // COMPONENTS_DATA_USE_MEASUREMENT_CONTENT_DATA_USE_MEASUREMENT_H_