1 files changed, 10 insertions, 4 deletions
diff --git a/chrome/browser/prerender/prerender_link_manager.h b/chrome/browser/prerender/prerender_link_manager.h
index 5f091c2..d9e9c00 100644
--- a/chrome/browser/prerender/prerender_link_manager.h
+++ b/chrome/browser/prerender/prerender_link_manager.h
@@ -9,7 +9,6 @@
 #include <utility>
 
 #include "base/basictypes.h"
-#include "base/gtest_prod_util.h"
 #include "chrome/browser/profiles/profile_keyed_service.h"
 #include "googleurl/src/gurl.h"
 
@@ -25,6 +24,7 @@ class Size;
 
 namespace prerender {
 
+class PrerenderHandle;
 class PrerenderManager;
 
 // PrerenderLinkManager implements the API on Link elements for all documents
@@ -69,14 +69,20 @@ class PrerenderLinkManager : public ProfileKeyedService {
   friend class PrerenderTest;
 
   typedef std::pair<int, int> ChildAndPrerenderIdPair;
-  typedef std::map<ChildAndPrerenderIdPair, GURL> IdPairToUrlMap;
+  typedef std::map<ChildAndPrerenderIdPair, PrerenderHandle*>
+      IdPairToPrerenderHandleMap;
 
-  void RemovePrerender(const IdPairToUrlMap::iterator& id_url_iter);
+  void RemovePrerender(
+      const IdPairToPrerenderHandleMap::iterator& id_to_handle_iter);
 
   bool IsEmpty() const;
 
   PrerenderManager* manager_;
-  IdPairToUrlMap ids_to_url_map_;
+
+  // A map from child process id and prerender id to PrerenderHandles. We map
+  // from this pair because the prerender ids are only unique within their
+  // renderer process.
+  IdPairToPrerenderHandleMap ids_to_handle_map_;
 
   DISALLOW_COPY_AND_ASSIGN(PrerenderLinkManager);
 };