| OLD | NEW |
| (Empty) |
| 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_ | |
| OLD | NEW |