OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 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 "chrome/browser/prefs/tracked/segregated_pref_store.h" | 5 #include "chrome/browser/prefs/tracked/segregated_pref_store.h" |
6 | 6 |
7 #include "base/logging.h" | 7 #include "base/logging.h" |
8 #include "base/stl_util.h" | 8 #include "base/stl_util.h" |
9 #include "base/values.h" | 9 #include "base/values.h" |
10 | 10 |
(...skipping 17 matching lines...) Expand all Loading... |
28 void SegregatedPrefStore::AggregatingObserver::OnInitializationCompleted( | 28 void SegregatedPrefStore::AggregatingObserver::OnInitializationCompleted( |
29 bool succeeded) { | 29 bool succeeded) { |
30 if (succeeded) | 30 if (succeeded) |
31 ++successful_sub_initializations_; | 31 ++successful_sub_initializations_; |
32 else | 32 else |
33 ++failed_sub_initializations_; | 33 ++failed_sub_initializations_; |
34 | 34 |
35 DCHECK_LE(failed_sub_initializations_ + successful_sub_initializations_, 2); | 35 DCHECK_LE(failed_sub_initializations_ + successful_sub_initializations_, 2); |
36 | 36 |
37 if (failed_sub_initializations_ + successful_sub_initializations_ == 2) { | 37 if (failed_sub_initializations_ + successful_sub_initializations_ == 2) { |
| 38 if (!outer_->on_initialization_.is_null()) |
| 39 outer_->on_initialization_.Run(); |
| 40 |
38 if (successful_sub_initializations_ == 2 && outer_->read_error_delegate_) { | 41 if (successful_sub_initializations_ == 2 && outer_->read_error_delegate_) { |
39 PersistentPrefStore::PrefReadError read_error = outer_->GetReadError(); | 42 PersistentPrefStore::PrefReadError read_error = outer_->GetReadError(); |
40 if (read_error != PersistentPrefStore::PREF_READ_ERROR_NONE) | 43 if (read_error != PersistentPrefStore::PREF_READ_ERROR_NONE) |
41 outer_->read_error_delegate_->OnError(read_error); | 44 outer_->read_error_delegate_->OnError(read_error); |
42 } | 45 } |
43 | 46 |
44 FOR_EACH_OBSERVER( | 47 FOR_EACH_OBSERVER( |
45 PrefStore::Observer, | 48 PrefStore::Observer, |
46 outer_->observers_, | 49 outer_->observers_, |
47 OnInitializationCompleted(successful_sub_initializations_ == 2)); | 50 OnInitializationCompleted(successful_sub_initializations_ == 2)); |
48 } | 51 } |
49 } | 52 } |
50 | 53 |
51 SegregatedPrefStore::SegregatedPrefStore( | 54 SegregatedPrefStore::SegregatedPrefStore( |
52 const scoped_refptr<PersistentPrefStore>& default_pref_store, | 55 const scoped_refptr<PersistentPrefStore>& default_pref_store, |
53 const scoped_refptr<PersistentPrefStore>& selected_pref_store, | 56 const scoped_refptr<PersistentPrefStore>& selected_pref_store, |
54 const std::set<std::string>& selected_pref_names) | 57 const std::set<std::string>& selected_pref_names, |
| 58 const base::Closure& on_initialization) |
55 : default_pref_store_(default_pref_store), | 59 : default_pref_store_(default_pref_store), |
56 selected_pref_store_(selected_pref_store), | 60 selected_pref_store_(selected_pref_store), |
57 selected_preference_names_(selected_pref_names), | 61 selected_preference_names_(selected_pref_names), |
| 62 on_initialization_(on_initialization), |
58 aggregating_observer_(this) { | 63 aggregating_observer_(this) { |
| 64 |
59 default_pref_store_->AddObserver(&aggregating_observer_); | 65 default_pref_store_->AddObserver(&aggregating_observer_); |
60 selected_pref_store_->AddObserver(&aggregating_observer_); | 66 selected_pref_store_->AddObserver(&aggregating_observer_); |
61 } | 67 } |
62 | 68 |
63 void SegregatedPrefStore::AddObserver(Observer* observer) { | 69 void SegregatedPrefStore::AddObserver(Observer* observer) { |
64 observers_.AddObserver(observer); | 70 observers_.AddObserver(observer); |
65 } | 71 } |
66 | 72 |
67 void SegregatedPrefStore::RemoveObserver(Observer* observer) { | 73 void SegregatedPrefStore::RemoveObserver(Observer* observer) { |
68 observers_.RemoveObserver(observer); | 74 observers_.RemoveObserver(observer); |
(...skipping 46 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
115 if (read_error == PersistentPrefStore::PREF_READ_ERROR_NONE) { | 121 if (read_error == PersistentPrefStore::PREF_READ_ERROR_NONE) { |
116 read_error = selected_pref_store_->GetReadError(); | 122 read_error = selected_pref_store_->GetReadError(); |
117 // Ignore NO_FILE from selected_pref_store_. | 123 // Ignore NO_FILE from selected_pref_store_. |
118 if (read_error == PersistentPrefStore::PREF_READ_ERROR_NO_FILE) | 124 if (read_error == PersistentPrefStore::PREF_READ_ERROR_NO_FILE) |
119 read_error = PersistentPrefStore::PREF_READ_ERROR_NONE; | 125 read_error = PersistentPrefStore::PREF_READ_ERROR_NONE; |
120 } | 126 } |
121 return read_error; | 127 return read_error; |
122 } | 128 } |
123 | 129 |
124 PersistentPrefStore::PrefReadError SegregatedPrefStore::ReadPrefs() { | 130 PersistentPrefStore::PrefReadError SegregatedPrefStore::ReadPrefs() { |
125 // Note: Both of these stores own PrefFilters which makes ReadPrefs | |
126 // asynchronous. This is okay in this case as only the first call will be | |
127 // truly asynchronous, the second call will then unblock the migration in | |
128 // TrackedPreferencesMigrator and complete synchronously. | |
129 default_pref_store_->ReadPrefs(); | 131 default_pref_store_->ReadPrefs(); |
130 PersistentPrefStore::PrefReadError selected_store_read_error = | 132 selected_pref_store_->ReadPrefs(); |
131 selected_pref_store_->ReadPrefs(); | |
132 DCHECK_NE(PersistentPrefStore::PREF_READ_ERROR_ASYNCHRONOUS_TASK_INCOMPLETE, | |
133 selected_store_read_error); | |
134 | 133 |
135 return GetReadError(); | 134 return GetReadError(); |
136 } | 135 } |
137 | 136 |
138 void SegregatedPrefStore::ReadPrefsAsync(ReadErrorDelegate* error_delegate) { | 137 void SegregatedPrefStore::ReadPrefsAsync(ReadErrorDelegate* error_delegate) { |
139 read_error_delegate_.reset(error_delegate); | 138 read_error_delegate_.reset(error_delegate); |
140 default_pref_store_->ReadPrefsAsync(NULL); | 139 default_pref_store_->ReadPrefsAsync(NULL); |
141 selected_pref_store_->ReadPrefsAsync(NULL); | 140 selected_pref_store_->ReadPrefsAsync(NULL); |
142 } | 141 } |
143 | 142 |
144 void SegregatedPrefStore::CommitPendingWrite() { | 143 void SegregatedPrefStore::CommitPendingWrite() { |
145 default_pref_store_->CommitPendingWrite(); | 144 default_pref_store_->CommitPendingWrite(); |
146 selected_pref_store_->CommitPendingWrite(); | 145 selected_pref_store_->CommitPendingWrite(); |
147 } | 146 } |
148 | 147 |
149 SegregatedPrefStore::~SegregatedPrefStore() { | 148 SegregatedPrefStore::~SegregatedPrefStore() { |
150 default_pref_store_->RemoveObserver(&aggregating_observer_); | 149 default_pref_store_->RemoveObserver(&aggregating_observer_); |
151 selected_pref_store_->RemoveObserver(&aggregating_observer_); | 150 selected_pref_store_->RemoveObserver(&aggregating_observer_); |
152 } | 151 } |
153 | 152 |
154 PersistentPrefStore* SegregatedPrefStore::StoreForKey(const std::string& key) { | |
155 return ContainsKey(selected_preference_names_, key) ? selected_pref_store_ | |
156 : default_pref_store_; | |
157 } | |
158 | |
159 const PersistentPrefStore* SegregatedPrefStore::StoreForKey( | 153 const PersistentPrefStore* SegregatedPrefStore::StoreForKey( |
160 const std::string& key) const { | 154 const std::string& key) const { |
161 return ContainsKey(selected_preference_names_, key) ? selected_pref_store_ | 155 if (ContainsKey(selected_preference_names_, key) || |
162 : default_pref_store_; | 156 selected_pref_store_->GetValue(key, NULL)) { |
| 157 return selected_pref_store_.get(); |
| 158 } |
| 159 return default_pref_store_.get(); |
163 } | 160 } |
| 161 |
| 162 PersistentPrefStore* SegregatedPrefStore::StoreForKey(const std::string& key) { |
| 163 if (ContainsKey(selected_preference_names_, key)) |
| 164 return selected_pref_store_.get(); |
| 165 |
| 166 // Check if this unselected value was previously selected. If so, migrate it |
| 167 // back to the unselected store. |
| 168 // It's hard to do this in a single pass at startup because PrefStore does not |
| 169 // permit us to enumerate its contents. |
| 170 const base::Value* value = NULL; |
| 171 if (selected_pref_store_->GetValue(key, &value)) { |
| 172 default_pref_store_->SetValue(key, value->DeepCopy()); |
| 173 // Commit |default_pref_store_| to guarantee that the migrated value is |
| 174 // flushed to disk before the removal from |selected_pref_store_| is |
| 175 // eventually flushed to disk. |
| 176 default_pref_store_->CommitPendingWrite(); |
| 177 |
| 178 value = NULL; |
| 179 selected_pref_store_->RemoveValue(key); |
| 180 } |
| 181 |
| 182 return default_pref_store_.get(); |
| 183 } |
OLD | NEW |