1// Copyright 2014 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#ifndef CHROME_BROWSER_PREDICTORS_RESOURCE_PREFETCHER_MANAGER_H_
6#define CHROME_BROWSER_PREDICTORS_RESOURCE_PREFETCHER_MANAGER_H_
7
8#include <map>
9
10#include "base/memory/ref_counted.h"
11#include "chrome/browser/predictors/resource_prefetch_common.h"
12#include "chrome/browser/predictors/resource_prefetcher.h"
13
14namespace net {
15class URLRequestContextGetter;
16}
17
18namespace predictors {
19
20struct NavigationID;
21class ResourcePrefetchPredictor;
22
23// Manages prefetches for multple navigations.
24//  - Created and owned by the resource prefetch predictor.
25//  - Needs to be refcounted as it is de-referenced on two different threads.
26//  - Created on the UI thread, but most functions are called in the IO thread.
27//  - Will only allow one inflight prefresh per main frame URL.
28class ResourcePrefetcherManager
29    :  public ResourcePrefetcher::Delegate,
30       public base::RefCountedThreadSafe<ResourcePrefetcherManager> {
31 public:
32  // The |predictor| should be alive till ShutdownOnIOThread is called.
33  ResourcePrefetcherManager(ResourcePrefetchPredictor* predictor,
34                            const ResourcePrefetchPredictorConfig& config,
35                            net::URLRequestContextGetter* getter);
36
37  // UI thread.
38  void ShutdownOnUIThread();
39
40  // --- IO Thread methods.
41
42  // The prefetchers need to be deleted on the IO thread.
43  void ShutdownOnIOThread();
44
45  // Will create a new ResourcePrefetcher for the main frame url of the input
46  // navigation if there isn't one already for the same URL or host (for host
47  // based).
48  void MaybeAddPrefetch(const NavigationID& navigation_id,
49                        PrefetchKeyType key_type,
50                        scoped_ptr<ResourcePrefetcher::RequestVector> requests);
51
52  // Stops the ResourcePrefetcher for the input navigation, if one was in
53  // progress.
54  void MaybeRemovePrefetch(const NavigationID& navigation_id);
55
56  // ResourcePrefetcher::Delegate methods.
57  virtual void ResourcePrefetcherFinished(
58      ResourcePrefetcher* prefetcher,
59      ResourcePrefetcher::RequestVector* requests) OVERRIDE;
60  virtual net::URLRequestContext* GetURLRequestContext() OVERRIDE;
61
62 private:
63  friend class base::RefCountedThreadSafe<ResourcePrefetcherManager>;
64  friend class MockResourcePrefetcherManager;
65
66  typedef std::map<std::string, ResourcePrefetcher*> PrefetcherMap;
67
68  virtual ~ResourcePrefetcherManager();
69
70  // UI Thread. |predictor_| needs to be called on the UI thread.
71  void ResourcePrefetcherFinishedOnUI(
72      const NavigationID& navigation_id,
73      PrefetchKeyType key_type,
74      scoped_ptr<ResourcePrefetcher::RequestVector> requests);
75
76  ResourcePrefetchPredictor* predictor_;
77  const ResourcePrefetchPredictorConfig config_;
78  net::URLRequestContextGetter* const context_getter_;
79
80  PrefetcherMap prefetcher_map_;  // Owns the ResourcePrefetcher pointers.
81
82  DISALLOW_COPY_AND_ASSIGN(ResourcePrefetcherManager);
83};
84
85}  // namespace predictors
86
87#endif  // CHROME_BROWSER_PREDICTORS_RESOURCE_PREFETCHER_MANAGER_H_
88