Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(175)

Side by Side Diff: base/files/important_file_writer.cc

Issue 1128733002: Update from https://crrev.com/328418 (Closed) Base URL: git@github.com:domokit/mojo.git@master
Patch Set: Created 5 years, 7 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « base/files/important_file_writer.h ('k') | base/files/important_file_writer_unittest.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. 1 // Copyright (c) 2011 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 "base/files/important_file_writer.h" 5 #include "base/files/important_file_writer.h"
6 6
7 #include <stdio.h> 7 #include <stdio.h>
8 8
9 #include <string> 9 #include <string>
10 10
(...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after
44 }; 44 };
45 45
46 void LogFailure(const FilePath& path, TempFileFailure failure_code, 46 void LogFailure(const FilePath& path, TempFileFailure failure_code,
47 const std::string& message) { 47 const std::string& message) {
48 UMA_HISTOGRAM_ENUMERATION("ImportantFile.TempFileFailures", failure_code, 48 UMA_HISTOGRAM_ENUMERATION("ImportantFile.TempFileFailures", failure_code,
49 TEMP_FILE_FAILURE_MAX); 49 TEMP_FILE_FAILURE_MAX);
50 DPLOG(WARNING) << "temp file failure: " << path.value().c_str() 50 DPLOG(WARNING) << "temp file failure: " << path.value().c_str()
51 << " : " << message; 51 << " : " << message;
52 } 52 }
53 53
54 // Helper function to call WriteFileAtomically() with a scoped_ptr<std::string>.
55 bool WriteScopedStringToFileAtomically(const FilePath& path,
56 scoped_ptr<std::string> data) {
57 return ImportantFileWriter::WriteFileAtomically(path, *data);
58 }
59
54 } // namespace 60 } // namespace
55 61
56 // static 62 // static
57 bool ImportantFileWriter::WriteFileAtomically(const FilePath& path, 63 bool ImportantFileWriter::WriteFileAtomically(const FilePath& path,
58 const std::string& data) { 64 const std::string& data) {
59 #if defined(OS_CHROMEOS) 65 #if defined(OS_CHROMEOS)
60 // On Chrome OS, chrome gets killed when it cannot finish shutdown quickly, 66 // On Chrome OS, chrome gets killed when it cannot finish shutdown quickly,
61 // and this function seems to be one of the slowest shutdown steps. 67 // and this function seems to be one of the slowest shutdown steps.
62 // Include some info to the report for investigation. crbug.com/418627 68 // Include some info to the report for investigation. crbug.com/418627
63 // TODO(hashimoto): Remove this. 69 // TODO(hashimoto): Remove this.
(...skipping 53 matching lines...) Expand 10 before | Expand all | Expand 10 after
117 123
118 ImportantFileWriter::ImportantFileWriter( 124 ImportantFileWriter::ImportantFileWriter(
119 const FilePath& path, 125 const FilePath& path,
120 const scoped_refptr<base::SequencedTaskRunner>& task_runner) 126 const scoped_refptr<base::SequencedTaskRunner>& task_runner)
121 : path_(path), 127 : path_(path),
122 task_runner_(task_runner), 128 task_runner_(task_runner),
123 serializer_(NULL), 129 serializer_(NULL),
124 commit_interval_(TimeDelta::FromMilliseconds(kDefaultCommitIntervalMs)), 130 commit_interval_(TimeDelta::FromMilliseconds(kDefaultCommitIntervalMs)),
125 weak_factory_(this) { 131 weak_factory_(this) {
126 DCHECK(CalledOnValidThread()); 132 DCHECK(CalledOnValidThread());
127 DCHECK(task_runner_.get()); 133 DCHECK(task_runner_);
128 } 134 }
129 135
130 ImportantFileWriter::~ImportantFileWriter() { 136 ImportantFileWriter::~ImportantFileWriter() {
131 // We're usually a member variable of some other object, which also tends 137 // We're usually a member variable of some other object, which also tends
132 // to be our serializer. It may not be safe to call back to the parent object 138 // to be our serializer. It may not be safe to call back to the parent object
133 // being destructed. 139 // being destructed.
134 DCHECK(!HasPendingWrite()); 140 DCHECK(!HasPendingWrite());
135 } 141 }
136 142
137 bool ImportantFileWriter::HasPendingWrite() const { 143 bool ImportantFileWriter::HasPendingWrite() const {
138 DCHECK(CalledOnValidThread()); 144 DCHECK(CalledOnValidThread());
139 return timer_.IsRunning(); 145 return timer_.IsRunning();
140 } 146 }
141 147
142 void ImportantFileWriter::WriteNow(const std::string& data) { 148 void ImportantFileWriter::WriteNow(scoped_ptr<std::string> data) {
143 DCHECK(CalledOnValidThread()); 149 DCHECK(CalledOnValidThread());
144 if (data.length() > static_cast<size_t>(kint32max)) { 150 if (data->length() > static_cast<size_t>(kint32max)) {
145 NOTREACHED(); 151 NOTREACHED();
146 return; 152 return;
147 } 153 }
148 154
149 if (HasPendingWrite()) 155 if (HasPendingWrite())
150 timer_.Stop(); 156 timer_.Stop();
151 157
152 if (!PostWriteTask(data)) { 158 auto task = Bind(&WriteScopedStringToFileAtomically, path_, Passed(&data));
159 if (!PostWriteTask(task)) {
153 // Posting the task to background message loop is not expected 160 // Posting the task to background message loop is not expected
154 // to fail, but if it does, avoid losing data and just hit the disk 161 // to fail, but if it does, avoid losing data and just hit the disk
155 // on the current thread. 162 // on the current thread.
156 NOTREACHED(); 163 NOTREACHED();
157 164
158 WriteFileAtomically(path_, data); 165 task.Run();
159 } 166 }
160 } 167 }
161 168
162 void ImportantFileWriter::ScheduleWrite(DataSerializer* serializer) { 169 void ImportantFileWriter::ScheduleWrite(DataSerializer* serializer) {
163 DCHECK(CalledOnValidThread()); 170 DCHECK(CalledOnValidThread());
164 171
165 DCHECK(serializer); 172 DCHECK(serializer);
166 serializer_ = serializer; 173 serializer_ = serializer;
167 174
168 if (!timer_.IsRunning()) { 175 if (!timer_.IsRunning()) {
169 timer_.Start(FROM_HERE, commit_interval_, this, 176 timer_.Start(FROM_HERE, commit_interval_, this,
170 &ImportantFileWriter::DoScheduledWrite); 177 &ImportantFileWriter::DoScheduledWrite);
171 } 178 }
172 } 179 }
173 180
174 void ImportantFileWriter::DoScheduledWrite() { 181 void ImportantFileWriter::DoScheduledWrite() {
175 DCHECK(serializer_); 182 DCHECK(serializer_);
176 std::string data; 183 scoped_ptr<std::string> data(new std::string);
177 if (serializer_->SerializeData(&data)) { 184 if (serializer_->SerializeData(data.get())) {
178 WriteNow(data); 185 WriteNow(data.Pass());
179 } else { 186 } else {
180 DLOG(WARNING) << "failed to serialize data to be saved in " 187 DLOG(WARNING) << "failed to serialize data to be saved in "
181 << path_.value().c_str(); 188 << path_.value().c_str();
182 } 189 }
183 serializer_ = NULL; 190 serializer_ = NULL;
184 } 191 }
185 192
186 void ImportantFileWriter::RegisterOnNextSuccessfulWriteCallback( 193 void ImportantFileWriter::RegisterOnNextSuccessfulWriteCallback(
187 const base::Closure& on_next_successful_write) { 194 const base::Closure& on_next_successful_write) {
188 DCHECK(on_next_successful_write_.is_null()); 195 DCHECK(on_next_successful_write_.is_null());
189 on_next_successful_write_ = on_next_successful_write; 196 on_next_successful_write_ = on_next_successful_write;
190 } 197 }
191 198
192 bool ImportantFileWriter::PostWriteTask(const std::string& data) { 199 bool ImportantFileWriter::PostWriteTask(const Callback<bool()>& task) {
193 // TODO(gab): This code could always use PostTaskAndReplyWithResult and let 200 // TODO(gab): This code could always use PostTaskAndReplyWithResult and let
194 // ForwardSuccessfulWrite() no-op if |on_next_successful_write_| is null, but 201 // ForwardSuccessfulWrite() no-op if |on_next_successful_write_| is null, but
195 // PostTaskAndReply causes memory leaks in tests (crbug.com/371974) and 202 // PostTaskAndReply causes memory leaks in tests (crbug.com/371974) and
196 // suppressing all of those is unrealistic hence we avoid most of them by 203 // suppressing all of those is unrealistic hence we avoid most of them by
197 // using PostTask() in the typical scenario below. 204 // using PostTask() in the typical scenario below.
198 if (!on_next_successful_write_.is_null()) { 205 if (!on_next_successful_write_.is_null()) {
199 return base::PostTaskAndReplyWithResult( 206 return base::PostTaskAndReplyWithResult(
200 task_runner_.get(), 207 task_runner_.get(),
201 FROM_HERE, 208 FROM_HERE,
202 MakeCriticalClosure( 209 MakeCriticalClosure(task),
203 Bind(&ImportantFileWriter::WriteFileAtomically, path_, data)),
204 Bind(&ImportantFileWriter::ForwardSuccessfulWrite, 210 Bind(&ImportantFileWriter::ForwardSuccessfulWrite,
205 weak_factory_.GetWeakPtr())); 211 weak_factory_.GetWeakPtr()));
206 } 212 }
207 return task_runner_->PostTask( 213 return task_runner_->PostTask(
208 FROM_HERE, 214 FROM_HERE,
209 MakeCriticalClosure( 215 MakeCriticalClosure(base::Bind(IgnoreResult(task))));
210 Bind(IgnoreResult(&ImportantFileWriter::WriteFileAtomically),
211 path_, data)));
212 } 216 }
213 217
214 void ImportantFileWriter::ForwardSuccessfulWrite(bool result) { 218 void ImportantFileWriter::ForwardSuccessfulWrite(bool result) {
215 DCHECK(CalledOnValidThread()); 219 DCHECK(CalledOnValidThread());
216 if (result && !on_next_successful_write_.is_null()) { 220 if (result && !on_next_successful_write_.is_null()) {
217 on_next_successful_write_.Run(); 221 on_next_successful_write_.Run();
218 on_next_successful_write_.Reset(); 222 on_next_successful_write_.Reset();
219 } 223 }
220 } 224 }
221 225
222 } // namespace base 226 } // namespace base
OLDNEW
« no previous file with comments | « base/files/important_file_writer.h ('k') | base/files/important_file_writer_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698