Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 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 | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "chrome/browser/prerender/prerender_link_manager.h" | |
| 6 | |
| 7 #include <utility> | |
| 8 | |
| 9 #include "chrome/browser/profiles/profile.h" | |
| 10 #include "chrome/browser/prerender/prerender_manager.h" | |
| 11 #include "chrome/browser/prerender/prerender_manager_factory.h" | |
| 12 #include "content/public/common/referrer.h" | |
| 13 #include "googleurl/src/gurl.h" | |
| 14 #include "googleurl/src/url_canon.h" | |
| 15 #include "ui/gfx/size.h" | |
| 16 | |
| 17 namespace prerender { | |
| 18 | |
| 19 PrerenderLinkManager::PrerenderLinkManager( | |
| 20 PrerenderManager* manager) | |
| 21 : manager_(manager) { | |
| 22 } | |
| 23 | |
| 24 PrerenderLinkManager::~PrerenderLinkManager() { | |
| 25 } | |
| 26 | |
| 27 void PrerenderLinkManager::OnNewLinkPrerenderImpl( | |
| 28 int prerender_id, | |
| 29 int child_id, | |
| 30 int render_view_route_id, | |
| 31 const GURL& orig_url, | |
| 32 const content::Referrer& referrer, | |
| 33 const gfx::Size& ALLOW_UNUSED size) { | |
|
gavinp
2012/03/28 19:51:31
I carried this size all the way up for two reasons
| |
| 34 // TODO(gavinp): Add tests to insure fragments work. | |
| 35 url_canon::Replacements<char> replacements; | |
| 36 replacements.ClearRef(); | |
| 37 const GURL url = orig_url.ReplaceComponents(replacements); | |
|
gavinp
2012/03/28 19:51:31
In the spirit of not changing too much as I add t
| |
| 38 | |
| 39 manager_->AddPrerenderFromLinkRelPrerender( | |
| 40 child_id, render_view_route_id, url, referrer); | |
| 41 const ChildAndPrerenderIdPair child_and_prerender_id(child_id, prerender_id); | |
|
cbentzel
2012/03/29 00:13:37
Why don't you need to include route_id as well? Do
gavinp
2012/04/20 17:54:14
The prerender_id should be unique across the WebKi
| |
| 42 DCHECK(id_map_.find(child_and_prerender_id) == id_map_.end()); | |
| 43 id_map_.insert(std::make_pair(child_and_prerender_id, url)); | |
| 44 url_map_.insert(std::make_pair(url, child_and_prerender_id)); | |
|
cbentzel
2012/03/29 00:13:37
I recommend adding a CheckInvariant() function whi
gavinp
2012/04/20 17:54:15
Done.
| |
| 45 } | |
| 46 | |
| 47 void PrerenderLinkManager::OnRemovedLinkPrerenderImpl(const int prerender_id, | |
| 48 const int child_id) { | |
| 49 const ChildAndPrerenderIdPair child_and_prerender_id(child_id, prerender_id); | |
| 50 | |
| 51 PrerenderIdToUrlMap::iterator id_url_iter = | |
| 52 id_map_.find(child_and_prerender_id); | |
| 53 if (id_url_iter == id_map_.end()) | |
| 54 return; | |
| 55 const GURL url = id_url_iter->second; | |
| 56 id_map_.erase(id_url_iter); | |
| 57 | |
| 58 // Are any other link elements prerendering this URL? | |
| 59 UrlToPrerenderIdMap::iterator url_id_iter = url_map_.find(url); | |
|
cbentzel
2012/03/29 00:13:37
This doesn't guarantee to find the first entry in
gavinp
2012/04/20 17:54:15
_The C++ Programming Language_ claims otherwise in
| |
| 60 int remaining_prerender_link_count = 0; | |
| 61 while (url_id_iter != url_map_.end() && url_id_iter->first == url) { | |
| 62 if (url_id_iter->second == child_and_prerender_id) { | |
| 63 UrlToPrerenderIdMap::iterator to_erase = url_id_iter; | |
| 64 ++url_id_iter; | |
| 65 url_map_.erase(to_erase); | |
|
cbentzel
2012/03/29 00:13:37
I always get nervous about erasing while iterating
mmenke
2012/03/29 01:15:39
I suggest that you only delete entries in one plac
gavinp
2012/04/20 17:54:15
Done.
| |
| 66 continue; | |
| 67 } | |
| 68 ++remaining_prerender_link_count; | |
| 69 ++url_id_iter; | |
| 70 } | |
| 71 if (remaining_prerender_link_count) | |
| 72 return; | |
| 73 | |
| 74 // TODO(gavinp): Track down the correct prerender and stop it, rather than | |
|
cbentzel
2012/03/29 00:13:37
Please fix this. Also, it doesn't look like you ha
gavinp
2012/04/20 17:54:15
Done, in a basic way using the existing PrerenderM
| |
| 75 // this nuclear option, which assumes that only one prerender at a time | |
| 76 // runs. | |
| 77 if (manager_->IsPrerendering(url)) | |
| 78 manager_->CancelAllPrerenders(); | |
| 79 } | |
| 80 | |
| 81 void PrerenderLinkManager::OnUnloadedLinkPrerenderImpl( | |
| 82 int prerender_id, | |
| 83 int child_id) { | |
| 84 // TODO(gavinp,cbentzel): Implement reasonable behaviour for | |
| 85 // navigation away from launcher. | |
| 86 const ChildAndPrerenderIdPair child_and_prerender_id(child_id, prerender_id); | |
| 87 PrerenderIdToUrlMap::iterator id_url_iter = | |
| 88 id_map_.find(child_and_prerender_id); | |
| 89 if (id_url_iter == id_map_.end()) | |
| 90 return; | |
| 91 const GURL& url = id_url_iter->second; | |
| 92 DCHECK(url_map_.find(url) != url_map_.end()); | |
| 93 ignore_result(url_map_.erase(url)); | |
| 94 id_map_.erase(id_url_iter); | |
| 95 } | |
| 96 | |
| 97 // static | |
| 98 void PrerenderLinkManager::OnNewLinkPrerender( | |
| 99 Profile* profile, | |
| 100 int prerender_id, | |
| 101 int child_id, | |
| 102 int render_view_route_id, | |
| 103 const GURL& url, | |
| 104 const content::Referrer& referrer, | |
| 105 const gfx::Size& size) { | |
| 106 if (prerender::PrerenderManager* prerender_manager = | |
| 107 prerender::PrerenderManagerFactory::GetForProfile(profile)) { | |
| 108 prerender_manager->link_manager()->OnNewLinkPrerenderImpl( | |
|
cbentzel
2012/03/29 00:13:37
Not sure I like that this goes "Get the prerender
gavinp
2012/04/20 17:54:15
Done.
| |
| 109 prerender_id, child_id, render_view_route_id, | |
| 110 url, referrer, size); | |
| 111 } | |
| 112 } | |
| 113 | |
| 114 // static | |
| 115 void PrerenderLinkManager::OnRemovedLinkPrerender(Profile* profile, | |
| 116 int prerender_id, | |
| 117 int child_id) { | |
| 118 if (prerender::PrerenderManager* prerender_manager = | |
| 119 prerender::PrerenderManagerFactory::GetForProfile(profile)) | |
| 120 prerender_manager->link_manager()->OnRemovedLinkPrerenderImpl(prerender_id, | |
| 121 child_id); | |
| 122 } | |
| 123 | |
| 124 // static | |
| 125 void PrerenderLinkManager::OnUnloadedLinkPrerender(Profile* profile, | |
| 126 int prerender_id, | |
| 127 int child_id) { | |
| 128 if (prerender::PrerenderManager* prerender_manager = | |
| 129 prerender::PrerenderManagerFactory::GetForProfile(profile)) | |
| 130 prerender_manager->link_manager()->OnUnloadedLinkPrerenderImpl(prerender_id, | |
| 131 child_id); | |
| 132 } | |
| 133 | |
| 134 } // namespace prerender | |
| OLD | NEW |