Refactor the DataUseMeasurement class

components/data_use_measurement/content/data_use_measurement.cc

Index: components/data_use_measurement/content/data_use_measurement.cc
diff --git a/components/data_use_measurement/content/data_use_measurement.cc b/components/data_use_measurement/content/data_use_measurement.cc
index e45368ae1b65ca1c67fdb70a3ddccd2e30be4ff1..a671c2dd9496f54d247ce4a5d6bbb749c8d68aa5 100644
--- a/components/data_use_measurement/content/data_use_measurement.cc
+++ b/components/data_use_measurement/content/data_use_measurement.cc
@@ -59,6 +59,18 @@ DataUseMeasurement::DataUseMeasurement(
 
 DataUseMeasurement::~DataUseMeasurement(){};
 
+void DataUseMeasurement::OnBeforeRedirect(net::URLRequest* request,
+                                          const GURL& new_location) {
+  // Recording data use of request on redirects.
+  ReportDataUseUMA(request);
+}
+
+void DataUseMeasurement::OnCompleted(net::URLRequest* request, bool started) {
+  // TODO(amohammadkhan): Verify that there is no double recording in data use
+  // of redirected requests.
+  ReportDataUseUMA(request);
+}
+
 void DataUseMeasurement::ReportDataUseUMA(
     const net::URLRequest* request) const {