Delegate WebSocket message handlers to the loading task runner

content/child/websocket_dispatcher.h

 // 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.
 
 #ifndef CONTENT_CHILD_WEBSOCKET_DISPATCHER_H_
 #define CONTENT_CHILD_WEBSOCKET_DISPATCHER_H_
 
 #include <stdint.h>
 #include <map>
 #include <string>
 #include <vector>
 
 #include "base/compiler_specific.h"
 #include "base/macros.h"
+#include "base/memory/ref_counted.h"
+#include "base/single_thread_task_runner.h"
 #include "ipc/ipc_listener.h"
 
 namespace content {
 
 class WebSocketBridge;
 
 // Dispatches WebSocket related messages sent to a child process from the
 // main browser process.  There is one instance per child process. Messages
 // are dispatched on the main child thread.  The ChildThread class
 // creates an instance of WebSocketDispatcher and delegates calls to it.
 class WebSocketDispatcher : public IPC::Listener {
  public:
   WebSocketDispatcher();
   ~WebSocketDispatcher() override;
 
   // Returns a unique channel id
   int AddBridge(WebSocketBridge* bridge);
   void RemoveBridge(int channel_id);
 
   // IPC::Listener implementation.
   bool OnMessageReceived(const IPC::Message& msg) override;
 
+  void SetLoadingTaskRunner(scoped_refptr<base::SingleThreadTaskRunner>
+                            loading_task_runner);
+
  private:
   WebSocketBridge* GetBridge(int channel_id, uint32_t type);
+  void OnMessageReceivedOnLoadingTaskRunner(IPC::Message msg);
 
   std::map<int, WebSocketBridge*> bridges_;
   int channel_id_max_;
+  scoped_refptr<base::SingleThreadTaskRunner> loading_task_runner_;
 
   DISALLOW_COPY_AND_ASSIGN(WebSocketDispatcher);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_CHILD_WEBSOCKET_DISPATCHER_H_