Index: chrome/renderer/prerender/prerender_dispatcher.cc |
diff --git a/chrome/renderer/prerender/prerender_dispatcher.cc b/chrome/renderer/prerender/prerender_dispatcher.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..500d4bebc28eaebea4e3f361d88ecc88d32b8168 |
--- /dev/null |
+++ b/chrome/renderer/prerender/prerender_dispatcher.cc |
@@ -0,0 +1,60 @@ |
+// 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. |
+ |
+#include "chrome/renderer/prerender/prerender_dispatcher.h" |
+ |
+#include "base/logging.h" |
+#include "chrome/common/prerender_messages.h" |
+#include "googleurl/src/gurl.h" |
+ |
+namespace prerender { |
+ |
+PrerenderDispatcher::PrerenderDispatcher() { |
+} |
+ |
+PrerenderDispatcher::~PrerenderDispatcher() { |
+} |
+ |
+bool PrerenderDispatcher::IsPrerenderURL(const GURL& url) const { |
+ return prerender_urls_.find(url) != prerender_urls_.end(); |
+} |
+ |
+bool PrerenderDispatcher::OnControlMessageReceived( |
+ const IPC::Message& message) { |
+ bool handled = true; |
+ IPC_BEGIN_MESSAGE_MAP(PrerenderDispatcher, message) |
+ IPC_MESSAGE_HANDLER(PrerenderMsg_AddPrerenderURL, OnAddPrerenderURL) |
+ IPC_MESSAGE_HANDLER(PrerenderMsg_RemovePrerenderURL, OnRemovePrerenderURL) |
+ IPC_MESSAGE_UNHANDLED(handled = false) |
+ IPC_END_MESSAGE_MAP() |
+ |
+ return handled; |
+} |
+ |
+void PrerenderDispatcher::OnAddPrerenderURL(const GURL& url) { |
+ PrerenderMap::iterator it = prerender_urls_.find(url); |
+ if (it != prerender_urls_.end()) { |
+ DCHECK(it->second > 0); |
+ it->second += 1; |
+ } else { |
+ prerender_urls_[url] = 1; |
+ } |
+} |
+ |
+void PrerenderDispatcher::OnRemovePrerenderURL(const GURL& url) { |
+ PrerenderMap::iterator it = prerender_urls_.find(url); |
+ // This is possible with a spurious remove. |
+ // TODO(cbentzel): We'd also want to send the map of active prerenders when |
+ // creating a new render process, so the Add/Remove go relative to that. |
+ // This may not be that big of a deal in practice, since the newly created tab |
+ // is unlikely to go to the prerendered page. |
+ if (it == prerender_urls_.end()) |
+ return; |
+ DCHECK(it->second > 0); |
+ it->second -= 1; |
+ if (it->second == 0) |
+ prerender_urls_.erase(it); |
+} |
+ |
+} // namespace prerender |