Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 "base/logging.h" | 5 #include "base/logging.h" |
| 6 #include "chrome/browser/browser_process.h" | |
| 6 #include "chrome/browser/prerender/prerender_manager.h" | 7 #include "chrome/browser/prerender/prerender_manager.h" |
| 7 #include "chrome/browser/prerender/prerender_tracker.h" | 8 #include "chrome/browser/prerender/prerender_tracker.h" |
|
mmenke
2011/05/25 00:56:24
nit: This should go first.
dominich
2011/05/25 16:42:28
Done.
| |
| 8 #include "content/browser/browser_thread.h" | 9 #include "content/browser/browser_thread.h" |
| 10 #include "content/browser/resource_context.h" | |
| 11 #include "content/common/resource_messages.h" | |
| 12 #include "net/base/load_flags.h" | |
| 9 | 13 |
| 10 namespace prerender { | 14 namespace prerender { |
| 11 | 15 |
| 12 namespace { | |
| 13 | |
| 14 void DestroyPreloadForRenderView( | |
| 15 const base::WeakPtr<PrerenderManager>& prerender_manager_weak_ptr, | |
| 16 int child_id, | |
| 17 int route_id, | |
| 18 FinalStatus final_status) { | |
| 19 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
| 20 PrerenderManager* prerender_manager = prerender_manager_weak_ptr.get(); | |
| 21 if (!prerender_manager) | |
| 22 return; | |
| 23 | |
| 24 prerender_manager->DestroyPreloadForChildRouteIdPair( | |
| 25 std::make_pair(child_id, route_id), | |
| 26 final_status); | |
| 27 } | |
| 28 | |
| 29 } // namespace | |
| 30 | |
| 31 struct RenderViewInfo { | 16 struct RenderViewInfo { |
| 32 explicit RenderViewInfo(PrerenderManager* prerender_manager) | 17 explicit RenderViewInfo(PrerenderManager* prerender_manager) |
| 33 : final_status(FINAL_STATUS_MAX), | 18 : final_status(FINAL_STATUS_MAX), |
| 34 prerender_manager(prerender_manager->AsWeakPtr()) { | 19 prerender_manager(prerender_manager->AsWeakPtr()) { |
| 35 } | 20 } |
| 21 ~RenderViewInfo() {} | |
| 36 | 22 |
| 37 FinalStatus final_status; | 23 FinalStatus final_status; |
| 38 base::WeakPtr<PrerenderManager> prerender_manager; | 24 base::WeakPtr<PrerenderManager> prerender_manager; |
| 39 }; | 25 }; |
| 40 | 26 |
| 41 // static | 27 PrerenderTracker::PrerenderTracker() { |
| 42 PrerenderTracker* PrerenderTracker::GetInstance() { | 28 } |
| 43 return Singleton<PrerenderTracker>::get(); | 29 |
| 30 PrerenderTracker::~PrerenderTracker() { | |
| 44 } | 31 } |
| 45 | 32 |
| 46 bool PrerenderTracker::TryUse(int child_id, int route_id) { | 33 bool PrerenderTracker::TryUse(int child_id, int route_id) { |
| 47 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 34 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
| 48 | 35 |
| 49 return SetFinalStatus(child_id, route_id, FINAL_STATUS_USED, NULL); | 36 return SetFinalStatus(child_id, route_id, FINAL_STATUS_USED, NULL); |
| 50 } | 37 } |
| 51 | 38 |
| 52 bool PrerenderTracker::TryCancel( | 39 bool PrerenderTracker::TryCancel( |
| 53 int child_id, | 40 int child_id, |
| (...skipping 13 matching lines...) Expand all Loading... | |
| 67 int route_id, | 54 int route_id, |
| 68 FinalStatus final_status) { | 55 FinalStatus final_status) { |
| 69 DCHECK_NE(FINAL_STATUS_USED, final_status); | 56 DCHECK_NE(FINAL_STATUS_USED, final_status); |
| 70 DCHECK(final_status >= 0 && final_status < FINAL_STATUS_MAX); | 57 DCHECK(final_status >= 0 && final_status < FINAL_STATUS_MAX); |
| 71 | 58 |
| 72 if (!IsPrerenderingOnIOThread(child_id, route_id)) | 59 if (!IsPrerenderingOnIOThread(child_id, route_id)) |
| 73 return false; | 60 return false; |
| 74 return TryCancel(child_id, route_id, final_status); | 61 return TryCancel(child_id, route_id, final_status); |
| 75 } | 62 } |
| 76 | 63 |
| 77 bool PrerenderTracker::IsPrerenderingOnIOThread(int child_id, | |
| 78 int route_id) const { | |
| 79 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | |
| 80 | |
| 81 ChildRouteIdPair child_route_id_pair(child_id, route_id); | |
| 82 return possibly_prerendering_io_thread_set_.end() != | |
| 83 possibly_prerendering_io_thread_set_.find(child_route_id_pair); | |
| 84 } | |
| 85 | |
| 86 bool PrerenderTracker::GetFinalStatus(int child_id, int route_id, | 64 bool PrerenderTracker::GetFinalStatus(int child_id, int route_id, |
| 87 FinalStatus* final_status) const { | 65 FinalStatus* final_status) const { |
| 88 ChildRouteIdPair child_route_id_pair(child_id, route_id); | 66 ChildRouteIdPair child_route_id_pair(child_id, route_id); |
| 89 | 67 |
| 90 base::AutoLock lock(final_status_map_lock_); | 68 base::AutoLock lock(final_status_map_lock_); |
| 91 FinalStatusMap::const_iterator final_status_it = | 69 FinalStatusMap::const_iterator final_status_it = |
| 92 final_status_map_.find(child_route_id_pair); | 70 final_status_map_.find(child_route_id_pair); |
| 93 if (final_status_map_.end() == final_status_map_.find(child_route_id_pair)) | 71 if (final_status_map_.end() == final_status_map_.find(child_route_id_pair)) |
| 94 return false; | 72 return false; |
| 95 *final_status = final_status_it->second.final_status; | 73 *final_status = final_status_it->second.final_status; |
| (...skipping 15 matching lines...) Expand all Loading... | |
| 111 BrowserThread::PostTask( | 89 BrowserThread::PostTask( |
| 112 BrowserThread::IO, FROM_HERE, | 90 BrowserThread::IO, FROM_HERE, |
| 113 NewRunnableFunction(&AddPrerenderOnIOThreadTask, child_route_id_pair)); | 91 NewRunnableFunction(&AddPrerenderOnIOThreadTask, child_route_id_pair)); |
| 114 | 92 |
| 115 base::AutoLock lock(final_status_map_lock_); | 93 base::AutoLock lock(final_status_map_lock_); |
| 116 | 94 |
| 117 final_status_map_.insert( | 95 final_status_map_.insert( |
| 118 std::make_pair(child_route_id_pair, RenderViewInfo(prerender_manager))); | 96 std::make_pair(child_route_id_pair, RenderViewInfo(prerender_manager))); |
| 119 } | 97 } |
| 120 | 98 |
| 121 PrerenderTracker::PrerenderTracker() { | |
| 122 } | |
| 123 | |
| 124 PrerenderTracker::~PrerenderTracker() { | |
| 125 } | |
| 126 | |
| 127 void PrerenderTracker::OnPrerenderingFinished(int child_id, int route_id) { | 99 void PrerenderTracker::OnPrerenderingFinished(int child_id, int route_id) { |
| 128 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 100 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
| 129 DCHECK_GE(child_id, 0); | 101 DCHECK_GE(child_id, 0); |
| 130 DCHECK_GE(route_id, 0); | 102 DCHECK_GE(route_id, 0); |
| 131 | 103 |
| 132 ChildRouteIdPair child_route_id_pair(child_id, route_id); | 104 ChildRouteIdPair child_route_id_pair(child_id, route_id); |
| 133 | 105 |
| 134 BrowserThread::PostTask( | 106 BrowserThread::PostTask( |
| 135 BrowserThread::IO, FROM_HERE, | 107 BrowserThread::IO, FROM_HERE, |
| 136 NewRunnableFunction(&RemovePrerenderOnIOThreadTask, child_route_id_pair)); | 108 NewRunnableFunction(&RemovePrerenderOnIOThreadTask, child_route_id_pair)); |
| (...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 173 if (actual_final_status) | 145 if (actual_final_status) |
| 174 *actual_final_status = desired_final_status; | 146 *actual_final_status = desired_final_status; |
| 175 return true; | 147 return true; |
| 176 } | 148 } |
| 177 | 149 |
| 178 if (actual_final_status) | 150 if (actual_final_status) |
| 179 *actual_final_status = final_status_it->second.final_status; | 151 *actual_final_status = final_status_it->second.final_status; |
| 180 return false; | 152 return false; |
| 181 } | 153 } |
| 182 | 154 |
| 155 bool PrerenderTracker::IsPrerenderingOnIOThread(int child_id, | |
| 156 int route_id) const { | |
| 157 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | |
| 158 | |
| 159 ChildRouteIdPair child_route_id_pair(child_id, route_id); | |
| 160 return possibly_prerendering_io_thread_set_.end() != | |
| 161 possibly_prerendering_io_thread_set_.find(child_route_id_pair); | |
| 162 } | |
| 163 | |
| 183 void PrerenderTracker::AddPrerenderOnIOThread( | 164 void PrerenderTracker::AddPrerenderOnIOThread( |
| 184 const ChildRouteIdPair& child_route_id_pair) { | 165 const ChildRouteIdPair& child_route_id_pair) { |
| 185 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | 166 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
| 186 DCHECK(!IsPrerenderingOnIOThread(child_route_id_pair.first, | 167 DCHECK(!IsPrerenderingOnIOThread(child_route_id_pair.first, |
| 187 child_route_id_pair.second)); | 168 child_route_id_pair.second)); |
| 188 | 169 |
| 189 possibly_prerendering_io_thread_set_.insert(child_route_id_pair); | 170 possibly_prerendering_io_thread_set_.insert(child_route_id_pair); |
| 190 } | 171 } |
| 191 | 172 |
| 192 void PrerenderTracker::RemovePrerenderOnIOThread( | 173 void PrerenderTracker::RemovePrerenderOnIOThread( |
| 193 const ChildRouteIdPair& child_route_id_pair) { | 174 const ChildRouteIdPair& child_route_id_pair) { |
| 194 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | 175 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
| 195 DCHECK(IsPrerenderingOnIOThread(child_route_id_pair.first, | 176 DCHECK(IsPrerenderingOnIOThread(child_route_id_pair.first, |
| 196 child_route_id_pair.second)); | 177 child_route_id_pair.second)); |
| 197 | 178 |
| 198 possibly_prerendering_io_thread_set_.erase(child_route_id_pair); | 179 possibly_prerendering_io_thread_set_.erase(child_route_id_pair); |
| 199 } | 180 } |
| 200 | 181 |
| 201 // static | 182 // static |
| 202 void PrerenderTracker::AddPrerenderOnIOThreadTask( | 183 void PrerenderTracker::AddPrerenderOnIOThreadTask( |
| 203 const ChildRouteIdPair& child_route_id_pair) { | 184 const ChildRouteIdPair& child_route_id_pair) { |
| 204 GetInstance()->AddPrerenderOnIOThread(child_route_id_pair); | 185 g_browser_process->prerender_tracker()->AddPrerenderOnIOThread( |
| 186 child_route_id_pair); | |
| 205 } | 187 } |
| 206 | 188 |
| 207 // static | 189 // static |
| 208 void PrerenderTracker::RemovePrerenderOnIOThreadTask( | 190 void PrerenderTracker::RemovePrerenderOnIOThreadTask( |
| 209 const ChildRouteIdPair& child_route_id_pair) { | 191 const ChildRouteIdPair& child_route_id_pair) { |
| 210 GetInstance()->RemovePrerenderOnIOThread(child_route_id_pair); | 192 g_browser_process->prerender_tracker()->RemovePrerenderOnIOThread( |
| 193 child_route_id_pair); | |
| 211 } | 194 } |
| 212 | 195 |
| 213 } // namespace prerender | 196 } // namespace prerender |
| OLD | NEW |