Use mojo for Chrome Loading, Part 1

content/browser/loader/url_loader_factory_impl.cc

+// Copyright 2016 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/browser/loader/url_loader_factory_impl.h"
+
+#include <memory>
+
+#include "base/memory/ptr_util.h"
+#include "base/memory/weak_ptr.h"
+#include "content/browser/loader/resource_dispatcher_host_impl.h"
+#include "content/browser/loader/resource_message_filter.h"
+#include "content/common/resource_messages.h"
+#include "content/common/resource_request.h"
+#include "content/common/url_loader.mojom.h"
+
+namespace content {
+
+namespace {
+
+// This class is an implementation of mojom::URLLoader which is a resource
+// loader interface with Mojo.
+class URLLoaderImpl final : public mojom::URLLoader {
+ public:
+  // Creates a URLLoader. This function returns nothing and registers the
+  // created instance to the ResourceDispatcherHostImpl instead.
+  static void Create(scoped_refptr<ResourceMessageFilter> filter,
+                     mojom::URLLoaderRequest request) {
+    int child_id = filter->child_id();
+    ResourceDispatcherHostImpl::Get()->AddUninitiatedURLLoader(
+        child_id,
+        base::WrapUnique(new URLLoaderImpl(filter.get(), std::move(request))));
+  }
+
+  ~URLLoaderImpl() override {}
+
+  void Load(int32_t request_id,
+            const ResourceRequest& request,
+            mojom::URLLoaderClientPtr client) override {
+    DCHECK_CURRENTLY_ON(BrowserThread::IO);
+
+    ResourceMessageFilter* filter = resource_message_filter_.get();
+    if (!filter) {
+      // TODO(yhirano): Cancel the request.
+      return;
+    }
+
+    std::unique_ptr<mojom::URLLoader> that =
+        ResourceDispatcherHostImpl::Get()->TakeUninitiatedURLLoader(this);
+    DCHECK(that);
+
+    // TODO(yhirano): Provide the right routing ID.
+    const int routing_id = 0;
+    ResourceDispatcherHostImpl* rdh = ResourceDispatcherHostImpl::Get();
+    rdh->SetMojoLoaderAndClientForNextLoadRequest(std::move(that),
+                                                  std::move(client));
+    rdh->OnMessageReceived(
+        ResourceHostMsg_RequestResource(routing_id, request_id, request),
+        filter);

[kinuko, 2016/05/20 15:56:47]

Actually... is there a reason we want to call OnRequestResource via
OnMessageReceived?  If we just directly call OnRequestResourceWithMojo or
something that'd allow us to remove SetMojoLoaderAndClientForNextLoadRequest?

[yhirano, 2016/05/23 11:29:37]

Done.

+  }
+  void FollowRedirect() override {}
+  void Cancel() override {}
+
+ private:
+  URLLoaderImpl(ResourceMessageFilter* filter,
+                mojo::InterfaceRequest<mojom::URLLoader> request)
+      : resource_message_filter_(filter->GetWeakPtr()),
+        binding_(this, std::move(request)) {}
+
+  base::WeakPtr<ResourceMessageFilter> resource_message_filter_;
+  mojo::Binding<mojom::URLLoader> binding_;
+};
+
+}  // namespace
+
+URLLoaderFactoryImpl::URLLoaderFactoryImpl(
+    scoped_refptr<ResourceMessageFilter> resource_message_filter,
+    mojo::InterfaceRequest<mojom::URLLoaderFactory> request)
+    : resource_message_filter_(resource_message_filter),
+      binding_(this, std::move(request)) {}
+
+URLLoaderFactoryImpl::~URLLoaderFactoryImpl() {
+  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+}
+
+void URLLoaderFactoryImpl::CreateURLLoader(mojom::URLLoaderRequest request) {
+  BrowserThread::PostTask(
+      BrowserThread::IO, FROM_HERE,
+      base::Bind(&URLLoaderImpl::Create, resource_message_filter_,
+                 base::Passed(std::move(request))));
+}
+
+}  // namespace content