src/chrome/browser/feedback/feedback_uploader.cc

   1 // Copyright 2013 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 "chrome/browser/feedback/feedback_uploader.h"
   6
   7 #include "base/callback.h"
   8 #include "base/command_line.h"
   9 #include "base/files/file_path.h"
  10 #include "base/task_runner_util.h"
  11 #include "base/threading/sequenced_worker_pool.h"
  12 #include "chrome/browser/feedback/feedback_report.h"
  13 #include "chrome/common/chrome_switches.h"
  14 #include "content/public/browser/browser_context.h"
  15 #include "content/public/browser/browser_thread.h"
  16 #include "net/base/load_flags.h"
  17 #include "net/url_request/url_fetcher.h"
  18 #include "url/gurl.h"
  19
  20 using content::BrowserThread;
  21
  22 namespace feedback {
  23 namespace {
  24
  25 const char kFeedbackPostUrl[] =
  26     "https://www.google.com/tools/feedback/chrome/__submit";
  27 const char kProtBufMimeType[] = "application/x-protobuf";
  28
  29 const int64 kRetryDelayMinutes = 60;
  30
  31 }  // namespace
  32
  33 bool FeedbackUploader::ReportsUploadTimeComparator::operator()(
  34     FeedbackReport* a, FeedbackReport* b) const {
  35   return a->upload_at() > b->upload_at();
  36 }
  37
  38 FeedbackUploader::FeedbackUploader(content::BrowserContext* context)
  39     : context_(context),
  40       retry_delay_(base::TimeDelta::FromMinutes(kRetryDelayMinutes)) {
  41   CHECK(context_);
  42   dispatch_callback_ = base::Bind(&FeedbackUploader::DispatchReport,
  43                                   AsWeakPtr());
  44 }
  45
  46 FeedbackUploader::~FeedbackUploader() {}
  47
  48 void FeedbackUploader::QueueReport(const std::string& data) {
  49   reports_queue_.push(
  50       new FeedbackReport(context_, base::Time::Now(), data));
  51   UpdateUploadTimer();
  52 }
  53
  54 void FeedbackUploader::DispatchReport(const std::string& data) {
  55   GURL post_url;
  56   if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kFeedbackServer))
  57     post_url = GURL(CommandLine::ForCurrentProcess()->GetSwitchValueASCII(
  58         switches::kFeedbackServer));
  59   else
  60     post_url = GURL(kFeedbackPostUrl);
  61
  62   net::URLFetcher* fetcher = net::URLFetcher::Create(
  63       post_url, net::URLFetcher::POST,
  64       new FeedbackUploaderDelegate(
  65           data,
  66           base::Bind(&FeedbackUploader::UpdateUploadTimer, AsWeakPtr()),
  67           base::Bind(&FeedbackUploader::RetryReport, AsWeakPtr())));
  68
  69   fetcher->SetUploadData(std::string(kProtBufMimeType), data);
  70   fetcher->SetRequestContext(context_->GetRequestContext());
  71   fetcher->SetLoadFlags(net::LOAD_DO_NOT_SAVE_COOKIES |
  72                         net::LOAD_DO_NOT_SEND_COOKIES);
  73   fetcher->Start();
  74 }
  75
  76 void FeedbackUploader::UpdateUploadTimer() {
  77   if (reports_queue_.empty())
  78     return;
  79
  80   scoped_refptr<FeedbackReport> report = reports_queue_.top();
  81   base::Time now = base::Time::Now();
  82   if (report->upload_at() <= now) {
  83     reports_queue_.pop();
  84     dispatch_callback_.Run(report->data());
  85     report->DeleteReportOnDisk();
  86   } else {
  87     // Stop the old timer and start an updated one.
  88     if (upload_timer_.IsRunning())
  89       upload_timer_.Stop();
  90     upload_timer_.Start(
  91         FROM_HERE, report->upload_at() - now, this,
  92         &FeedbackUploader::UpdateUploadTimer);
  93   }
  94 }
  95
  96 void FeedbackUploader::RetryReport(const std::string& data) {
  97   reports_queue_.push(new FeedbackReport(context_,
  98                                          base::Time::Now() + retry_delay_,
  99                                          data));
 100   UpdateUploadTimer();
 101 }
 102
 103 void FeedbackUploader::setup_for_test(
 104     const ReportDataCallback& dispatch_callback,
 105     const base::TimeDelta& retry_delay) {
 106   dispatch_callback_ = dispatch_callback;
 107   retry_delay_ = retry_delay;
 108 }
 109
 110 }  // namespace feedback