Delegate WebSocket message handlers to the loading task runner

content/child/websocket_dispatcher.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/websocket_dispatcher.h"
 
 #include <stdint.h>
 #include <map>
 
 #include "base/logging.h"
@@ skipping 17 matching lines @@
 void WebSocketDispatcher::RemoveBridge(int channel_id) {
   std::map<int, WebSocketBridge*>::iterator iter = bridges_.find(channel_id);
   if (iter == bridges_.end()) {
     DVLOG(1) << "Remove a non-existent bridge(" << channel_id << ")";
     return;
   }
   bridges_.erase(iter);
 }
 
 bool WebSocketDispatcher::OnMessageReceived(const IPC::Message& msg) {
+  if (!CanHandleMessage(msg))
+    return false;
+  WebSocketBridge* bridge = GetBridge(msg.routing_id(), msg.type());
+  if (!bridge)
+    return true;
+  return bridge->OnMessageReceived(msg);
+}
+
+bool WebSocketDispatcher::CanHandleMessage(const IPC::Message& msg) const {
   switch (msg.type()) {
     case WebSocketMsg_AddChannelResponse::ID:
     case WebSocketMsg_NotifyStartOpeningHandshake::ID:
     case WebSocketMsg_NotifyFinishOpeningHandshake::ID:
     case WebSocketMsg_NotifyFailure::ID:
     case WebSocketMsg_SendFrame::ID:
     case WebSocketMsg_FlowControl::ID:
     case WebSocketMsg_DropChannel::ID:
     case WebSocketMsg_NotifyClosing::ID:
-      break;
+      return true;
     default:
       return false;
   }
-
-  WebSocketBridge* bridge = GetBridge(msg.routing_id(), msg.type());
-  if (!bridge)
-    return true;
-  return bridge->OnMessageReceived(msg);
 }
 
 WebSocketBridge* WebSocketDispatcher::GetBridge(int channel_id, uint32_t type) {
   std::map<int, WebSocketBridge*>::iterator iter = bridges_.find(channel_id);
   if (iter == bridges_.end()) {
     DVLOG(1) << "No bridge for channel_id=" << channel_id
              << ", type=" << type;
     return NULL;
   }
   return iter->second;
 }
 
 }  // namespace content