OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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/net/sdch_dictionary_fetcher.h" | 5 #include "chrome/browser/net/sdch_dictionary_fetcher.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/compiler_specific.h" | 8 #include "base/compiler_specific.h" |
9 #include "base/message_loop/message_loop.h" | 9 #include "base/message_loop/message_loop.h" |
10 #include "chrome/browser/profiles/profile.h" | 10 #include "chrome/browser/profiles/profile.h" |
(...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
42 if (attempted_load_.find(dictionary_url) != attempted_load_.end()) { | 42 if (attempted_load_.find(dictionary_url) != attempted_load_.end()) { |
43 net::SdchManager::SdchErrorRecovery( | 43 net::SdchManager::SdchErrorRecovery( |
44 net::SdchManager::DICTIONARY_ALREADY_TRIED_TO_DOWNLOAD); | 44 net::SdchManager::DICTIONARY_ALREADY_TRIED_TO_DOWNLOAD); |
45 return; | 45 return; |
46 } | 46 } |
47 attempted_load_.insert(dictionary_url); | 47 attempted_load_.insert(dictionary_url); |
48 fetch_queue_.push(dictionary_url); | 48 fetch_queue_.push(dictionary_url); |
49 ScheduleDelayedRun(); | 49 ScheduleDelayedRun(); |
50 } | 50 } |
51 | 51 |
| 52 void SdchDictionaryFetcher::Cancel() { |
| 53 DCHECK(CalledOnValidThread()); |
| 54 |
| 55 while (!fetch_queue_.empty()) |
| 56 fetch_queue_.pop(); |
| 57 attempted_load_.clear(); |
| 58 weak_factory_.InvalidateWeakPtrs(); |
| 59 current_fetch_.reset(NULL); |
| 60 } |
| 61 |
52 void SdchDictionaryFetcher::ScheduleDelayedRun() { | 62 void SdchDictionaryFetcher::ScheduleDelayedRun() { |
53 if (fetch_queue_.empty() || current_fetch_.get() || task_is_pending_) | 63 if (fetch_queue_.empty() || current_fetch_.get() || task_is_pending_) |
54 return; | 64 return; |
55 base::MessageLoop::current()->PostDelayedTask(FROM_HERE, | 65 base::MessageLoop::current()->PostDelayedTask(FROM_HERE, |
56 base::Bind(&SdchDictionaryFetcher::StartFetching, | 66 base::Bind(&SdchDictionaryFetcher::StartFetching, |
57 weak_factory_.GetWeakPtr()), | 67 weak_factory_.GetWeakPtr()), |
58 base::TimeDelta::FromMilliseconds(kMsDelayFromRequestTillDownload)); | 68 base::TimeDelta::FromMilliseconds(kMsDelayFromRequestTillDownload)); |
59 task_is_pending_ = true; | 69 task_is_pending_ = true; |
60 } | 70 } |
61 | 71 |
62 void SdchDictionaryFetcher::StartFetching() { | 72 void SdchDictionaryFetcher::StartFetching() { |
63 DCHECK(CalledOnValidThread()); | 73 DCHECK(CalledOnValidThread()); |
64 DCHECK(task_is_pending_); | 74 DCHECK(task_is_pending_); |
65 task_is_pending_ = false; | 75 task_is_pending_ = false; |
66 | 76 |
| 77 // Handle losing the race against Cancel(). |
| 78 if (fetch_queue_.empty()) |
| 79 return; |
| 80 |
67 DCHECK(context_.get()); | 81 DCHECK(context_.get()); |
68 current_fetch_.reset(net::URLFetcher::Create( | 82 current_fetch_.reset(net::URLFetcher::Create( |
69 fetch_queue_.front(), net::URLFetcher::GET, this)); | 83 fetch_queue_.front(), net::URLFetcher::GET, this)); |
70 fetch_queue_.pop(); | 84 fetch_queue_.pop(); |
71 current_fetch_->SetRequestContext(context_.get()); | 85 current_fetch_->SetRequestContext(context_.get()); |
72 current_fetch_->SetLoadFlags(net::LOAD_DO_NOT_SEND_COOKIES | | 86 current_fetch_->SetLoadFlags(net::LOAD_DO_NOT_SEND_COOKIES | |
73 net::LOAD_DO_NOT_SAVE_COOKIES); | 87 net::LOAD_DO_NOT_SAVE_COOKIES); |
74 current_fetch_->Start(); | 88 current_fetch_->Start(); |
75 } | 89 } |
76 | 90 |
77 void SdchDictionaryFetcher::OnURLFetchComplete( | 91 void SdchDictionaryFetcher::OnURLFetchComplete( |
78 const net::URLFetcher* source) { | 92 const net::URLFetcher* source) { |
79 DCHECK(CalledOnValidThread()); | 93 DCHECK(CalledOnValidThread()); |
80 if ((200 == source->GetResponseCode()) && | 94 if ((200 == source->GetResponseCode()) && |
81 (source->GetStatus().status() == net::URLRequestStatus::SUCCESS)) { | 95 (source->GetStatus().status() == net::URLRequestStatus::SUCCESS)) { |
82 std::string data; | 96 std::string data; |
83 source->GetResponseAsString(&data); | 97 source->GetResponseAsString(&data); |
84 manager_->AddSdchDictionary(data, source->GetURL()); | 98 manager_->AddSdchDictionary(data, source->GetURL()); |
85 } | 99 } |
86 current_fetch_.reset(NULL); | 100 current_fetch_.reset(NULL); |
87 ScheduleDelayedRun(); | 101 ScheduleDelayedRun(); |
88 } | 102 } |
OLD | NEW |