components/domain_reliability/uploader.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "components/domain_reliability/uploader.h"
   6
   7 #include "base/bind.h"
   8 #include "base/callback.h"
   9 #include "base/memory/scoped_vector.h"
  10 #include "base/metrics/sparse_histogram.h"
  11 #include "base/stl_util.h"
  12 #include "base/supports_user_data.h"
  13 #include "net/base/load_flags.h"
  14 #include "net/url_request/url_fetcher.h"
  15 #include "net/url_request/url_fetcher_delegate.h"
  16 #include "net/url_request/url_request_context_getter.h"
  17
  18 namespace domain_reliability {
  19
  20 namespace {
  21
  22 const char* kJsonMimeType = "application/json; charset=utf-8";
  23
  24 class UploadUserData : public base::SupportsUserData::Data {
  25  public:
  26   static net::URLFetcher::CreateDataCallback CreateCreateDataCallback() {
  27     return base::Bind(&UploadUserData::CreateUploadUserData);
  28   }
  29
  30   static const void* kUserDataKey;
  31
  32  private:
  33   static base::SupportsUserData::Data* CreateUploadUserData() {
  34     return new UploadUserData();
  35   }
  36 };
  37
  38 const void* UploadUserData::kUserDataKey =
  39     static_cast<const void*>(&UploadUserData::kUserDataKey);
  40
  41 class DomainReliabilityUploaderImpl
  42     : public DomainReliabilityUploader, net::URLFetcherDelegate {
  43  public:
  44   DomainReliabilityUploaderImpl(const scoped_refptr<
  45       net::URLRequestContextGetter>& url_request_context_getter)
  46       : url_request_context_getter_(url_request_context_getter),
  47         discard_uploads_(true) {}
  48
  49   virtual ~DomainReliabilityUploaderImpl() {
  50     // Delete any in-flight URLFetchers.
  51     STLDeleteContainerPairFirstPointers(
  52         upload_callbacks_.begin(), upload_callbacks_.end());
  53   }
  54
  55   // DomainReliabilityUploader implementation:
  56   virtual void UploadReport(
  57       const std::string& report_json,
  58       const GURL& upload_url,
  59       const DomainReliabilityUploader::UploadCallback& callback) OVERRIDE {
  60     VLOG(1) << "Uploading report to " << upload_url;
  61     VLOG(2) << "Report JSON: " << report_json;
  62
  63     if (discard_uploads_) {
  64       VLOG(1) << "Discarding report instead of uploading.";
  65       callback.Run(true);
  66       return;
  67     }
  68
  69     net::URLFetcher* fetcher =
  70         net::URLFetcher::Create(0, upload_url, net::URLFetcher::POST, this);
  71     fetcher->SetRequestContext(url_request_context_getter_.get());
  72     fetcher->SetLoadFlags(net::LOAD_DO_NOT_SEND_COOKIES |
  73                           net::LOAD_DO_NOT_SAVE_COOKIES);
  74     fetcher->SetUploadData(kJsonMimeType, report_json);
  75     fetcher->SetAutomaticallyRetryOn5xx(false);
  76     fetcher->SetURLRequestUserData(
  77         UploadUserData::kUserDataKey,
  78         UploadUserData::CreateCreateDataCallback());
  79     fetcher->Start();
  80
  81     upload_callbacks_[fetcher] = callback;
  82   }
  83
  84   virtual void set_discard_uploads(bool discard_uploads) OVERRIDE {
  85     discard_uploads_ = discard_uploads;
  86     VLOG(1) << "Setting discard_uploads to " << discard_uploads;
  87   }
  88
  89   // net::URLFetcherDelegate implementation:
  90   virtual void OnURLFetchComplete(
  91       const net::URLFetcher* fetcher) OVERRIDE {
  92     DCHECK(fetcher);
  93
  94     UploadCallbackMap::iterator callback_it = upload_callbacks_.find(fetcher);
  95     DCHECK(callback_it != upload_callbacks_.end());
  96
  97     VLOG(1) << "Upload finished with " << fetcher->GetResponseCode();
  98
  99     UMA_HISTOGRAM_SPARSE_SLOWLY("DomainReliability.UploadResponseCode",
 100                                 fetcher->GetResponseCode());
 101
 102     bool success = fetcher->GetResponseCode() == 200;
 103     callback_it->second.Run(success);
 104
 105     delete callback_it->first;
 106     upload_callbacks_.erase(callback_it);
 107   }
 108
 109  private:
 110   using DomainReliabilityUploader::UploadCallback;
 111   typedef std::map<const net::URLFetcher*, UploadCallback> UploadCallbackMap;
 112
 113   scoped_refptr<net::URLRequestContextGetter> url_request_context_getter_;
 114   UploadCallbackMap upload_callbacks_;
 115   bool discard_uploads_;
 116 };
 117
 118 }  // namespace
 119
 120 DomainReliabilityUploader::DomainReliabilityUploader() {}
 121 DomainReliabilityUploader::~DomainReliabilityUploader() {}
 122
 123 // static
 124 scoped_ptr<DomainReliabilityUploader> DomainReliabilityUploader::Create(
 125     const scoped_refptr<net::URLRequestContextGetter>&
 126         url_request_context_getter) {
 127   return scoped_ptr<DomainReliabilityUploader>(
 128       new DomainReliabilityUploaderImpl(url_request_context_getter));
 129 }
 130
 131 // static
 132 bool DomainReliabilityUploader::URLRequestIsUpload(
 133     const net::URLRequest& request) {
 134   return request.GetUserData(UploadUserData::kUserDataKey) != NULL;
 135 }
 136
 137 }  // namespace domain_reliability