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