Implement Worker-MainThread bridge for FileWriter in Chrome

content/child/fileapi/webfilesystem_impl.cc

 // Copyright 2013 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 "content/child/fileapi/webfilesystem_impl.h"
 
 #include "base/bind.h"
 #include "base/id_map.h"
 #include "base/lazy_instance.h"
 #include "base/logging.h"
@@ skipping 76 matching lines @@
  private:
   CallbacksMap() {
     g_callbacks_map_tls.Pointer()->Set(this);
   }
 
   IDMap<WebFileSystemCallbacks> callbacks_;
 
   DISALLOW_COPY_AND_ASSIGN(CallbacksMap);
 };
 
-void DidReadMetadataForCreateFileWriter(
-    const GURL& path,
-    WebKit::WebFileWriterClient* client,
-    WebKit::WebFileSystemCallbacks* callbacks,
-    const base::PlatformFileInfo& file_info) {
-  if (file_info.is_directory || file_info.size < 0) {
-    callbacks->didFail(WebKit::WebFileErrorInvalidState);
-    return;
-  }
-  callbacks->didCreateFileWriter(new WebFileWriterImpl(path, client),
-                                 file_info.size);
-}
-
 void DidReceiveSnapshotFile(int request_id) {
   if (ChildThread::current())
     ChildThread::current()->Send(
         new FileSystemHostMsg_DidReceiveSnapshotFile(request_id));
 }
 
 int CurrentWorkerId() {
   return WorkerTaskRunner::Instance()->CurrentWorkerId();
 }
 
@@ skipping 68 matching lines @@
     file_system_entries[i].name =
         base::FilePath(entries[i].name).AsUTF16Unsafe();
     file_system_entries[i].isDirectory = entries[i].is_directory;
   }
   CallbackFileSystemCallbacks(
       thread_id, callbacks_id,
       &WebFileSystemCallbacks::didReadDirectory,
       MakeTuple(file_system_entries, has_more));
 }
 
+void CreateFileWriterCallbackAdapter(
+    int thread_id, int callbacks_id,
+    base::MessageLoopProxy* main_thread_loop,
+    const GURL& path,
+    WebKit::WebFileWriterClient* client,
+    const base::PlatformFileInfo& file_info) {
+  if (thread_id != CurrentWorkerId()) {
+    WorkerTaskRunner::Instance()->PostTask(
+        thread_id,
+        base::Bind(&CreateFileWriterCallbackAdapter,
+                    thread_id, callbacks_id,
+                    make_scoped_refptr(main_thread_loop),
+                    path, client, file_info));
+    return;
+  }
+
+  if (!CallbacksMap::Get())
+    return;
+
+  WebFileSystemCallbacks* callbacks =
+      CallbacksMap::Get()->GetAndUnregisterCallbacks(callbacks_id);
+  DCHECK(callbacks);
+
+  if (file_info.is_directory || file_info.size < 0) {
+    callbacks->didFail(WebKit::WebFileErrorInvalidState);
+    return;
+  }
+  callbacks->didCreateFileWriter(
+      new WebFileWriterImpl(path, client, main_thread_loop), file_info.size);
+}
+
 void CreateSnapshotFileCallbackAdapter(
     int thread_id, int callbacks_id,
     base::MessageLoopProxy* main_thread_loop,
     const base::PlatformFileInfo& file_info,
     const base::FilePath& platform_path,
     int request_id) {
   if (thread_id != CurrentWorkerId()) {
     WorkerTaskRunner::Instance()->PostTask(
         thread_id,
         base::Bind(&CreateSnapshotFileCallbackAdapter,
@@ skipping 153 matching lines @@
       &FileSystemDispatcher::ReadDirectory,
       MakeTuple(GURL(path),
                 base::Bind(&ReadDirectoryCallbackAdapater,
                            CurrentWorkerId(), callbacks_id),
                 base::Bind(&StatusCallbackAdapter,
                            CurrentWorkerId(), callbacks_id)));
 }
 
 WebKit::WebFileWriter* WebFileSystemImpl::createFileWriter(
     const WebURL& path, WebKit::WebFileWriterClient* client) {
-  return new WebFileWriterImpl(GURL(path), client);
+  return new WebFileWriterImpl(GURL(path), client, main_thread_loop_.get());
 }
 
 void WebFileSystemImpl::createFileWriter(
     const WebURL& path,
     WebKit::WebFileWriterClient* client,
     WebKit::WebFileSystemCallbacks* callbacks) {
-  // TODO(kinuko): Convert this method to use bridge model. (crbug.com/257349)
-  DCHECK(main_thread_loop_->RunsTasksOnCurrentThread());
-  FileSystemDispatcher* dispatcher =
-      ChildThread::current()->file_system_dispatcher();
-  dispatcher->ReadMetadata(
-      GURL(path),
-      base::Bind(&DidReadMetadataForCreateFileWriter,
-                 GURL(path), client, callbacks),
-      base::Bind(&FileStatusCallbackAdapter, callbacks));
+  int callbacks_id = CallbacksMap::GetOrCreate()->RegisterCallbacks(callbacks);
+  CallDispatcherOnMainThread(
+      main_thread_loop_.get(),
+      &FileSystemDispatcher::ReadMetadata,
+      MakeTuple(GURL(path),
+                base::Bind(&CreateFileWriterCallbackAdapter,
+                           CurrentWorkerId(), callbacks_id, main_thread_loop_,
+                           GURL(path), client),
+                base::Bind(&StatusCallbackAdapter,
+                           CurrentWorkerId(), callbacks_id)));
 }
 
 void WebFileSystemImpl::createSnapshotFileAndReadMetadata(
     const WebKit::WebURL& path,
     WebKit::WebFileSystemCallbacks* callbacks) {
   int callbacks_id = CallbacksMap::GetOrCreate()->RegisterCallbacks(callbacks);
   CallDispatcherOnMainThread(
       main_thread_loop_.get(),
       &FileSystemDispatcher::CreateSnapshotFile,
       MakeTuple(GURL(path),
                 base::Bind(&CreateSnapshotFileCallbackAdapter,
                            CurrentWorkerId(), callbacks_id,
                            main_thread_loop_),
                 base::Bind(&StatusCallbackAdapter,
                            CurrentWorkerId(), callbacks_id)));
 }
 
 }  // namespace content