Modifying prefetch to account for multi-profile.

chrome/browser/net/connect_interceptor.cc

Index: chrome/browser/net/connect_interceptor.cc
===================================================================
--- chrome/browser/net/connect_interceptor.cc	(revision 96009)
+++ chrome/browser/net/connect_interceptor.cc	(working copy)
@@ -4,8 +4,9 @@
 
 #include "chrome/browser/net/connect_interceptor.h"
 
-#include "chrome/browser/net/predictor_api.h"
+#include "chrome/browser/net/predictor.h"
 #include "net/base/load_flags.h"
+#include "net/url_request/url_request.h"
 
 namespace chrome_browser_net {
 
@@ -16,24 +17,24 @@
 // TODO(jar): We should do a persistent field trial to validate/optimize this.
 static const int kMaxUnusedSocketLifetimeSecondsWithoutAGet = 10;
 
-ConnectInterceptor::ConnectInterceptor()
+ConnectInterceptor::ConnectInterceptor(Predictor* predictor)
     : timed_cache_(base::TimeDelta::FromSeconds(
-          kMaxUnusedSocketLifetimeSecondsWithoutAGet)) {
-  net::URLRequest::Deprecated::RegisterRequestInterceptor(this);

[willchan no longer on Chromium, 2011/08/10 16:10:35]

there's a friend list in net/url_request/url_request.h, please remove the line
for this use:
http://codesearch.google.com/codesearch#OAMlx_jo-ck/src/net/url_request/url_r....

[rpetterson, 2011/08/12 03:12:36]

Done.

+          kMaxUnusedSocketLifetimeSecondsWithoutAGet)),
+      predictor_(predictor) {
 }
 
 ConnectInterceptor::~ConnectInterceptor() {
-  net::URLRequest::Deprecated::UnregisterRequestInterceptor(this);
 }
 
 net::URLRequestJob* ConnectInterceptor::MaybeIntercept(
-    net::URLRequest* request) {
+    net::URLRequest* request) const {
   GURL request_scheme_host(Predictor::CanonicalizeUrl(request->url()));
   if (request_scheme_host == GURL::EmptyGURL())
     return NULL;
 
   // Learn what URLs are likely to be needed during next startup.
-  LearnAboutInitialNavigation(request_scheme_host);
+  if (predictor_)

[willchan no longer on Chromium, 2011/08/10 16:10:35]

|predictor_| should never be NULL, right? It was possible before due to
incognito, but now the incognito profile never creates a ConnectInterceptor
object. So we shouldn't need this NULL check. Please DCHECK(predictor); in the
constructor.

[rpetterson, 2011/08/12 03:12:36]

Yes, you're right since we only create the connect_interceptor in the
profile_impl_io_data.

DCHECK in constructor and removed this null check and others.

+    predictor_->LearnAboutInitialNavigation(request_scheme_host);
 
   bool redirected_host = false;
   if (request->referrer().empty()) {
@@ -55,7 +56,9 @@
         if (request->original_url().path().length() <= 1 &&
             timed_cache_.WasRecentlySeen(original_scheme_host)) {
           // TODO(jar): These definite redirects could be learned much faster.
-          LearnFromNavigation(original_scheme_host, request_scheme_host);
+          if (predictor_)
+            predictor_->LearnFromNavigation(original_scheme_host,
+                                            request_scheme_host);
         }
       }
     }
@@ -64,7 +67,9 @@
     bool is_subresource = !(request->load_flags() & net::LOAD_MAIN_FRAME);
     // Learn about our referring URL, for use in the future.
     if (is_subresource && timed_cache_.WasRecentlySeen(referring_scheme_host))
-      LearnFromNavigation(referring_scheme_host, request_scheme_host);
+      if (predictor_)
+        predictor_->LearnFromNavigation(referring_scheme_host,
+                                        request_scheme_host);
     if (referring_scheme_host == request_scheme_host) {
       // We've already made any/all predictions when we navigated to the
       // referring host, so we can bail out here.
@@ -80,18 +85,19 @@
   // main frame request - way back in RenderViewHost::Navigate.  So only handle
   // predictions now for subresources or for redirected hosts.
   if ((request->load_flags() & net::LOAD_SUB_FRAME) || redirected_host)
-    PredictFrameSubresources(request_scheme_host);
+    if (predictor_)
+      predictor_->PredictFrameSubresources(request_scheme_host);
   return NULL;
 }
 
 net::URLRequestJob* ConnectInterceptor::MaybeInterceptResponse(
-    net::URLRequest* request) {
+    net::URLRequest* request) const {
   return NULL;
 }
 
 net::URLRequestJob* ConnectInterceptor::MaybeInterceptRedirect(
-    net::URLRequest* request,
-    const GURL& location) {
+    const GURL& location,
+    net::URLRequest* request) const {
   return NULL;
 }
 
@@ -103,7 +109,7 @@
 // Make Clang compilation happy with explicit destructor.
 ConnectInterceptor::TimedCache::~TimedCache() {}
 
-bool ConnectInterceptor::TimedCache::WasRecentlySeen(const GURL& url) {
+bool ConnectInterceptor::TimedCache::WasRecentlySeen(const GURL& url) const {
   DCHECK_EQ(url.GetWithEmptyPath(), url);
   // Evict any overly old entries.
   base::TimeTicks now = base::TimeTicks::Now();
@@ -117,7 +123,7 @@
   return mru_cache_.end() != mru_cache_.Peek(url);
 }
 
-void ConnectInterceptor::TimedCache::SetRecentlySeen(const GURL& url) {
+void ConnectInterceptor::TimedCache::SetRecentlySeen(const GURL& url) const {
   DCHECK_EQ(url.GetWithEmptyPath(), url);
   mru_cache_.Put(url, base::TimeTicks::Now());
 }