OLD | NEW |
| (Empty) |
1 // Copyright 2016 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 #include "components/offline_pages/offline_page_storage_manager.h" | |
6 | |
7 #include <algorithm> | |
8 | |
9 #include "base/bind.h" | |
10 #include "base/time/clock.h" | |
11 #include "base/time/default_clock.h" | |
12 #include "components/offline_pages/client_policy_controller.h" | |
13 #include "components/offline_pages/offline_page_client_policy.h" | |
14 #include "components/offline_pages/offline_page_item.h" | |
15 #include "components/offline_pages/offline_page_model.h" | |
16 | |
17 using LifetimeType = offline_pages::LifetimePolicy::LifetimeType; | |
18 | |
19 namespace offline_pages { | |
20 | |
21 constexpr double constants::kOfflinePageStorageLimit; | |
22 constexpr double constants::kOfflinePageStorageClearThreshold; | |
23 constexpr base::TimeDelta constants::kClearStorageInterval; | |
24 constexpr base::TimeDelta constants::kRemovePageItemInterval; | |
25 | |
26 OfflinePageStorageManager::OfflinePageStorageManager( | |
27 OfflinePageModel* model, | |
28 ClientPolicyController* policy_controller, | |
29 ArchiveManager* archive_manager) | |
30 : model_(model), | |
31 policy_controller_(policy_controller), | |
32 archive_manager_(archive_manager), | |
33 clock_(new base::DefaultClock()), | |
34 weak_ptr_factory_(this) {} | |
35 | |
36 OfflinePageStorageManager::~OfflinePageStorageManager() {} | |
37 | |
38 void OfflinePageStorageManager::ClearPagesIfNeeded( | |
39 const ClearStorageCallback& callback) { | |
40 if (IsInProgress()) | |
41 return; | |
42 clear_time_ = clock_->Now(); | |
43 archive_manager_->GetStorageStats(base::Bind( | |
44 &OfflinePageStorageManager::OnGetStorageStatsDoneForClearingPages, | |
45 weak_ptr_factory_.GetWeakPtr(), callback)); | |
46 } | |
47 | |
48 void OfflinePageStorageManager::SetClockForTesting( | |
49 std::unique_ptr<base::Clock> clock) { | |
50 clock_ = std::move(clock); | |
51 } | |
52 | |
53 void OfflinePageStorageManager::OnGetStorageStatsDoneForClearingPages( | |
54 const ClearStorageCallback& callback, | |
55 const ArchiveManager::StorageStats& stats) { | |
56 DCHECK(IsInProgress()); | |
57 ClearMode mode = ShouldClearPages(stats); | |
58 if (mode == ClearMode::NOT_NEEDED) { | |
59 last_clear_time_ = clear_time_; | |
60 callback.Run(0, ClearStorageResult::UNNECESSARY); | |
61 return; | |
62 } | |
63 model_->GetAllPages( | |
64 base::Bind(&OfflinePageStorageManager::OnGetAllPagesDoneForClearingPages, | |
65 weak_ptr_factory_.GetWeakPtr(), callback, stats)); | |
66 } | |
67 | |
68 void OfflinePageStorageManager::OnGetAllPagesDoneForClearingPages( | |
69 const ClearStorageCallback& callback, | |
70 const ArchiveManager::StorageStats& stats, | |
71 const MultipleOfflinePageItemResult& pages) { | |
72 std::vector<int64_t> page_ids_to_expire; | |
73 std::vector<int64_t> page_ids_to_remove; | |
74 GetPageIdsToClear(pages, stats, &page_ids_to_expire, &page_ids_to_remove); | |
75 model_->ExpirePages( | |
76 page_ids_to_expire, clear_time_, | |
77 base::Bind(&OfflinePageStorageManager::OnPagesExpired, | |
78 weak_ptr_factory_.GetWeakPtr(), callback, | |
79 page_ids_to_expire.size(), page_ids_to_remove)); | |
80 } | |
81 | |
82 void OfflinePageStorageManager::OnPagesExpired( | |
83 const ClearStorageCallback& callback, | |
84 size_t pages_expired, | |
85 const std::vector<int64_t>& page_ids_to_remove, | |
86 bool expiration_succeeded) { | |
87 // We want to delete the outdated page records regardless the expiration | |
88 // succeeded or not. | |
89 model_->DeletePagesByOfflineId( | |
90 page_ids_to_remove, | |
91 base::Bind(&OfflinePageStorageManager::OnOutdatedPagesCleared, | |
92 weak_ptr_factory_.GetWeakPtr(), callback, pages_expired, | |
93 expiration_succeeded)); | |
94 } | |
95 | |
96 void OfflinePageStorageManager::OnOutdatedPagesCleared( | |
97 const ClearStorageCallback& callback, | |
98 size_t pages_cleared, | |
99 bool expiration_succeeded, | |
100 DeletePageResult result) { | |
101 ClearStorageResult clear_result = ClearStorageResult::SUCCESS; | |
102 if (!expiration_succeeded) { | |
103 clear_result = ClearStorageResult::EXPIRE_FAILURE; | |
104 if (result != DeletePageResult::SUCCESS) | |
105 clear_result = ClearStorageResult::EXPIRE_AND_DELETE_FAILURES; | |
106 } else if (result != DeletePageResult::SUCCESS) { | |
107 clear_result = ClearStorageResult::DELETE_FAILURE; | |
108 } | |
109 last_clear_time_ = clear_time_; | |
110 callback.Run(pages_cleared, clear_result); | |
111 } | |
112 | |
113 void OfflinePageStorageManager::GetPageIdsToClear( | |
114 const MultipleOfflinePageItemResult& pages, | |
115 const ArchiveManager::StorageStats& stats, | |
116 std::vector<int64_t>* page_ids_to_expire, | |
117 std::vector<int64_t>* page_ids_to_remove) { | |
118 // TODO(romax): See how persistent should be considered here. | |
119 // Creating a map from namespace to a vector of page items. | |
120 // Sort each vector based on last accessed time and all pages after index | |
121 // min{size(), page_limit} should be expired. And then start iterating | |
122 // backwards to expire pages. | |
123 std::map<std::string, std::vector<OfflinePageItem>> pages_map; | |
124 std::vector<OfflinePageItem> kept_pages; | |
125 int64_t kept_pages_size = 0; | |
126 | |
127 for (const auto& page : pages) { | |
128 if (!page.IsExpired()) { | |
129 pages_map[page.client_id.name_space].push_back(page); | |
130 } else if (clear_time_ - page.expiration_time >= | |
131 constants::kRemovePageItemInterval) { | |
132 page_ids_to_remove->push_back(page.offline_id); | |
133 } | |
134 } | |
135 | |
136 for (auto& iter : pages_map) { | |
137 std::string name_space = iter.first; | |
138 std::vector<OfflinePageItem>& page_list = iter.second; | |
139 | |
140 LifetimePolicy policy = | |
141 policy_controller_->GetPolicy(name_space).lifetime_policy; | |
142 | |
143 std::sort(page_list.begin(), page_list.end(), | |
144 [](const OfflinePageItem& a, const OfflinePageItem& b) -> bool { | |
145 return a.last_access_time > b.last_access_time; | |
146 }); | |
147 | |
148 size_t page_list_size = page_list.size(); | |
149 size_t pos = 0; | |
150 while (pos < page_list_size && | |
151 (policy.page_limit == kUnlimitedPages || pos < policy.page_limit) && | |
152 !ShouldBeExpired(page_list.at(pos))) { | |
153 kept_pages_size += page_list.at(pos).file_size; | |
154 kept_pages.push_back(page_list.at(pos)); | |
155 pos++; | |
156 } | |
157 | |
158 for (; pos < page_list_size; pos++) | |
159 page_ids_to_expire->push_back(page_list.at(pos).offline_id); | |
160 } | |
161 | |
162 // If we're still over the clear threshold, we're going to clear remaining | |
163 // pages from oldest last access time. | |
164 int64_t free_space = stats.free_disk_space; | |
165 int64_t total_size = stats.total_archives_size; | |
166 int64_t space_to_release = | |
167 kept_pages_size - | |
168 (total_size + free_space) * constants::kOfflinePageStorageClearThreshold; | |
169 if (space_to_release > 0) { | |
170 // Here we're sorting the |kept_pages| with oldest first. | |
171 std::sort(kept_pages.begin(), kept_pages.end(), | |
172 [](const OfflinePageItem& a, const OfflinePageItem& b) -> bool { | |
173 return a.last_access_time < b.last_access_time; | |
174 }); | |
175 size_t kept_pages_size = kept_pages.size(); | |
176 size_t pos = 0; | |
177 while (pos < kept_pages_size && space_to_release > 0) { | |
178 space_to_release -= kept_pages.at(pos).file_size; | |
179 page_ids_to_expire->push_back(kept_pages.at(pos).offline_id); | |
180 pos++; | |
181 } | |
182 } | |
183 } | |
184 | |
185 OfflinePageStorageManager::ClearMode | |
186 OfflinePageStorageManager::ShouldClearPages( | |
187 const ArchiveManager::StorageStats& storage_stats) { | |
188 int64_t total_size = storage_stats.total_archives_size; | |
189 int64_t free_space = storage_stats.free_disk_space; | |
190 if (total_size == 0) | |
191 return ClearMode::NOT_NEEDED; | |
192 | |
193 // If the size of all offline pages is more than limit, or it's larger than a | |
194 // specified percentage of all available storage space on the disk we'll clear | |
195 // all offline pages. | |
196 if (total_size >= | |
197 (total_size + free_space) * constants::kOfflinePageStorageLimit) | |
198 return ClearMode::DEFAULT; | |
199 // If it's been more than the pre-defined interval since the last time we | |
200 // clear the storage, we should clear pages. | |
201 if (last_clear_time_ == base::Time() || | |
202 clear_time_ - last_clear_time_ >= constants::kClearStorageInterval) { | |
203 return ClearMode::DEFAULT; | |
204 } | |
205 // Otherwise there's no need to clear storage right now. | |
206 return ClearMode::NOT_NEEDED; | |
207 } | |
208 | |
209 bool OfflinePageStorageManager::ShouldBeExpired( | |
210 const OfflinePageItem& page) const { | |
211 const LifetimePolicy& policy = | |
212 policy_controller_->GetPolicy(page.client_id.name_space).lifetime_policy; | |
213 return policy.lifetime_type == LifetimeType::TEMPORARY && | |
214 clear_time_ - page.last_access_time >= policy.expiration_period; | |
215 } | |
216 | |
217 bool OfflinePageStorageManager::IsInProgress() const { | |
218 return clear_time_ > last_clear_time_; | |
219 } | |
220 | |
221 } // namespace offline_pages | |
OLD | NEW |