HTML MessagePort as mojo::MessagePipeHandle

content/renderer/websharedworker_proxy.cc

 // Copyright (c) 2011 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/renderer/websharedworker_proxy.h"
 
 #include <stddef.h>
 
 #include "content/child/webmessageportchannel_impl.h"
 #include "content/common/view_messages.h"
@@ skipping 11 matching lines @@
       connect_listener_(nullptr),
       created_(false) {
   DCHECK_NE(MSG_ROUTING_NONE, route_id_);
   router_->AddRoute(route_id_, this);
 }
 
 WebSharedWorkerProxy::~WebSharedWorkerProxy() {
   router_->RemoveRoute(route_id_);
 }
 
-bool WebSharedWorkerProxy::Send(std::unique_ptr<IPC::Message> message) {
-  // The worker object can be interacted with before the browser process told us
-  // that it started, in which case we want to queue the message.
-  if (!created_) {
-    queued_messages_.push_back(std::move(message));
-    return true;
-  }
-
-  // For now we proxy all messages to the worker process through the browser.
-  // Revisit if we find this slow.
-  // TODO(jabdelmalek): handle sync messages if we need them.
-  return router_->Send(message.release());
-}
-
-void WebSharedWorkerProxy::SendQueuedMessages() {
-  DCHECK(created_);
-  DCHECK(queued_messages_.size());
-  std::vector<std::unique_ptr<IPC::Message>> queued_messages;
-  queued_messages.swap(queued_messages_);
-  for (size_t i = 0; i < queued_messages.size(); ++i) {
-    queued_messages[i]->set_routing_id(route_id_);
-    Send(std::move(queued_messages[i]));
-  }
-}
-
 void WebSharedWorkerProxy::connect(blink::WebMessagePortChannel* channel,
                                    ConnectListener* listener) {
-  WebMessagePortChannelImpl* webchannel =
-        static_cast<WebMessagePortChannelImpl*>(channel);
+  // Accept ownership of the channel.
+  std::unique_ptr<WebMessagePortChannelImpl> channel_impl(
+      static_cast<WebMessagePortChannelImpl*>(channel));
 
-  int message_port_id = webchannel->message_port_id();
-  DCHECK_NE(MSG_ROUTING_NONE, message_port_id);
-  webchannel->QueueMessages();
+  DCHECK(!queued_connect_message_);
 
-  Send(base::MakeUnique<ViewHostMsg_ConnectToWorker>(route_id_,
-                                                     message_port_id));
+  queued_connect_message_ = base::MakeUnique<ViewHostMsg_ConnectToWorker>(
+      route_id_, channel_impl->ReleaseMessagePort());
+  if (created_)
+    router_->Send(queued_connect_message_.release());
+
   connect_listener_ = listener;
 }
 
 bool WebSharedWorkerProxy::OnMessageReceived(const IPC::Message& message) {
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP(WebSharedWorkerProxy, message)
     IPC_MESSAGE_HANDLER(ViewMsg_WorkerCreated, OnWorkerCreated)
     IPC_MESSAGE_HANDLER(ViewMsg_WorkerScriptLoadFailed,
                         OnWorkerScriptLoadFailed)
     IPC_MESSAGE_HANDLER(ViewMsg_WorkerConnected,
                         OnWorkerConnected)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   return handled;
 }
 
 void WebSharedWorkerProxy::OnWorkerCreated() {
   created_ = true;
-  // The worker is created - now send off the WorkerMsg_Connect message.
-  SendQueuedMessages();
+
+  // The worker is created - now send off the connect message.
+  if (queued_connect_message_)
+    router_->Send(queued_connect_message_.release());
 }
 
 void WebSharedWorkerProxy::OnWorkerScriptLoadFailed() {
   if (connect_listener_) {
     // This can result in this object being freed.
     connect_listener_->scriptLoadFailed();
   }
 }
 
 void WebSharedWorkerProxy::OnWorkerConnected() {
   if (connect_listener_) {
     // This can result in this object being freed.
     connect_listener_->connected();
   }
 }
 
 }  // namespace content