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/renderer/prerender/prerender_dispatcher.h" | 5 #include "chrome/renderer/prerender/prerender_dispatcher.h" |
6 | 6 |
7 #include "base/logging.h" | 7 #include "base/logging.h" |
8 #include "chrome/common/prerender_messages.h" | 8 #include "chrome/common/prerender_messages.h" |
9 #include "chrome/renderer/prerender/prerendering_support.h" | 9 #include "chrome/renderer/prerender/prerender_extra_data.h" |
10 #include "content/public/common/referrer.h" | |
11 #include "content/public/renderer/render_thread.h" | |
12 #include "content/public/renderer/render_view.h" | |
10 #include "googleurl/src/gurl.h" | 13 #include "googleurl/src/gurl.h" |
14 #include "third_party/WebKit/Source/Platform/chromium/public/WebString.h" | |
15 #include "third_party/WebKit/Source/Platform/chromium/public/WebURL.h" | |
11 #include "third_party/WebKit/Source/Platform/chromium/public/WebPrerenderingSupp ort.h" | 16 #include "third_party/WebKit/Source/Platform/chromium/public/WebPrerenderingSupp ort.h" |
12 | 17 |
13 namespace prerender { | 18 namespace prerender { |
14 | 19 |
15 PrerenderDispatcher::PrerenderDispatcher() | 20 using WebKit::WebPrerender; |
16 : prerendering_support_(new PrerenderingSupport()) { | 21 using WebKit::WebPrerenderingSupport; |
17 WebKit::WebPrerenderingSupport::initialize(prerendering_support_.get()); | 22 |
23 PrerenderDispatcher::PrerenderDispatcher() { | |
24 WebPrerenderingSupport::initialize(this); | |
18 } | 25 } |
19 | 26 |
20 PrerenderDispatcher::~PrerenderDispatcher() { | 27 PrerenderDispatcher::~PrerenderDispatcher() { |
21 if (prerendering_support_.get()) | 28 WebPrerenderingSupport::shutdown(); |
22 WebKit::WebPrerenderingSupport::shutdown(); | |
23 } | 29 } |
24 | 30 |
25 bool PrerenderDispatcher::IsPrerenderURL(const GURL& url) const { | 31 bool PrerenderDispatcher::IsPrerenderURL(const GURL& url) const { |
26 return prerender_urls_.find(url) != prerender_urls_.end(); | 32 for (std::map<int, WebKit::WebPrerender*>::const_iterator |
33 it = prerenders_.begin(); | |
34 it != prerenders_.end(); ++it) { | |
35 if (it->second->url() == url) | |
36 return true; | |
37 } | |
38 for (std::multimap<int, GURL>::const_iterator it = prerender_aliases_.begin(); | |
39 it != prerender_aliases_.end(); ++it) { | |
40 if (it->second == url) | |
41 return true; | |
42 } | |
43 return false; | |
27 } | 44 } |
28 | 45 |
29 bool PrerenderDispatcher::OnControlMessageReceived( | 46 bool PrerenderDispatcher::OnControlMessageReceived( |
30 const IPC::Message& message) { | 47 const IPC::Message& message) { |
31 bool handled = true; | 48 bool handled = true; |
32 IPC_BEGIN_MESSAGE_MAP(PrerenderDispatcher, message) | 49 IPC_BEGIN_MESSAGE_MAP(PrerenderDispatcher, message) |
33 IPC_MESSAGE_HANDLER(PrerenderMsg_AddPrerenderURL, OnAddPrerenderURL) | 50 IPC_MESSAGE_HANDLER(PrerenderMsg_StartedPrerender, OnStartedPrerender) |
34 IPC_MESSAGE_HANDLER(PrerenderMsg_RemovePrerenderURL, OnRemovePrerenderURL) | 51 IPC_MESSAGE_HANDLER(PrerenderMsg_AddAliasToPrerender, OnAddAliasToPrerender) |
52 IPC_MESSAGE_HANDLER(PrerenderMsg_StoppedPrerender, OnStoppedPrerender) | |
35 IPC_MESSAGE_UNHANDLED(handled = false) | 53 IPC_MESSAGE_UNHANDLED(handled = false) |
36 IPC_END_MESSAGE_MAP() | 54 IPC_END_MESSAGE_MAP() |
37 | 55 |
38 return handled; | 56 return handled; |
39 } | 57 } |
40 | 58 |
41 void PrerenderDispatcher::OnAddPrerenderURL(const GURL& url) { | 59 void PrerenderDispatcher::OnStartedPrerender(int prerender_id) { |
42 PrerenderMap::iterator it = prerender_urls_.find(url); | 60 running_prerenders_.insert(prerender_id); |
43 if (it != prerender_urls_.end()) { | 61 |
44 DCHECK(it->second > 0); | 62 std::map<int, WebPrerender*>::iterator it = prerenders_.find(prerender_id); |
45 it->second += 1; | 63 if (it != prerenders_.end()) { |
dominich
2012/09/12 15:21:04
this would mean that we started a prerender before
| |
46 } else { | 64 it->second->sendPrerenderStatusEvent( |
47 prerender_urls_[url] = 1; | 65 WebKit::WebString::fromUTF8("started")); |
48 } | 66 } |
49 } | 67 } |
50 | 68 |
51 void PrerenderDispatcher::OnRemovePrerenderURL(const GURL& url) { | 69 void PrerenderDispatcher::OnAddAliasToPrerender(int prerender_id, |
52 PrerenderMap::iterator it = prerender_urls_.find(url); | 70 const GURL& url) { |
53 // This is possible with a spurious remove. | 71 prerender_aliases_.insert( |
dominich
2012/09/12 15:21:04
are there any invariants expected here - such as t
| |
72 std::multimap<int, GURL>::value_type(prerender_id, url)); | |
73 } | |
74 | |
75 void PrerenderDispatcher::OnStoppedPrerender(int prerender_id) { | |
76 // It's possible we will receive a spurious remove, so we don't crash in that | |
77 // situation. | |
78 | |
54 // TODO(cbentzel): We'd also want to send the map of active prerenders when | 79 // TODO(cbentzel): We'd also want to send the map of active prerenders when |
55 // creating a new render process, so the Add/Remove go relative to that. | 80 // creating a new render process, so the Add/Remove go relative to that. |
56 // This may not be that big of a deal in practice, since the newly created tab | 81 // This may not be that big of a deal in practice, since the newly created tab |
57 // is unlikely to go to the prerendered page. | 82 // is unlikely to go to the prerendered page. |
58 if (it == prerender_urls_.end()) | 83 |
59 return; | 84 running_prerenders_.erase(prerender_id); |
60 DCHECK(it->second > 0); | 85 prerender_aliases_.erase(prerender_id); |
61 it->second -= 1; | 86 |
62 if (it->second == 0) | 87 std::map<int, WebPrerender*>::iterator it = prerenders_.find(prerender_id); |
63 prerender_urls_.erase(it); | 88 if (it != prerenders_.end()) { |
89 it->second->sendPrerenderStatusEvent( | |
90 WebKit::WebString::fromUTF8("stopped")); | |
91 } | |
92 } | |
93 | |
94 void PrerenderDispatcher::add(WebPrerender* prerender) { | |
95 const PrerenderExtraData& extra_data = | |
96 PrerenderExtraData::FromPrerender(*prerender); | |
97 prerenders_.insert(std::map<int, WebPrerender*>::value_type( | |
98 extra_data.prerender_id(), prerender)); | |
99 content::RenderThread::Get()->Send(new PrerenderHostMsg_AddLinkRelPrerender( | |
100 extra_data.prerender_id(), GURL(prerender->url()), | |
101 content::Referrer(GURL(prerender->referrer()), | |
102 prerender->referrerPolicy()), | |
103 extra_data.size(), extra_data.render_view_route_id())); | |
mmenke
2012/09/12 16:31:45
So if the same prerender is added twice, then the
| |
104 } | |
105 | |
106 void PrerenderDispatcher::cancel(WebPrerender* prerender) { | |
107 const PrerenderExtraData& extra_data = | |
108 PrerenderExtraData::FromPrerender(*prerender); | |
109 std::map<int, WebPrerender*>::iterator it = | |
110 prerenders_.find(extra_data.prerender_id()); | |
111 if (it != prerenders_.end()) | |
112 prerenders_.erase(it); | |
mmenke
2012/09/12 16:31:45
Is the find / check / erase needed?
Doesn't just
| |
113 content::RenderThread::Get()->Send( | |
114 new PrerenderHostMsg_CancelLinkRelPrerender(extra_data.prerender_id())); | |
115 } | |
116 | |
117 void PrerenderDispatcher::abandon(WebPrerender* prerender) { | |
118 const PrerenderExtraData& extra_data = | |
119 PrerenderExtraData::FromPrerender(*prerender); | |
120 std::map<int, WebPrerender*>::iterator it = | |
121 prerenders_.find(extra_data.prerender_id()); | |
122 if (it != prerenders_.end()) | |
123 prerenders_.erase(it); | |
124 content::RenderThread::Get()->Send( | |
125 new PrerenderHostMsg_AbandonLinkRelPrerender(extra_data.prerender_id())); | |
64 } | 126 } |
65 | 127 |
66 } // namespace prerender | 128 } // namespace prerender |
OLD | NEW |