Report data usage UMA for Chrome services

sync/internal_api/public/http_bridge.h

 // Copyright 2012 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 SYNC_INTERNAL_API_PUBLIC_HTTP_BRIDGE_H_
 #define SYNC_INTERNAL_API_PUBLIC_HTTP_BRIDGE_H_
 
 #include <string>
 
 #include "base/basictypes.h"
+#include "base/callback.h"
 #include "base/compiler_specific.h"
 #include "base/gtest_prod_util.h"
 #include "base/memory/ref_counted.h"
 #include "base/synchronization/lock.h"
 #include "base/synchronization/waitable_event.h"
 #include "base/timer/timer.h"
 #include "net/url_request/url_fetcher_delegate.h"
 #include "net/url_request/url_request_context.h"
 #include "net/url_request/url_request_context_getter.h"
 #include "sync/base/sync_export.h"
@@ skipping 26 matching lines @@
 // This is a one-time use bridge. Create one for each request you want to make.
 // It is RefCountedThreadSafe because it can PostTask to the io loop, and thus
 // needs to stick around across context switches, etc.
 class SYNC_EXPORT_PRIVATE HttpBridge
     : public base::RefCountedThreadSafe<HttpBridge>,
       public HttpPostProviderInterface,
       public net::URLFetcherDelegate {
  public:
   HttpBridge(const std::string& user_agent,
              const scoped_refptr<net::URLRequestContextGetter>& context,
-             const NetworkTimeUpdateCallback& network_time_update_callback);
+             const NetworkTimeUpdateCallback& network_time_update_callback,
+             base::Callback<void(net::URLFetcher*)> bind_to_tracker_callback);

[Nicolas Zea, 2015/09/09 19:56:51]

nit: pass the callback by const ref here and elsewhere, e.g. const
base::Callback<void(net::URLFetcher*)>&. Also consider making a typedef for the
callback and using that instead of redefining the callback type everywhere.

[amohammadkhan, 2015/09/09 22:21:56]

Done. I put the typedef of my callback in http_post_provider_factory.h, it
caused minimum change to other files, please let me know if it is not a good
place for it.

 
   // HttpPostProvider implementation.
   void SetExtraRequestHeaders(const char* headers) override;
   void SetURL(const char* url, int port) override;
   void SetPostPayload(const char* content_type,
                       int content_length,
                       const char* content) override;
   bool MakeSynchronousPost(int* error_code, int* response_code) override;
   void Abort() override;
 
@@ skipping 105 matching lines @@
   mutable base::Lock fetch_state_lock_;
   URLFetchState fetch_state_;
 
   scoped_refptr<net::URLRequestContextGetter> request_context_getter_;
 
   const scoped_refptr<base::SingleThreadTaskRunner> network_task_runner_;
 
   // Callback for updating network time.
   NetworkTimeUpdateCallback network_time_update_callback_;
 
+  // A callback to tag Sync request to be able to record data use of this
+  // service by data_use_measurement component.
+  base::Callback<void(net::URLFetcher*)> bind_to_tracker_callback_;
+
   DISALLOW_COPY_AND_ASSIGN(HttpBridge);
 };
 
 class SYNC_EXPORT HttpBridgeFactory : public HttpPostProviderFactory,
                                       public CancelationObserver {
  public:
   HttpBridgeFactory(
       const scoped_refptr<net::URLRequestContextGetter>&
           baseline_context_getter,
       const NetworkTimeUpdateCallback& network_time_update_callback,
       CancelationSignal* cancelation_signal);
   ~HttpBridgeFactory() override;
 
   // HttpPostProviderFactory:
-  void Init(const std::string& user_agent) override;
+  void Init(
+      const std::string& user_agent,
+      base::Callback<void(net::URLFetcher*)> bind_to_tracker_callback) override;
   HttpPostProviderInterface* Create() override;
   void Destroy(HttpPostProviderInterface* http) override;
 
   // CancelationObserver implementation:
   void OnSignalReceived() override;
 
  private:
   // The user agent to use in all requests.
   std::string user_agent_;
 
   // Protects |request_context_getter_| to allow releasing it's reference from
   // the sync thread, even when it's in use on the IO thread.
   base::Lock request_context_getter_lock_;
 
   // The request context getter used for making all requests.
   scoped_refptr<net::URLRequestContextGetter> request_context_getter_;
 
   NetworkTimeUpdateCallback network_time_update_callback_;
 
   CancelationSignal* const cancelation_signal_;
 
+  // A callback to tag Sync request to be able to record data use of this
+  // service by data_use_measurement component.
+  base::Callback<void(net::URLFetcher*)> bind_to_tracker_callback_;
+
   DISALLOW_COPY_AND_ASSIGN(HttpBridgeFactory);
 };
 
 }  //  namespace syncer
 
 #endif  // SYNC_INTERNAL_API_PUBLIC_HTTP_BRIDGE_H_