Add pending preloads indexed by routing id. Start preloading once we navigate.

chrome/browser/prerender/prerender_resource_handler.cc

 // Copyright (c) 2011 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/browser/prerender/prerender_resource_handler.h"
 
 #include "chrome/browser/net/chrome_url_request_context.h"
 #include "content/common/resource_response.h"
 #include "net/base/load_flags.h"
 #include "net/http/http_response_headers.h"
@@ skipping 34 matching lines @@
   if (rrh.headers->response_code() != 200)
     return false;
   return true;
 }
 
 }  // namespace
 
 PrerenderResourceHandler* PrerenderResourceHandler::MaybeCreate(
     const net::URLRequest& request,
     ChromeURLRequestContext* context,
-    ResourceHandler* next_handler) {
+    ResourceHandler* next_handler,
+    bool is_from_prerender,
+    int route_id) {
   if (!context || !context->prerender_manager())
     return NULL;
   if (!(request.load_flags() & net::LOAD_PREFETCH))
     return NULL;
   if (!ShouldPrerenderURL(request.url()))
     return NULL;
   if (request.method() != "GET")
     return NULL;
+
   return new PrerenderResourceHandler(request,
                                       next_handler,
-                                      context->prerender_manager());
+                                      context->prerender_manager(),
+                                      is_from_prerender,
+                                      route_id);
 }
 
 PrerenderResourceHandler::PrerenderResourceHandler(
     const net::URLRequest& request,
     ResourceHandler* next_handler,
-    PrerenderManager* prerender_manager)
+    PrerenderManager* prerender_manager,
+    bool make_pending,
+    int route_id)
     : next_handler_(next_handler),
       prerender_manager_(prerender_manager),
       ALLOW_THIS_IN_INITIALIZER_LIST(
           prerender_callback_(NewCallback(
               this, &PrerenderResourceHandler::StartPrerender))),
-      request_(request) {
+      request_(request),
+      route_id_(route_id),
+      make_pending_(make_pending) {
   DCHECK(next_handler);
   DCHECK(prerender_manager);
 }
 
 PrerenderResourceHandler::PrerenderResourceHandler(
     const net::URLRequest& request,
     ResourceHandler* next_handler,
     PrerenderCallback* callback)
     : next_handler_(next_handler),
       prerender_callback_(callback),
@@ skipping 29 matching lines @@
 bool PrerenderResourceHandler::OnResponseStarted(int request_id,
                                                  ResourceResponse* response) {
   if (ShouldPrerender(response)) {
     DCHECK(ValidateAliasURLs(alias_urls_));
     BrowserThread::PostTask(
         BrowserThread::UI,
         FROM_HERE,
         NewRunnableMethod(
             this,
             &PrerenderResourceHandler::RunCallbackFromUIThread,
+            route_id_,
             url_,
             alias_urls_,
-            GURL(request_.referrer())));
+            GURL(request_.referrer()),
+            make_pending_));
   }
   return next_handler_->OnResponseStarted(request_id, response);
 }
 
 bool PrerenderResourceHandler::OnWillStart(int request_id,
                                            const GURL& url,
                                            bool* defer) {
   bool will_start = next_handler_->OnWillStart(request_id, url, defer);
   if (will_start) {
     if (!ShouldPrerenderURL(url))
@@ skipping 21 matching lines @@
     const net::URLRequestStatus& status,
     const std::string& security_info) {
   return next_handler_->OnResponseCompleted(request_id, status, security_info);
 }
 
 void PrerenderResourceHandler::OnRequestClosed() {
   next_handler_->OnRequestClosed();
 }
 
 void PrerenderResourceHandler::RunCallbackFromUIThread(
+    int route_id,
     const GURL& url,
     const std::vector<GURL>& alias_urls,
-    const GURL& referrer) {
+    const GURL& referrer,
+    bool make_pending) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
-  prerender_callback_->Run(url, alias_urls, referrer);
+  prerender_callback_->Run(route_id, url, alias_urls, referrer, make_pending);
 }
 
 void PrerenderResourceHandler::StartPrerender(
+    int route_id,
     const GURL& url,
     const std::vector<GURL>& alias_urls,
-    const GURL& referrer) {
+    const GURL& referrer,
+    bool make_pending) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
-  prerender_manager_->AddPreload(url, alias_urls, referrer);
+  if (make_pending)
+    prerender_manager_->AddPendingPreload(route_id, url, alias_urls, referrer);
+  else
+    prerender_manager_->AddPreload(url, alias_urls, referrer);
 }
 
 }  // namespace prerender