OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/translate/translate_url_fetcher.h" | 5 #include "chrome/browser/translate/translate_url_fetcher.h" |
6 | 6 |
7 #include "chrome/browser/browser_process.h" | 7 #include "chrome/browser/browser_process.h" |
8 #include "net/base/load_flags.h" | 8 #include "net/base/load_flags.h" |
9 #include "net/http/http_status_code.h" | 9 #include "net/http/http_status_code.h" |
10 #include "net/url_request/url_fetcher.h" | 10 #include "net/url_request/url_fetcher.h" |
11 #include "net/url_request/url_request_status.h" | 11 #include "net/url_request/url_request_status.h" |
12 | 12 |
13 namespace { | |
14 | |
15 // Retry parameter for fetching. | |
16 const int kMaxRetry = 5; | |
17 | |
18 } // namespace | |
19 | |
20 TranslateURLFetcher::TranslateURLFetcher(int id) | 13 TranslateURLFetcher::TranslateURLFetcher(int id) |
21 : id_(id), | 14 : id_(id), |
22 state_(IDLE) { | 15 state_(IDLE), |
| 16 retry_count_(0) { |
23 } | 17 } |
24 | 18 |
25 TranslateURLFetcher::~TranslateURLFetcher() { | 19 TranslateURLFetcher::~TranslateURLFetcher() { |
26 } | 20 } |
27 | 21 |
28 bool TranslateURLFetcher::Request( | 22 bool TranslateURLFetcher::Request( |
29 const GURL& url, | 23 const GURL& url, |
30 const TranslateURLFetcher::Callback& callback) { | 24 const TranslateURLFetcher::Callback& callback) { |
31 // This function is not supporsed to be called before previous operaion is not | 25 // This function is not supporsed to be called before previous operaion is not |
32 // finished. | 26 // finished. |
33 if (state_ == REQUESTING) { | 27 if (state_ == REQUESTING) { |
34 NOTREACHED(); | 28 NOTREACHED(); |
35 return false; | 29 return false; |
36 } | 30 } |
37 | 31 |
38 state_ = REQUESTING; | 32 state_ = REQUESTING; |
39 url_ = url; | 33 url_ = url; |
40 callback_ = callback; | 34 callback_ = callback; |
41 | 35 |
42 fetcher_.reset(net::URLFetcher::Create( | 36 fetcher_.reset(net::URLFetcher::Create( |
43 id_, | 37 id_, |
44 url_, | 38 url_, |
45 net::URLFetcher::GET, | 39 net::URLFetcher::GET, |
46 this)); | 40 this)); |
47 fetcher_->SetLoadFlags(net::LOAD_DO_NOT_SEND_COOKIES | | 41 fetcher_->SetLoadFlags(net::LOAD_DO_NOT_SEND_COOKIES | |
48 net::LOAD_DO_NOT_SAVE_COOKIES); | 42 net::LOAD_DO_NOT_SAVE_COOKIES); |
49 fetcher_->SetRequestContext(g_browser_process->system_request_context()); | 43 fetcher_->SetRequestContext(g_browser_process->system_request_context()); |
50 fetcher_->SetMaxRetriesOn5xx(kMaxRetry); | 44 fetcher_->SetMaxRetriesOn5xx(retry_count_); |
| 45 if (!extra_request_header_.empty()) |
| 46 fetcher_->SetExtraRequestHeaders(extra_request_header_); |
| 47 |
51 fetcher_->Start(); | 48 fetcher_->Start(); |
52 | 49 |
53 return true; | 50 return true; |
54 } | 51 } |
55 | 52 |
56 void TranslateURLFetcher::OnURLFetchComplete(const net::URLFetcher* source) { | 53 void TranslateURLFetcher::OnURLFetchComplete(const net::URLFetcher* source) { |
57 DCHECK(fetcher_.get() == source); | 54 DCHECK(fetcher_.get() == source); |
58 | 55 |
59 std::string data; | 56 std::string data; |
60 if (source->GetStatus().status() == net::URLRequestStatus::SUCCESS && | 57 if (source->GetStatus().status() == net::URLRequestStatus::SUCCESS && |
61 source->GetResponseCode() == net::HTTP_OK) { | 58 source->GetResponseCode() == net::HTTP_OK) { |
62 state_ = COMPLETED; | 59 state_ = COMPLETED; |
63 source->GetResponseAsString(&data); | 60 source->GetResponseAsString(&data); |
64 } else { | 61 } else { |
65 state_ = FAILED; | 62 state_ = FAILED; |
66 } | 63 } |
67 | 64 |
68 scoped_ptr<const net::URLFetcher> delete_ptr(fetcher_.release()); | 65 scoped_ptr<const net::URLFetcher> delete_ptr(fetcher_.release()); |
69 callback_.Run(id_, state_ == COMPLETED, data); | 66 callback_.Run(id_, state_ == COMPLETED, data); |
70 } | 67 } |
OLD | NEW |