src/chrome/browser/predictors/resource_prefetcher_manager.h

   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
  14 namespace net {
  15 class URLRequestContextGetter;
  16 }
  17
  18 namespace predictors {
  19
  20 struct NavigationID;
  21 class 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.
  28 class 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_