| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2012 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 "net/disk_cache/in_flight_io.h" | |
| 6 | |
| 7 #include "base/bind.h" | |
| 8 #include "base/location.h" | |
| 9 #include "base/logging.h" | |
| 10 #include "base/threading/thread_restrictions.h" | |
| 11 | |
| 12 namespace disk_cache { | |
| 13 | |
| 14 BackgroundIO::BackgroundIO(InFlightIO* controller) | |
| 15 : result_(-1), io_completed_(true, false), controller_(controller) { | |
| 16 } | |
| 17 | |
| 18 // Runs on the primary thread. | |
| 19 void BackgroundIO::OnIOSignalled() { | |
| 20 if (controller_) | |
| 21 controller_->InvokeCallback(this, false); | |
| 22 } | |
| 23 | |
| 24 void BackgroundIO::Cancel() { | |
| 25 // controller_ may be in use from the background thread at this time. | |
| 26 base::AutoLock lock(controller_lock_); | |
| 27 DCHECK(controller_); | |
| 28 controller_ = NULL; | |
| 29 } | |
| 30 | |
| 31 BackgroundIO::~BackgroundIO() { | |
| 32 } | |
| 33 | |
| 34 // --------------------------------------------------------------------------- | |
| 35 | |
| 36 InFlightIO::InFlightIO() | |
| 37 : callback_thread_(base::MessageLoopProxy::current()), | |
| 38 running_(false), single_thread_(false) { | |
| 39 } | |
| 40 | |
| 41 InFlightIO::~InFlightIO() { | |
| 42 } | |
| 43 | |
| 44 // Runs on the background thread. | |
| 45 void BackgroundIO::NotifyController() { | |
| 46 base::AutoLock lock(controller_lock_); | |
| 47 if (controller_) | |
| 48 controller_->OnIOComplete(this); | |
| 49 } | |
| 50 | |
| 51 void InFlightIO::WaitForPendingIO() { | |
| 52 while (!io_list_.empty()) { | |
| 53 // Block the current thread until all pending IO completes. | |
| 54 IOList::iterator it = io_list_.begin(); | |
| 55 InvokeCallback(it->get(), true); | |
| 56 } | |
| 57 } | |
| 58 | |
| 59 void InFlightIO::DropPendingIO() { | |
| 60 while (!io_list_.empty()) { | |
| 61 IOList::iterator it = io_list_.begin(); | |
| 62 BackgroundIO* operation = it->get(); | |
| 63 operation->Cancel(); | |
| 64 DCHECK(io_list_.find(operation) != io_list_.end()); | |
| 65 io_list_.erase(make_scoped_refptr(operation)); | |
| 66 } | |
| 67 } | |
| 68 | |
| 69 // Runs on a background thread. | |
| 70 void InFlightIO::OnIOComplete(BackgroundIO* operation) { | |
| 71 #ifndef NDEBUG | |
| 72 if (callback_thread_->BelongsToCurrentThread()) { | |
| 73 DCHECK(single_thread_ || !running_); | |
| 74 single_thread_ = true; | |
| 75 } | |
| 76 #endif | |
| 77 | |
| 78 callback_thread_->PostTask(FROM_HERE, | |
| 79 base::Bind(&BackgroundIO::OnIOSignalled, | |
| 80 operation)); | |
| 81 operation->io_completed()->Signal(); | |
| 82 } | |
| 83 | |
| 84 // Runs on the primary thread. | |
| 85 void InFlightIO::InvokeCallback(BackgroundIO* operation, bool cancel_task) { | |
| 86 { | |
| 87 // http://crbug.com/74623 | |
| 88 base::ThreadRestrictions::ScopedAllowWait allow_wait; | |
| 89 operation->io_completed()->Wait(); | |
| 90 } | |
| 91 running_ = true; | |
| 92 | |
| 93 if (cancel_task) | |
| 94 operation->Cancel(); | |
| 95 | |
| 96 // Make sure that we remove the operation from the list before invoking the | |
| 97 // callback (so that a subsequent cancel does not invoke the callback again). | |
| 98 DCHECK(io_list_.find(operation) != io_list_.end()); | |
| 99 DCHECK(!operation->HasOneRef()); | |
| 100 io_list_.erase(make_scoped_refptr(operation)); | |
| 101 OnOperationComplete(operation, cancel_task); | |
| 102 } | |
| 103 | |
| 104 // Runs on the primary thread. | |
| 105 void InFlightIO::OnOperationPosted(BackgroundIO* operation) { | |
| 106 DCHECK(callback_thread_->BelongsToCurrentThread()); | |
| 107 io_list_.insert(make_scoped_refptr(operation)); | |
| 108 } | |
| 109 | |
| 110 } // namespace disk_cache | |
| OLD | NEW |