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 |