Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 1 #include "chrome/renderer/prerender/prerender_dispatcher.h" | |
| 2 | |
| 3 #include "base/logging.h" | |
| 4 #include "chrome/common/prerender_messages.h" | |
| 5 #include "googleurl/src/gurl.h" | |
| 6 | |
| 7 namespace prerender { | |
| 8 | |
| 9 bool PrerenderDispatcher::IsPrerenderURL(const GURL& url) const { | |
| 10 return prerender_urls_.find(url) != prerender_urls_.end(); | |
| 11 } | |
| 12 | |
| 13 bool PrerenderDispatcher::OnControlMessageReceived(const IPC::Message& message) { | |
| 14 bool handled = true; | |
| 15 IPC_BEGIN_MESSAGE_MAP(PrerenderDispatcher, message) | |
| 16 IPC_MESSAGE_HANDLER(PrerenderMsg_AddPrerenderURL, OnAddPrerenderURL) | |
| 17 IPC_MESSAGE_HANDLER(PrerenderMsg_RemovePrerenderURL, OnRemovePrerenderURL) | |
| 18 IPC_MESSAGE_UNHANDLED(handled = false) | |
| 19 IPC_END_MESSAGE_MAP() | |
| 20 | |
| 21 return handled; | |
| 22 } | |
| 23 | |
| 24 void PrerenderDispatcher::OnAddPrerenderURL(const GURL& url) { | |
| 25 PrerenderMap::iterator it = prerender_urls_.find(url); | |
| 26 if (it != prerender_urls_.end()) { | |
| 27 DCHECK(it->second > 0); | |
| 28 it->second += 1; | |
| 29 } else { | |
| 30 prerender_urls_[url] = 1; | |
| 31 } | |
| 32 } | |
| 33 | |
| 34 void PrerenderDispatcher::OnRemovePrerenderURL(const GURL& url) { | |
| 35 PrerenderMap::iterator it = prerender_urls_.find(url); | |
| 36 // This is possible with a spurious remove. | |
| 37 // TODO(cbentzel): We'd also want to send the map of active prerenders when | |
| 38 // creating a new render process, so the Add/Remove go relative to that. This may | |
| 39 // not be that big of a deal in practice, since the newly created tab is unlik ely | |
| 40 // to go to the prerendered page. | |
| 41 if (it == prerender_urls_.end()) | |
| 42 return; | |
| 43 DCHECK(it->second > 0); | |
| 44 if (it->second > 1) { | |
|
dominich
2012/02/28 16:21:00
this could also be
it->second -= 1;
if (it->secon
cbentzel
2012/02/29 18:16:13
I like it. Done.
| |
| 45 it->second -= 1; | |
| 46 } else { | |
| 47 prerender_urls_.erase(it); | |
| 48 } | |
| 49 } | |
| 50 | |
| 51 } // namespace prerender | |
| OLD | NEW |