[DO NOT COMMIT] mojo datapipe performance measurement

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"
+#include "content/child/child_thread_impl.h"
 #include "content/child/websocket_bridge.h"
 #include "content/common/websocket_messages.h"
+#include "content/public/common/service_registry.h"
 #include "ipc/ipc_message.h"
 #include "url/gurl.h"
 
 namespace content {
 
 WebSocketDispatcher::WebSocketDispatcher()
   : channel_id_max_(0),
-    weak_ptr_factory_(this) {}
+    weak_ptr_factory_(this) {
+  ServiceRegistry* registry = ChildThreadImpl::current()->service_registry();
+  registry->ConnectToRemoteService(mojo::GetProxy(&loader_test_service_));
+}
 
 WebSocketDispatcher::~WebSocketDispatcher() {}
 
 bool WebSocketDispatcher::CanHandleMessage(const IPC::Message& msg) {
   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:
+    case WebSocketMsg_LoaderTransferTest_SetDataBuffer::ID:
+    case WebSocketMsg_LoaderTransferTest_Ack::ID:
+    case WebSocketMsg_LoaderTransferTest_Done::ID:
       return true;
     default:
       return false;
   }
 }
 
 int WebSocketDispatcher::AddBridge(WebSocketBridge* bridge) {
   ++channel_id_max_;
   bridges_.insert(std::make_pair(channel_id_max_, bridge));
   return channel_id_max_;
 }
 
 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);
 }
 
 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