Adding opt out and previews type information to DRP pingback

components/data_reduction_proxy/core/browser/data_reduction_proxy_pingback_client.h

 // Copyright 2016 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_REDUCTION_PROXY_CORE_BROWSER_DATA_REDUCTION_PROXY_PINGBACK_CLIENT_H_
 #define COMPONENTS_DATA_REDUCTION_PROXY_CORE_BROWSER_DATA_REDUCTION_PROXY_PINGBACK_CLIENT_H_
 
+#include <stdint.h>
+
 #include <memory>
 #include <string>
 
 #include "base/macros.h"
 #include "base/threading/thread_checker.h"
 #include "components/data_reduction_proxy/proto/pageload_metrics.pb.h"
 #include "net/url_request/url_fetcher_delegate.h"
 #include "url/gurl.h"
 
 namespace base {
@@ skipping 15 matching lines @@
  public:
   // The caller must ensure that |url_request_context| remains alive for the
   // lifetime of the |DataReductionProxyPingbackClient| instance.
   explicit DataReductionProxyPingbackClient(
       net::URLRequestContextGetter* url_request_context);
   ~DataReductionProxyPingbackClient() override;
 
   // Sends a pingback to the data saver proxy server about various timing
   // information.
   virtual void SendPingback(const DataReductionProxyData& data,
-                            const DataReductionProxyPageLoadTiming& timing);
+                            const DataReductionProxyPageLoadTiming& timing,
+                            const void* tab_identifier_key);
 
   // Sets the probability of actually sending a pingback to the server for any
   // call to SendPingback.
   void SetPingbackReportingFraction(float pingback_reporting_fraction);
 
+  // Adds an opt out for |tab_identifier_key| for a data saver |page_id|.

[bengr, 2017/04/20 17:36:19]

What's an opt out? Define somewhere.

[RyanSturm, 2017/04/20 20:25:45]

Done.

+  void AddOptOut(const void* tab_identifier_key, uint64_t page_id);
+
+  // Removes any stored data associated with |tab_identifier_key| in a task that
+  // runs later.
+  void ClearTabKeyAsync(const void* tab_identifier_key);
+
+  // The total number of pending loads being tracked due to opt outs.
+  size_t PendingTabLoadsForTesting() const;
+
  protected:
   // Generates a float in the range [0, 1). Virtualized in testing.
   virtual float GenerateRandomFloat() const;
 
   // Returns the current time. Virtualized in testing.
   virtual base::Time CurrentTime() const;
 
  private:
   // URLFetcherDelegate implmentation:
   void OnURLFetchComplete(const net::URLFetcher* source) override;
 
   // Whether a pingback should be sent.
   bool ShouldSendPingback() const;
 
   // Creates an URLFetcher that will POST to |secure_proxy_url_| using
   // |url_request_context_|. The max retries is set to 5.
   // |data_to_send_| will be used to fill the body of the Fetcher, and will be
   // reset to an empty RecordPageloadMetricsRequest.
   void CreateFetcherForDataAndStart();
 
+  // Removes any stored data associated with |tab_identifier_key|.
+  void ClearTabKeySync(const void* tab_identifier_key);
+
   net::URLRequestContextGetter* url_request_context_;
 
   // The URL for the data saver proxy's ping back service.
   const GURL pingback_url_;
 
   // The currently running fetcher.
   std::unique_ptr<net::URLFetcher> current_fetcher_;
 
   // Serialized data to send to the data saver proxy server.
   RecordPageloadMetricsRequest metrics_request_;
 
   // The probability of sending a pingback to the server.
   float pingback_reporting_fraction_;
 
+  // The map of tab identifier keys to page IDs.
+  std::map<const void*, uint64_t> tab_page_id_map_;
+
   base::ThreadChecker thread_checker_;
 
   DISALLOW_COPY_AND_ASSIGN(DataReductionProxyPingbackClient);
 };
 
 }  // namespace data_reduction_proxy
 
 #endif  // COMPONENTS_DATA_REDUCTION_PROXY_CORE_BROWSER_DATA_REDUCTION_PROXY_PINGBACK_CLIENT_H_