Disk cache: Protect the code against misuse......

net/disk_cache/in_flight_io.h

 // 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.
 
 #ifndef NET_DISK_CACHE_IN_FLIGHT_IO_H_
 #define NET_DISK_CACHE_IN_FLIGHT_IO_H_
 
 #include <set>
 
-#include "base/message_loop.h"
+#include "base/message_loop_proxy.h"
 #include "base/waitable_event.h"
 
 namespace disk_cache {
 
 class InFlightIO;
 
 // This class represents a single asynchronous IO operation while it is being
 // bounced between threads.
 class BackgroundIO : public base::RefCountedThreadSafe<BackgroundIO> {
  public:
@@ skipping 61 matching lines @@
 //    8.  BackgroundIO::OnIOSignalled()
 //    9.  InFlightIO::InvokeCallback()
 //   10. DerivedInFlightIO::OnOperationComplete()
 //   11.       invoke callback
 //
 // Shutdown is a special case that is handled though WaitForPendingIO() instead
 // of just waiting for step 7.
 class InFlightIO {
  public:
   InFlightIO()
-      : callback_thread_(MessageLoop::current()), running_(false),
-        single_thread_(false) {}
+      : callback_thread_(base::MessageLoopProxy::CreateForCurrentThread()),
+        running_(false), single_thread_(false) {}
   virtual ~InFlightIO() {}
 
   // Blocks the current thread until all IO operations tracked by this object
   // complete.
   void WaitForPendingIO();
 
   // Called on a background thread when |operation| completes.
   void OnIOComplete(BackgroundIO* operation);
 
   // Invokes the users' completion callback at the end of the IO operation.
@@ skipping 10 matching lines @@
 
   // Signals this object that the derived class just posted the |operation| to
   // be executed on a background thread. This method must be called on the same
   // thread used to create this object.
   void OnOperationPosted(BackgroundIO* operation);
 
  private:
   typedef std::set<scoped_refptr<BackgroundIO> > IOList;
 
   IOList io_list_;  // List of pending, in-flight io operations.
-  MessageLoop* callback_thread_;
+  scoped_refptr<base::MessageLoopProxy> callback_thread_;
 
   bool running_;  // True after the first posted operation completes.
   bool single_thread_;  // True if we only have one thread.
 
   DISALLOW_COPY_AND_ASSIGN(InFlightIO);
 };
 
 }  // namespace disk_cache
 
 #endif  // NET_DISK_CACHE_IN_FLIGHT_IO_H_