Worker: Enable UseCounter for SharedWorkerGlobalScope

content/renderer/shared_worker/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/shared_worker/websharedworker_proxy.h"
 
 #include <stddef.h>
 
 #include "content/child/child_thread_impl.h"
 #include "content/child/webmessageportchannel_impl.h"
 #include "content/common/view_messages.h"
 #include "content/common/worker_messages.h"
+#include "content/common/worker_use_counter.h"
 #include "ipc/message_router.h"
 
 namespace content {
 
 WebSharedWorkerProxy::WebSharedWorkerProxy(
     std::unique_ptr<blink::WebSharedWorkerConnectListener> listener,
     ViewHostMsg_CreateWorker_Params params,
     blink::WebMessagePortChannel* channel)
     : route_id_(MSG_ROUTING_NONE),
       router_(ChildThreadImpl::current()->GetRouter()),
@@ skipping 30 matching lines @@
 }
 
 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_HANDLER(ViewMsg_WorkerDestroyed, OnWorkerDestroyed)
+    IPC_MESSAGE_HANDLER(ViewMsg_CountFeatureOnSharedWorker, OnCountFeature)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   return handled;
 }
 
 void WebSharedWorkerProxy::OnWorkerCreated() {
   // The worker is created - now send off the connection request.
   router_->Send(
       new ViewHostMsg_ConnectToWorker(route_id_, message_port_id_));
 }
 
 void WebSharedWorkerProxy::OnWorkerScriptLoadFailed() {
   listener_->scriptLoadFailed();
   delete this;
 }
 
-void WebSharedWorkerProxy::OnWorkerConnected() {
+void WebSharedWorkerProxy::OnWorkerConnected(std::vector<char> features) {
   listener_->connected();
+  WorkerUseCounter counter(features);
+  for (size_t i = 0; i < counter.GetNumberOfBits(); ++i) {
+    if (!counter.IsCounted(i))
+      continue;
+    listener_->countFeature(i);
+  }
+}
+
+void WebSharedWorkerProxy::OnWorkerDestroyed() {
   delete this;
 }
 
+void WebSharedWorkerProxy::OnCountFeature(uint32_t feature) {
+  listener_->countFeature(feature);
+}
+
 }  // namespace content