| OLD | NEW |
| (Empty) |
| 1 // Copyright 2013 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_COMMON_INSTANT_RESTRICTED_ID_CACHE_H_ | |
| 6 #define CHROME_COMMON_INSTANT_RESTRICTED_ID_CACHE_H_ | |
| 7 | |
| 8 #include <set> | |
| 9 #include <utility> | |
| 10 #include <vector> | |
| 11 | |
| 12 #include "base/containers/mru_cache.h" | |
| 13 #include "base/gtest_prod_util.h" | |
| 14 #include "base/logging.h" | |
| 15 #include "chrome/common/instant_types.h" | |
| 16 | |
| 17 // In InstantExtended, iframes are used to display objects which can only be | |
| 18 // referenced by the Instant page using an ID (restricted ID). These IDs need to | |
| 19 // be unique and cached for a while so that the SearchBox API can fetch the | |
| 20 // object info based on the ID when required by the Instant page. The reason to | |
| 21 // use a cache of N items as against just the last set of results is that there | |
| 22 // may be race conditions - e.g. the user clicks on a result being shown but the | |
| 23 // result set has internally changed but not yet been displayed. | |
| 24 // | |
| 25 // The cache can be used in two modes: | |
| 26 // | |
| 27 // 1. To store items and assign restricted IDs to them. The cache will store | |
| 28 // a max of |max_cache_size_| items and assign them unique IDs. | |
| 29 // | |
| 30 // 2. To store items that already have restricted IDs assigned to them (e.g. | |
| 31 // from another instance of the cache). The cache will then not generate IDs | |
| 32 // and does not make any guarantees of the uniqueness of the IDs. If multiple | |
| 33 // items are inserted with the same ID, the cache will return the last | |
| 34 // inserted item in GetItemWithRestrictedID() call. | |
| 35 | |
| 36 // T needs to be copyable. | |
| 37 template <typename T> | |
| 38 class InstantRestrictedIDCache { | |
| 39 public: | |
| 40 typedef std::pair<InstantRestrictedID, T> ItemIDPair; | |
| 41 typedef std::vector<T> ItemVector; | |
| 42 typedef std::vector<ItemIDPair> ItemIDVector; | |
| 43 | |
| 44 explicit InstantRestrictedIDCache(size_t max_cache_size); | |
| 45 ~InstantRestrictedIDCache(); | |
| 46 | |
| 47 // Adds items to the cache, assigning restricted IDs in the process. May | |
| 48 // delete older items from the cache. |items.size()| has to be less than max | |
| 49 // cache size. | |
| 50 void AddItems(const ItemVector& items); | |
| 51 | |
| 52 // Adds items to the cache using the supplied restricted IDs. May delete | |
| 53 // older items from the cache. No two entries in |items| should have the same | |
| 54 // InstantRestrictedID. |items.size()| has to be less than max cache size. | |
| 55 void AddItemsWithRestrictedID(const ItemIDVector& items); | |
| 56 | |
| 57 // Returns the last set of items added to the cache either via AddItems() or | |
| 58 // AddItemsWithRestrictedID(). | |
| 59 void GetCurrentItems(ItemIDVector* items) const; | |
| 60 | |
| 61 // Returns true if the |restricted_id| is present in the cache and if so, | |
| 62 // returns a copy of the item. | |
| 63 bool GetItemWithRestrictedID(InstantRestrictedID restricted_id, | |
| 64 T* item) const; | |
| 65 | |
| 66 private: | |
| 67 FRIEND_TEST_ALL_PREFIXES(InstantRestrictedIDCacheTest, AutoIDGeneration); | |
| 68 FRIEND_TEST_ALL_PREFIXES(InstantRestrictedIDCacheTest, CrazyIDGeneration); | |
| 69 FRIEND_TEST_ALL_PREFIXES(InstantRestrictedIDCacheTest, ManualIDGeneration); | |
| 70 FRIEND_TEST_ALL_PREFIXES(InstantRestrictedIDCacheTest, MixIDGeneration); | |
| 71 FRIEND_TEST_ALL_PREFIXES(InstantRestrictedIDCacheTest, AddEmptySet); | |
| 72 FRIEND_TEST_ALL_PREFIXES(InstantRestrictedIDCacheTest, | |
| 73 AddItemsWithRestrictedID); | |
| 74 | |
| 75 typedef base::MRUCache<InstantRestrictedID, T> CacheImpl; | |
| 76 | |
| 77 mutable CacheImpl cache_; | |
| 78 typename CacheImpl::reverse_iterator last_add_start_; | |
| 79 InstantRestrictedID last_restricted_id_; | |
| 80 | |
| 81 DISALLOW_COPY_AND_ASSIGN(InstantRestrictedIDCache); | |
| 82 }; | |
| 83 | |
| 84 template <typename T> | |
| 85 InstantRestrictedIDCache<T>::InstantRestrictedIDCache(size_t max_cache_size) | |
| 86 : cache_(max_cache_size), | |
| 87 last_add_start_(cache_.rend()), | |
| 88 last_restricted_id_(0) { | |
| 89 DCHECK(max_cache_size); | |
| 90 } | |
| 91 | |
| 92 template <typename T> | |
| 93 InstantRestrictedIDCache<T>::~InstantRestrictedIDCache() { | |
| 94 } | |
| 95 | |
| 96 template <typename T> | |
| 97 void InstantRestrictedIDCache<T>::AddItems(const ItemVector& items) { | |
| 98 DCHECK_LE(items.size(), cache_.max_size()); | |
| 99 | |
| 100 if (items.empty()) { | |
| 101 last_add_start_ = cache_.rend(); | |
| 102 return; | |
| 103 } | |
| 104 | |
| 105 for (size_t i = 0; i < items.size(); ++i) { | |
| 106 InstantRestrictedID id = ++last_restricted_id_; | |
| 107 cache_.Put(id, items[i]); | |
| 108 if (i == 0) | |
| 109 last_add_start_ = --cache_.rend(); | |
| 110 } | |
| 111 } | |
| 112 | |
| 113 template <typename T> | |
| 114 void InstantRestrictedIDCache<T>::AddItemsWithRestrictedID( | |
| 115 const ItemIDVector& items) { | |
| 116 DCHECK_LE(items.size(), cache_.max_size()); | |
| 117 | |
| 118 if (items.empty()) { | |
| 119 last_add_start_ = cache_.rend(); | |
| 120 return; | |
| 121 } | |
| 122 | |
| 123 std::set<InstantRestrictedID> ids_added; | |
| 124 for (size_t i = 0; i < items.size(); ++i) { | |
| 125 const ItemIDPair& item_id = items[i]; | |
| 126 | |
| 127 DCHECK(ids_added.find(item_id.first) == ids_added.end()); | |
| 128 ids_added.insert(item_id.first); | |
| 129 | |
| 130 cache_.Put(item_id.first, item_id.second); | |
| 131 last_restricted_id_ = std::max(item_id.first, last_restricted_id_); | |
| 132 } | |
| 133 | |
| 134 // cache_.Put() can invalidate the iterator |last_add_start_| is pointing to. | |
| 135 // Therefore, update |last_add_start_| after adding all the items to the | |
| 136 // |cache_|. | |
| 137 last_add_start_ = cache_.rend(); | |
| 138 for (size_t i = 0; i < items.size(); ++i) | |
| 139 --last_add_start_; | |
| 140 } | |
| 141 | |
| 142 template <typename T> | |
| 143 void InstantRestrictedIDCache<T>::GetCurrentItems(ItemIDVector* items) const { | |
| 144 items->clear(); | |
| 145 | |
| 146 for (typename CacheImpl::reverse_iterator it = last_add_start_; | |
| 147 it != cache_.rend(); ++it) { | |
| 148 items->push_back(std::make_pair(it->first, it->second)); | |
| 149 } | |
| 150 } | |
| 151 | |
| 152 template <typename T> | |
| 153 bool InstantRestrictedIDCache<T>::GetItemWithRestrictedID( | |
| 154 InstantRestrictedID restricted_id, | |
| 155 T* item) const { | |
| 156 DCHECK(item); | |
| 157 | |
| 158 typename CacheImpl::const_iterator cache_it = cache_.Peek(restricted_id); | |
| 159 if (cache_it == cache_.end()) | |
| 160 return false; | |
| 161 *item = cache_it->second; | |
| 162 return true; | |
| 163 } | |
| 164 | |
| 165 #endif // CHROME_COMMON_INSTANT_RESTRICTED_ID_CACHE_H_ | |
| OLD | NEW |