Home | History | Annotate | Download | only in translate 
      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/translate/translate_url_fetcher.h"
      6 
      7 #include "chrome/browser/browser_process.h"
      8 #include "net/base/load_flags.h"
      9 #include "net/http/http_status_code.h"
     10 #include "net/url_request/url_fetcher.h"
     11 #include "net/url_request/url_request_status.h"
     12 
     13 namespace {
     14 
     15 // Retry parameter for fetching.
     16 const int kMaxRetry = 16;
     17 
     18 }  // namespace
     19 
     20 TranslateURLFetcher::TranslateURLFetcher(int id)
     21     : id_(id),
     22       state_(IDLE),
     23       retry_count_(0) {
     24 }
     25 
     26 TranslateURLFetcher::~TranslateURLFetcher() {
     27 }
     28 
     29 bool TranslateURLFetcher::Request(
     30     const GURL& url,
     31     const TranslateURLFetcher::Callback& callback) {
     32   // This function is not supposed to be called before previous operaion is not
     33   // finished.
     34   if (state_ == REQUESTING) {
     35     NOTREACHED();
     36     return false;
     37   }
     38 
     39   if (retry_count_ >= kMaxRetry)
     40     return false;
     41   retry_count_++;
     42 
     43   state_ = REQUESTING;
     44   url_ = url;
     45   callback_ = callback;
     46 
     47   fetcher_.reset(net::URLFetcher::Create(
     48       id_,
     49       url_,
     50       net::URLFetcher::GET,
     51       this));
     52   fetcher_->SetLoadFlags(net::LOAD_DO_NOT_SEND_COOKIES |
     53                          net::LOAD_DO_NOT_SAVE_COOKIES);
     54   fetcher_->SetRequestContext(g_browser_process->system_request_context());
     55   // Set retry parameter for HTTP status code 5xx. This doesn't work against
     56   // 106 (net::ERR_INTERNET_DISCONNECTED) and so on.
     57   // TranslateLanguageList handles network status, and implements retry.
     58   fetcher_->SetMaxRetriesOn5xx(max_retry_on_5xx_);
     59   if (!extra_request_header_.empty())
     60     fetcher_->SetExtraRequestHeaders(extra_request_header_);
     61 
     62   fetcher_->Start();
     63 
     64   return true;
     65 }
     66 
     67 void TranslateURLFetcher::OnURLFetchComplete(const net::URLFetcher* source) {
     68   DCHECK(fetcher_.get() == source);
     69 
     70   std::string data;
     71   if (source->GetStatus().status() == net::URLRequestStatus::SUCCESS &&
     72       source->GetResponseCode() == net::HTTP_OK) {
     73     state_ = COMPLETED;
     74     source->GetResponseAsString(&data);
     75   } else {
     76     state_ = FAILED;
     77   }
     78 
     79   // Transfer URLFetcher's ownership before invoking a callback.
     80   scoped_ptr<const net::URLFetcher> delete_ptr(fetcher_.release());
     81   callback_.Run(id_, state_ == COMPLETED, data);
     82 }
     83