1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
// 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 <utility>
#include "base/bind.h"
#include "base/memory/ptr_util.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);
prefetcher_map_.clear();
}
void ResourcePrefetcherManager::MaybeAddPrefetch(
const GURL& main_frame_url,
const std::vector<GURL>& urls) {
DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
// Don't add a duplicate prefetch for the same host.
std::string key = main_frame_url.host();
if (base::ContainsKey(prefetcher_map_, key))
return;
auto prefetcher =
base::MakeUnique<ResourcePrefetcher>(this, config_, main_frame_url, urls);
prefetcher->Start();
prefetcher_map_[key] = std::move(prefetcher);
}
void ResourcePrefetcherManager::MaybeRemovePrefetch(
const GURL& main_frame_url) {
DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
std::string key = main_frame_url.host();
auto it = prefetcher_map_.find(key);
if (it != prefetcher_map_.end() &&
it->second->main_frame_url() == main_frame_url) {
it->second->Stop();
}
}
void ResourcePrefetcherManager::ResourcePrefetcherFinished(
ResourcePrefetcher* resource_prefetcher) {
DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
const GURL& main_frame_url = resource_prefetcher->main_frame_url();
BrowserThread::PostTask(
BrowserThread::UI, FROM_HERE,
base::Bind(&ResourcePrefetchPredictor::OnPrefetchingFinished,
base::Unretained(predictor_), main_frame_url));
const std::string key = main_frame_url.host();
auto it = prefetcher_map_.find(key);
DCHECK(it != prefetcher_map_.end());
prefetcher_map_.erase(it);
}
net::URLRequestContext* ResourcePrefetcherManager::GetURLRequestContext() {
DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
return context_getter_->GetURLRequestContext();
}
} // namespace predictors
|
