Record the data use by content type

components/data_use_measurement/core/data_use_measurement.h

 // 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 <memory>
 #include <string>
 
 #include "base/callback.h"
 #include "base/macros.h"
 #include "base/time/time.h"
 #include "build/build_config.h"
 #include "components/data_use_measurement/core/data_use_user_data.h"
 #include "components/metrics/data_use_tracker.h"
 
 #if defined(OS_ANDROID)
 #include "base/android/application_status_listener.h"
 #endif
 
 class GURL;
 
 namespace net {
+class HttpResponseHeaders;
 class URLRequest;
 }
 
 namespace data_use_measurement {
 
 class DataUseAscriber;
 class URLRequestClassifier;
 
 // 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.
 // TODO(amohammadkhan): Complete the layered architecture.
 // http://crbug.com/527460
 class DataUseMeasurement {
  public:
   DataUseMeasurement(
       std::unique_ptr<URLRequestClassifier> url_request_classifier,
       const metrics::UpdateUsagePrefCallbackType& metrics_data_use_forwarder,
       DataUseAscriber* ascriber);
   ~DataUseMeasurement();
 
   // Called before a request is sent.
   void OnBeforeURLRequest(net::URLRequest* request);
 
   // Called right after a redirect response code was received for |request|.
   void OnBeforeRedirect(const net::URLRequest& request,
                         const GURL& new_location);
 
+  // Called when response headers are received for |request|.
+  void OnHeadersReceived(net::URLRequest* request,
+                         const net::HttpResponseHeaders* response_headers);
+
   // Called when data is received or sent on the network, respectively.
   void OnNetworkBytesReceived(const net::URLRequest& request,
                               int64_t bytes_received);
   void OnNetworkBytesSent(const net::URLRequest& request, int64_t bytes_sent);
 
   // Indicates that |request| has been completed or failed.
   void OnCompleted(const net::URLRequest& request, bool started);
 
 #if defined(OS_ANDROID)
   // This function should just be used for testing purposes. A change in
@@ skipping 64 matching lines @@
       bool is_connection_cellular,
       int64_t message_size) const;
 
   // Records data use histograms split on TrafficDirection, AppState and
   // TabState.
   void RecordTabStateHistogram(TrafficDirection dir,
                                DataUseUserData::AppState app_state,
                                bool is_tab_visible,
                                int64_t bytes);
 
+  // Records data use histograms of user traffic and services traffic split on
+  // content type, AppState and TabState.
+  void RecordContentTypeHistogram(DataUseUserData::ContentType content_type,
+                                  bool is_user_traffic,
+                                  DataUseUserData::AppState app_state,
+                                  bool is_tab_visible,
+                                  int64_t bytes);
+
   // Classifier for identifying if an URL request is user initiated.
   std::unique_ptr<URLRequestClassifier> url_request_classifier_;
 
   // Callback for updating data use prefs.
   // TODO(rajendrant): If a similar mechanism would need be used for components
   // other than metrics, then the better approach would be to refactor this
   // class to support registering arbitrary observers. crbug.com/601185
   metrics::UpdateUsagePrefCallbackType metrics_data_use_forwarder_;
 
   // DataUseAscriber used to get the attributes of data use.
@@ skipping 26 matching lines @@
   // True if app is in background and first network read has not yet happened.
   bool no_reads_since_background_;
 #endif
 
   DISALLOW_COPY_AND_ASSIGN(DataUseMeasurement);
 };
 
 }  // namespace data_use_measurement
 
 #endif  // COMPONENTS_DATA_USE_MEASUREMENT_CONTENT_DATA_USE_MEASUREMENT_H_