Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(154)

Side by Side Diff: chrome/renderer/prerender/prerender_dispatcher.cc

Issue 10918189: Add PrerenderStatusEvent on Prerenders (Closed) Base URL: http://git.chromium.org/git/chromium.git@trunk
Patch Set: rebase to trunk, remove some unrelated fixes... Created 8 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
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/prerender_extra_data.h" 9 #include "chrome/renderer/prerender/prerender_extra_data.h"
10 #include "content/public/common/referrer.h"
10 #include "content/public/renderer/render_thread.h" 11 #include "content/public/renderer/render_thread.h"
12 #include "content/public/renderer/render_view.h"
11 #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"
12 #include "third_party/WebKit/Source/Platform/chromium/public/WebPrerenderingSupp ort.h" 16 #include "third_party/WebKit/Source/Platform/chromium/public/WebPrerenderingSupp ort.h"
13 17
14 namespace prerender { 18 namespace prerender {
15 19
20 using WebKit::WebPrerender;
21 using WebKit::WebPrerenderingSupport;
22
16 PrerenderDispatcher::PrerenderDispatcher() { 23 PrerenderDispatcher::PrerenderDispatcher() {
17 WebKit::WebPrerenderingSupport::initialize(this); 24 WebPrerenderingSupport::initialize(this);
18 } 25 }
19 26
20 PrerenderDispatcher::~PrerenderDispatcher() { 27 PrerenderDispatcher::~PrerenderDispatcher() {
21 WebKit::WebPrerenderingSupport::shutdown(); 28 WebPrerenderingSupport::shutdown();
22 } 29 }
23 30
24 bool PrerenderDispatcher::IsPrerenderURL(const GURL& url) const { 31 bool PrerenderDispatcher::IsPrerenderURL(const GURL& url) const {
25 return prerender_urls_.find(url) != prerender_urls_.end(); 32 for (std::map<int, WebKit::WebPrerender*>::const_iterator
26 } 33 it = prerenders_.begin();
27 34 it != prerenders_.end(); ++it) {
28 void PrerenderDispatcher::OnAddPrerenderURL(const GURL& url) { 35 if (it->second->url() == url)
29 PrerenderMap::iterator it = prerender_urls_.find(url); 36 return true;
30 if (it != prerender_urls_.end()) {
31 DCHECK_LT(0, it->second);
32 it->second += 1;
33 } else {
34 prerender_urls_[url] = 1;
35 } 37 }
36 } 38 for (std::multimap<int, GURL>::const_iterator it = prerender_aliases_.begin();
37 39 it != prerender_aliases_.end(); ++it) {
38 void PrerenderDispatcher::OnRemovePrerenderURL(const GURL& url) { 40 if (it->second == url)
39 PrerenderMap::iterator it = prerender_urls_.find(url); 41 return true;
40 // This is possible with a spurious remove. 42 }
41 // TODO(cbentzel): We'd also want to send the map of active prerenders when 43 return false;
42 // creating a new render process, so the Add/Remove go relative to that.
43 // This may not be that big of a deal in practice, since the newly created tab
44 // is unlikely to go to the prerendered page.
45 if (it == prerender_urls_.end())
46 return;
47 DCHECK_LT(0, it->second);
48 it->second -= 1;
49 if (it->second == 0)
50 prerender_urls_.erase(it);
51 } 44 }
52 45
53 bool PrerenderDispatcher::OnControlMessageReceived( 46 bool PrerenderDispatcher::OnControlMessageReceived(
54 const IPC::Message& message) { 47 const IPC::Message& message) {
55 bool handled = true; 48 bool handled = true;
56 IPC_BEGIN_MESSAGE_MAP(PrerenderDispatcher, message) 49 IPC_BEGIN_MESSAGE_MAP(PrerenderDispatcher, message)
57 IPC_MESSAGE_HANDLER(PrerenderMsg_AddPrerenderURL, OnAddPrerenderURL) 50 IPC_MESSAGE_HANDLER(PrerenderMsg_StartedPrerender, OnStartedPrerender)
58 IPC_MESSAGE_HANDLER(PrerenderMsg_RemovePrerenderURL, OnRemovePrerenderURL) 51 IPC_MESSAGE_HANDLER(PrerenderMsg_AddAliasToPrerender, OnAddAliasToPrerender)
52 IPC_MESSAGE_HANDLER(PrerenderMsg_StoppedPrerender, OnStoppedPrerender)
59 IPC_MESSAGE_UNHANDLED(handled = false) 53 IPC_MESSAGE_UNHANDLED(handled = false)
60 IPC_END_MESSAGE_MAP() 54 IPC_END_MESSAGE_MAP()
61 55
62 return handled; 56 return handled;
63 } 57 }
64 58
65 void PrerenderDispatcher::add(const WebKit::WebPrerender& prerender) { 59 void PrerenderDispatcher::OnStartedPrerender(int prerender_id) {
60 running_prerenders_.insert(prerender_id);
61
62 std::map<int, WebPrerender*>::iterator it = prerenders_.find(prerender_id);
63 if (it != prerenders_.end())
64 it->second->didStartPrerender();
65 }
66
67 void PrerenderDispatcher::OnAddAliasToPrerender(int prerender_id,
68 const GURL& url) {
69 prerender_aliases_.insert(
70 std::multimap<int, GURL>::value_type(prerender_id, url));
71 }
72
73 void PrerenderDispatcher::OnStoppedPrerender(int prerender_id) {
74 // It's possible we will receive a spurious remove, so we don't crash in that
75 // situation.
76
77 // TODO(cbentzel): We'd also want to send the map of active prerenders when
78 // creating a new render process, so the Add/Remove go relative to that.
79 // This may not be that big of a deal in practice, since the newly created tab
80 // is unlikely to go to the prerendered page.
81
82 running_prerenders_.erase(prerender_id);
83 prerender_aliases_.erase(prerender_id);
84
85 std::map<int, WebPrerender*>::iterator it = prerenders_.find(prerender_id);
86 if (it != prerenders_.end())
87 it->second->didStopPrerender();
88 }
89
90 void PrerenderDispatcher::add(WebPrerender* prerender) {
66 const PrerenderExtraData& extra_data = 91 const PrerenderExtraData& extra_data =
67 PrerenderExtraData::FromPrerender(prerender); 92 PrerenderExtraData::FromPrerender(*prerender);
93 prerenders_.insert(std::map<int, WebPrerender*>::value_type(
94 extra_data.prerender_id(), prerender));
68 content::RenderThread::Get()->Send(new PrerenderHostMsg_AddLinkRelPrerender( 95 content::RenderThread::Get()->Send(new PrerenderHostMsg_AddLinkRelPrerender(
69 extra_data.prerender_id(), GURL(prerender.url()), 96 extra_data.prerender_id(), GURL(prerender->url()),
70 content::Referrer(GURL(prerender.referrer()), prerender.referrerPolicy()), 97 content::Referrer(GURL(prerender->referrer()),
98 prerender->referrerPolicy()),
71 extra_data.size(), extra_data.render_view_route_id())); 99 extra_data.size(), extra_data.render_view_route_id()));
72 } 100 }
73 101
74 void PrerenderDispatcher::cancel(const WebKit::WebPrerender& prerender) { 102 void PrerenderDispatcher::cancel(WebPrerender* prerender) {
75 const PrerenderExtraData& extra_data = 103 const PrerenderExtraData& extra_data =
76 PrerenderExtraData::FromPrerender(prerender); 104 PrerenderExtraData::FromPrerender(*prerender);
105 std::map<int, WebPrerender*>::iterator it =
106 prerenders_.find(extra_data.prerender_id());
107 if (it != prerenders_.end())
108 prerenders_.erase(it);
77 content::RenderThread::Get()->Send( 109 content::RenderThread::Get()->Send(
78 new PrerenderHostMsg_CancelLinkRelPrerender(extra_data.prerender_id())); 110 new PrerenderHostMsg_CancelLinkRelPrerender(extra_data.prerender_id()));
79 } 111 }
80 112
81 void PrerenderDispatcher::abandon(const WebKit::WebPrerender& prerender) { 113 void PrerenderDispatcher::abandon(WebPrerender* prerender) {
82 const PrerenderExtraData& extra_data = 114 const PrerenderExtraData& extra_data =
83 PrerenderExtraData::FromPrerender(prerender); 115 PrerenderExtraData::FromPrerender(*prerender);
116 std::map<int, WebPrerender*>::iterator it =
117 prerenders_.find(extra_data.prerender_id());
118 if (it != prerenders_.end())
119 prerenders_.erase(it);
84 content::RenderThread::Get()->Send( 120 content::RenderThread::Get()->Send(
85 new PrerenderHostMsg_AbandonLinkRelPrerender(extra_data.prerender_id())); 121 new PrerenderHostMsg_AbandonLinkRelPrerender(extra_data.prerender_id()));
86 } 122 }
87 123
88 } // namespace prerender 124 } // namespace prerender
OLDNEW
« no previous file with comments | « chrome/renderer/prerender/prerender_dispatcher.h ('k') | chrome/renderer/prerender/prerender_dispatcher_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698