Support needed to measure user and service traffic in Chrome.

components/data_use_measurement/core/data_use_user_data.h

Index: components/data_use_measurement/core/data_use_user_data.h
diff --git a/components/data_use_measurement/core/data_use_user_data.h b/components/data_use_measurement/core/data_use_user_data.h
new file mode 100644
index 0000000000000000000000000000000000000000..0e7fa63453d74d9d878d4659ddc1699d3949fbbd
--- /dev/null
+++ b/components/data_use_measurement/core/data_use_user_data.h
@@ -0,0 +1,57 @@
+// 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_CORE_DATA_USE_USER_DATA_H_
+#define COMPONENTS_DATA_USE_MEASUREMENT_CORE_DATA_USE_USER_DATA_H_
+
+#include "base/supports_user_data.h"
+
+namespace net {
+class URLFetcher;
+}
+
+namespace data_use_measurement {
+
+// Used to annotate URLRequests with the service name if the URLRequest is used
+// by a service.
+class DataUseUserData : public base::SupportsUserData::Data {
+ public:
+  // This enum should be synced with DataUseServices enum in histograms.xml.
+  // Please keep them synced after any updates. Also add service name to
+  // GetServiceName function.
+  enum ServiceType {
+    NOT_TAGGED,
+    SUGGESTIONS,
+  };
+
+  explicit DataUseUserData(ServiceType service_type);
+  ~DataUseUserData() override;
+
+  // Helper function to create DataUseUserData. The caller takes the ownership
+  // of
+  // the returned object.
+  static base::SupportsUserData::Data* CreateDataUseUserData(

[Alexei Svitkine (slow), 2015/09/02 18:11:20]

Nit: No need to repeat the name of the type in the function. Can just be
DataUseUserData::Create()

[amohammadkhan, 2015/09/03 04:31:19]

Done.

+      DataUseUserData::ServiceType service);
+
+  // Return the service name of the ServiceType enum.
+  static const char* GetServiceName(ServiceType service);
+
+  // Services should use this function to attach their |service_type| to the
+  // URLFethcer serving them.
+  static void AttachDataUserUserDataToFetcher(net::URLFetcher* fetcher,

[Alexei Svitkine (slow), 2015/09/02 18:11:20]

Ditto, can be AttachToFetcher()

[amohammadkhan, 2015/09/03 04:31:19]

Done.

+                                              ServiceType service_type);
+
+  ServiceType service_type() const { return service_type_; }
+
+  // The key for retrieving back this type of user data.
+  static const void* kUserDataKey;
+
+ private:
+  ServiceType service_type_;

[Alexei Svitkine (slow), 2015/09/02 18:11:20]

Nit: const

[amohammadkhan, 2015/09/03 04:31:19]

Done.

+
+  DISALLOW_COPY_AND_ASSIGN(DataUseUserData);
+};
+
+}  // data_use_measurement namespace
+#endif  // COMPONENTS_DATA_USE_MEASUREMENT_CORE_DATA_USE_USER_DATA_H_