| // Copyright 2014 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/predictors/resource_prefetcher_manager.h" |
| |
| #include "base/bind.h" |
| #include "base/stl_util.h" |
| #include "chrome/browser/predictors/resource_prefetch_predictor.h" |
| #include "content/public/browser/browser_thread.h" |
| #include "net/url_request/url_request.h" |
| #include "net/url_request/url_request_context_getter.h" |
| |
| using content::BrowserThread; |
| |
| namespace predictors { |
| |
| ResourcePrefetcherManager::ResourcePrefetcherManager( |
| ResourcePrefetchPredictor* predictor, |
| const ResourcePrefetchPredictorConfig& config, |
| net::URLRequestContextGetter* context_getter) |
| : predictor_(predictor), |
| config_(config), |
| context_getter_(context_getter) { |
| DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| CHECK(predictor_); |
| CHECK(context_getter_); |
| } |
| |
| ResourcePrefetcherManager::~ResourcePrefetcherManager() { |
| DCHECK(prefetcher_map_.empty()) |
| << "Did not call ShutdownOnUIThread or ShutdownOnIOThread. " |
| " Will leak Prefetcher pointers."; |
| } |
| |
| void ResourcePrefetcherManager::ShutdownOnUIThread() { |
| DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| |
| predictor_ = NULL; |
| BrowserThread::PostTask(BrowserThread::IO, FROM_HERE, |
| base::Bind(&ResourcePrefetcherManager::ShutdownOnIOThread, |
| this)); |
| } |
| |
| void ResourcePrefetcherManager::ShutdownOnIOThread() { |
| DCHECK_CURRENTLY_ON(content::BrowserThread::IO); |
| STLDeleteContainerPairSecondPointers(prefetcher_map_.begin(), |
| prefetcher_map_.end()); |
| } |
| |
| void ResourcePrefetcherManager::MaybeAddPrefetch( |
| const NavigationID& navigation_id, |
| PrefetchKeyType key_type, |
| scoped_ptr<ResourcePrefetcher::RequestVector> requests) { |
| DCHECK_CURRENTLY_ON(content::BrowserThread::IO); |
| |
| // Don't add a duplicate prefetch for the same host or URL. |
| std::string key = key_type == PREFETCH_KEY_TYPE_HOST ? |
| navigation_id.main_frame_url.host() : navigation_id.main_frame_url.spec(); |
| if (ContainsKey(prefetcher_map_, key)) |
| return; |
| |
| ResourcePrefetcher* prefetcher = new ResourcePrefetcher( |
| this, config_, navigation_id, key_type, requests.Pass()); |
| prefetcher_map_.insert(std::make_pair(key, prefetcher)); |
| prefetcher->Start(); |
| } |
| |
| void ResourcePrefetcherManager::MaybeRemovePrefetch( |
| const NavigationID& navigation_id) { |
| DCHECK_CURRENTLY_ON(content::BrowserThread::IO); |
| |
| // Look for a URL based prefetch first. |
| PrefetcherMap::iterator it = prefetcher_map_.find( |
| navigation_id.main_frame_url.spec()); |
| if (it != prefetcher_map_.end() && |
| it->second->navigation_id() == navigation_id) { |
| it->second->Stop(); |
| return; |
| } |
| |
| // No URL based prefetching, look for host based. |
| it = prefetcher_map_.find(navigation_id.main_frame_url.host()); |
| if (it != prefetcher_map_.end() && |
| it->second->navigation_id() == navigation_id) { |
| it->second->Stop(); |
| } |
| } |
| |
| void ResourcePrefetcherManager::ResourcePrefetcherFinished( |
| ResourcePrefetcher* resource_prefetcher, |
| ResourcePrefetcher::RequestVector* requests) { |
| DCHECK_CURRENTLY_ON(content::BrowserThread::IO); |
| |
| // |predictor_| can only be accessed from the UI thread. |
| scoped_ptr<ResourcePrefetcher::RequestVector> scoped_requests(requests); |
| BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, |
| base::Bind(&ResourcePrefetcherManager::ResourcePrefetcherFinishedOnUI, |
| this, |
| resource_prefetcher->navigation_id(), |
| resource_prefetcher->key_type(), |
| base::Passed(&scoped_requests))); |
| |
| const GURL& main_frame_url = |
| resource_prefetcher->navigation_id().main_frame_url; |
| const std::string key = |
| resource_prefetcher->key_type() == PREFETCH_KEY_TYPE_HOST ? |
| main_frame_url.host() : main_frame_url.spec(); |
| PrefetcherMap::iterator it = prefetcher_map_.find(key); |
| DCHECK(it != prefetcher_map_.end()); |
| delete it->second; |
| prefetcher_map_.erase(it); |
| } |
| |
| void ResourcePrefetcherManager::ResourcePrefetcherFinishedOnUI( |
| const NavigationID& navigation_id, |
| PrefetchKeyType key_type, |
| scoped_ptr<ResourcePrefetcher::RequestVector> requests) { |
| DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| |
| // |predictor_| may have been set to NULL if the predictor is shutting down. |
| if (predictor_) { |
| predictor_->FinishedPrefetchForNavigation(navigation_id, |
| key_type, |
| requests.release()); |
| } |
| } |
| |
| net::URLRequestContext* ResourcePrefetcherManager::GetURLRequestContext() { |
| DCHECK_CURRENTLY_ON(content::BrowserThread::IO); |
| |
| return context_getter_->GetURLRequestContext(); |
| } |
| |
| } // namespace predictors |