blob: 24b0e9c693b008a88a5082c13a704ff321fc4bf0 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
// Copyright (c) 2006-2010 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "net/disk_cache/in_flight_io.h"
#include "base/logging.h"
namespace disk_cache {
// Runs on the primary thread.
void BackgroundIO::OnIOSignalled() {
if (controller_)
controller_->InvokeCallback(this, false);
}
void BackgroundIO::Cancel() {
DCHECK(controller_);
controller_ = NULL;
}
// Runs on the background thread.
void BackgroundIO::NotifyController() {
controller_->OnIOComplete(this);
}
// ---------------------------------------------------------------------------
void InFlightIO::WaitForPendingIO() {
while (!io_list_.empty()) {
// Block the current thread until all pending IO completes.
IOList::iterator it = io_list_.begin();
InvokeCallback(*it, true);
}
}
// Runs on a background thread.
void InFlightIO::OnIOComplete(BackgroundIO* operation) {
#ifndef NDEBUG
if (callback_thread_ == MessageLoop::current()) {
DCHECK(single_thread_ || !running_);
single_thread_ = true;
}
running_ = true;
#endif
callback_thread_->PostTask(FROM_HERE,
NewRunnableMethod(operation,
&BackgroundIO::OnIOSignalled));
operation->io_completed()->Signal();
}
// Runs on the primary thread.
void InFlightIO::InvokeCallback(BackgroundIO* operation, bool cancel_task) {
operation->io_completed()->Wait();
if (cancel_task)
operation->Cancel();
// Make sure that we remove the operation from the list before invoking the
// callback (so that a subsequent cancel does not invoke the callback again).
DCHECK(io_list_.find(operation) != io_list_.end());
io_list_.erase(operation);
OnOperationComplete(operation, cancel_task);
}
// Runs on the primary thread.
void InFlightIO::OnOperationPosted(BackgroundIO* operation) {
DCHECK(callback_thread_ == MessageLoop::current());
io_list_.insert(operation);
}
} // namespace disk_cache
|
