| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "content/browser/leveldb_wrapper_impl.h" | 5 #include "content/browser/leveldb_wrapper_impl.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/threading/thread_task_runner_handle.h" | 8 #include "base/threading/thread_task_runner_handle.h" |
| 9 #include "components/leveldb/public/cpp/util.h" | 9 #include "components/leveldb/public/cpp/util.h" |
| 10 #include "content/public/browser/browser_thread.h" | 10 #include "content/public/browser/browser_thread.h" |
| (...skipping 53 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 64 } | 64 } |
| 65 | 65 |
| 66 void LevelDBWrapperImpl::Bind(mojom::LevelDBWrapperRequest request) { | 66 void LevelDBWrapperImpl::Bind(mojom::LevelDBWrapperRequest request) { |
| 67 bindings_.AddBinding(this, std::move(request)); | 67 bindings_.AddBinding(this, std::move(request)); |
| 68 } | 68 } |
| 69 | 69 |
| 70 void LevelDBWrapperImpl::EnableAggressiveCommitDelay() { | 70 void LevelDBWrapperImpl::EnableAggressiveCommitDelay() { |
| 71 s_aggressive_flushing_enabled_ = true; | 71 s_aggressive_flushing_enabled_ = true; |
| 72 } | 72 } |
| 73 | 73 |
| 74 void LevelDBWrapperImpl::ScheduleImmediateCommit() { |
| 75 if (!on_load_complete_tasks_.empty()) { |
| 76 LoadMap(base::Bind(&LevelDBWrapperImpl::ScheduleImmediateCommit, |
| 77 base::Unretained(this))); |
| 78 return; |
| 79 } |
| 80 |
| 81 if (!database_ || !commit_batch_) |
| 82 return; |
| 83 CommitChanges(); |
| 84 } |
| 85 |
| 74 void LevelDBWrapperImpl::AddObserver( | 86 void LevelDBWrapperImpl::AddObserver( |
| 75 mojom::LevelDBObserverAssociatedPtrInfo observer) { | 87 mojom::LevelDBObserverAssociatedPtrInfo observer) { |
| 76 mojom::LevelDBObserverAssociatedPtr observer_ptr; | 88 mojom::LevelDBObserverAssociatedPtr observer_ptr; |
| 77 observer_ptr.Bind(std::move(observer)); | 89 observer_ptr.Bind(std::move(observer)); |
| 78 observers_.AddPtr(std::move(observer_ptr)); | 90 observers_.AddPtr(std::move(observer_ptr)); |
| 79 } | 91 } |
| 80 | 92 |
| 81 void LevelDBWrapperImpl::Put(const std::vector<uint8_t>& key, | 93 void LevelDBWrapperImpl::Put(const std::vector<uint8_t>& key, |
| 82 const std::vector<uint8_t>& value, | 94 const std::vector<uint8_t>& value, |
| 83 const std::string& source, | 95 const std::string& source, |
| (...skipping 77 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 161 bytes_used_ -= key.size() + old_value.size(); | 173 bytes_used_ -= key.size() + old_value.size(); |
| 162 observers_.ForAllPtrs( | 174 observers_.ForAllPtrs( |
| 163 [&key, &source, &old_value](mojom::LevelDBObserver* observer) { | 175 [&key, &source, &old_value](mojom::LevelDBObserver* observer) { |
| 164 observer->KeyDeleted(key, old_value, source); | 176 observer->KeyDeleted(key, old_value, source); |
| 165 }); | 177 }); |
| 166 callback.Run(true); | 178 callback.Run(true); |
| 167 } | 179 } |
| 168 | 180 |
| 169 void LevelDBWrapperImpl::DeleteAll(const std::string& source, | 181 void LevelDBWrapperImpl::DeleteAll(const std::string& source, |
| 170 const DeleteAllCallback& callback) { | 182 const DeleteAllCallback& callback) { |
| 171 if (!map_) { | 183 if (!map_ && !on_load_complete_tasks_.empty()) { |
| 172 LoadMap( | 184 LoadMap( |
| 173 base::Bind(&LevelDBWrapperImpl::DeleteAll, base::Unretained(this), | 185 base::Bind(&LevelDBWrapperImpl::DeleteAll, base::Unretained(this), |
| 174 source, callback)); | 186 source, callback)); |
| 175 return; | 187 return; |
| 176 } | 188 } |
| 177 | 189 |
| 178 if (database_ && !map_->empty()) { | 190 if (database_ && (!map_ || !map_->empty())) { |
| 179 CreateCommitBatchIfNeeded(); | 191 CreateCommitBatchIfNeeded(); |
| 180 commit_batch_->clear_all_first = true; | 192 commit_batch_->clear_all_first = true; |
| 181 commit_batch_->changed_keys.clear(); | 193 commit_batch_->changed_keys.clear(); |
| 182 } | 194 } |
| 195 if (!map_) |
| 196 map_.reset(new ValueMap); |
| 197 |
| 183 map_->clear(); | 198 map_->clear(); |
| 184 bytes_used_ = 0; | 199 bytes_used_ = 0; |
| 185 observers_.ForAllPtrs( | 200 observers_.ForAllPtrs( |
| 186 [&source](mojom::LevelDBObserver* observer) { | 201 [&source](mojom::LevelDBObserver* observer) { |
| 187 observer->AllDeleted(source); | 202 observer->AllDeleted(source); |
| 188 }); | 203 }); |
| 189 callback.Run(true); | 204 callback.Run(true); |
| 190 } | 205 } |
| 191 | 206 |
| 192 void LevelDBWrapperImpl::Get(const std::vector<uint8_t>& key, | 207 void LevelDBWrapperImpl::Get(const std::vector<uint8_t>& key, |
| (...skipping 183 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 376 weak_ptr_factory_.GetWeakPtr())); | 391 weak_ptr_factory_.GetWeakPtr())); |
| 377 } | 392 } |
| 378 | 393 |
| 379 void LevelDBWrapperImpl::OnCommitComplete(leveldb::mojom::DatabaseError error) { | 394 void LevelDBWrapperImpl::OnCommitComplete(leveldb::mojom::DatabaseError error) { |
| 380 // TODO(michaeln): What if it fails, uma here or in the DB class? | 395 // TODO(michaeln): What if it fails, uma here or in the DB class? |
| 381 --commit_batches_in_flight_; | 396 --commit_batches_in_flight_; |
| 382 StartCommitTimer(); | 397 StartCommitTimer(); |
| 383 } | 398 } |
| 384 | 399 |
| 385 } // namespace content | 400 } // namespace content |
| OLD | NEW |