Delegate WebSocket message handlers to the loading task runner

content/child/websocket_message_filter.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/child/websocket_message_filter.h"
+
+#include "base/bind.h"
+#include "base/single_thread_task_runner.h"
+#include "content/child/websocket_dispatcher.h"
+#include "ipc/ipc_message.h"
+
+namespace content {
+
+WebSocketMessageFilter::WebSocketMessageFilter(
+    WebSocketDispatcher* websocket_dispatcher)
+  : websocket_dispatcher_(websocket_dispatcher->GetWeakPtr()) {}
+
+WebSocketMessageFilter::~WebSocketMessageFilter() {}
+
+bool WebSocketMessageFilter::OnMessageReceived(const IPC::Message& message) {
+  if (!WebSocketDispatcher::CanHandleMessage(message))
+    return false;
+  loading_task_runner_->PostTask(FROM_HERE, base::Bind(
+      &WebSocketMessageFilter::OnMessageReceivedOnMainThread,
+      this, message));
+  return true;
+}
+
+void WebSocketMessageFilter::OnMessageReceivedOnMainThread(
+    const IPC::Message& message) {
+  if (!websocket_dispatcher_.get())
+    return;
+  websocket_dispatcher_->OnMessageReceived(message);
+}
+
+}  // namespace content