Make PrerenderHandle an observer of PrerenderContents.

chrome/browser/prerender/prerender_link_manager.h

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef CHROME_BROWSER_PRERENDER_PRERENDER_LINK_MANAGER_H_
 #define CHROME_BROWSER_PRERENDER_PRERENDER_LINK_MANAGER_H_
 
 #include <map>
 #include <utility>
 
 #include "base/basictypes.h"
+#include "chrome/browser/prerender/prerender_handle.h"
 #include "chrome/browser/profiles/profile_keyed_service.h"
 #include "googleurl/src/gurl.h"
 
 class Profile;
 
 namespace content {
 struct Referrer;
 }
 
 namespace gfx {
 class Size;
 }
 
 namespace prerender {
 
 class PrerenderHandle;
 class PrerenderManager;
 
 // PrerenderLinkManager implements the API on Link elements for all documents
 // being rendered in this chrome instance.  It receives messages from the
 // renderer indicating addition, cancelation and abandonment of link elements,
 // and controls the PrerenderManager accordingly.
-class PrerenderLinkManager : public ProfileKeyedService {
+class PrerenderLinkManager : public ProfileKeyedService,
+                             public PrerenderHandle::Observer {
  public:
   explicit PrerenderLinkManager(PrerenderManager* manager);
   virtual ~PrerenderLinkManager();
 
   // A <link rel=prerender ...> element has been inserted into the document.
   // The |prerender_id| must be unique per |child_id|, and is assigned by the
   // WebPrerendererClient.
   // Returns true if the prerender was accepted by the prerender manager,
   // and false if not.  In either case, the |prerender_id| is usable for
   // future OnCancelPrerender and OnAbandonPrerender calls.
@@ skipping 25 matching lines @@
 
   typedef std::pair<int, int> ChildAndPrerenderIdPair;
   typedef std::map<ChildAndPrerenderIdPair, PrerenderHandle*>
       IdPairToPrerenderHandleMap;
 
   void RemovePrerender(
       const IdPairToPrerenderHandleMap::iterator& id_to_handle_iter);
 
   bool IsEmpty() const;
 
+  IdPairToPrerenderHandleMap::iterator FindPrerenderHandle(
+      PrerenderHandle* prerender_handle);
+
+  // From PrerenderHandle::Observer:
+  virtual void OnPrerenderStart(PrerenderHandle* prerender_handle) OVERRIDE;
+  virtual void OnPrerenderStop(PrerenderHandle* prerender_handle) OVERRIDE;
+  virtual void OnPrerenderAddAlias(PrerenderHandle* prerender_handle,
+                                   const GURL& alias_url) OVERRIDE;
+
   PrerenderManager* manager_;
 
   // 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);
 };
 
 }  // namespace prerender
 
 #endif  // CHROME_BROWSER_PRERENDER_PRERENDER_LINK_MANAGER_H_