Implement some mothods in SharedWorkerHost and SharedWorkerServiceImpl

content/browser/shared_worker/shared_worker_service_impl.cc

 // Copyright 2014 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/browser/shared_worker/shared_worker_service_impl.h"
 
 #include "content/browser/shared_worker/shared_worker_host.h"
 #include "content/browser/shared_worker/shared_worker_instance.h"
 #include "content/browser/shared_worker/shared_worker_message_filter.h"
 #include "content/browser/worker_host/worker_document_set.h"
@@ skipping 70 matching lines @@
       params.name,
       params.content_security_policy,
       params.security_policy_type,
       resource_context,
       partition));
   instance->AddFilter(filter, route_id);
   instance->worker_document_set()->Add(
       filter, params.document_id, filter->render_process_id(),
       params.render_frame_route_id);
 
-  scoped_ptr<SharedWorkerHost> worker(new SharedWorkerHost(instance.release()));
-  worker->Init(filter);
-  const int worker_route_id = worker->worker_route_id();
-  worker_hosts_.push_back(worker.release());
+  scoped_ptr<SharedWorkerHost> host(new SharedWorkerHost(instance.release()));
+  host->Init(filter);
+  const int worker_route_id = host->worker_route_id();
+  worker_hosts_.set(std::make_pair(filter->render_process_id(),
+                                   worker_route_id),
+                    host.Pass());
 
   FOR_EACH_OBSERVER(
       WorkerServiceObserver, observers_,
       WorkerCreated(params.url,
                     params.name,
                     filter->render_process_id(),
                     worker_route_id));
 }
 
 void SharedWorkerServiceImpl::ForwardToWorker(
     const IPC::Message& message,
     SharedWorkerMessageFilter* filter) {
-  // TODO(horo): implement this.
-  NOTIMPLEMENTED();
+  for (WorkerHostMap::const_iterator iter = worker_hosts_.begin();
+       iter != worker_hosts_.end();
+       ++iter) {
+    if (iter->second->FilterMessage(message, filter))
+      return;
+  }
 }
 
 void SharedWorkerServiceImpl::DocumentDetached(
     unsigned long long document_id,
     SharedWorkerMessageFilter* filter) {
-  // TODO(horo): implement this.
-  NOTIMPLEMENTED();
+  for (WorkerHostMap::const_iterator iter = worker_hosts_.begin();
+       iter != worker_hosts_.end();
+       ++iter) {
+    iter->second->DocumentDetached(filter, document_id);
+  }
 }
 
 void SharedWorkerServiceImpl::WorkerContextClosed(
     int worker_route_id,
     SharedWorkerMessageFilter* filter) {
-  // TODO(horo): implement this.
-  NOTIMPLEMENTED();
+  if (SharedWorkerHost* host = FindSharedWorkerHost(filter, worker_route_id))
+    host->WorkerContextClosed();
 }
 
 void SharedWorkerServiceImpl::WorkerContextDestroyed(
     int worker_route_id,
     SharedWorkerMessageFilter* filter) {
-  // TODO(horo): implement this.
-  NOTIMPLEMENTED();
+  scoped_ptr<SharedWorkerHost> host =
+      worker_hosts_.take_and_erase(std::make_pair(filter->render_process_id(),
+                                                  worker_route_id));
+  if (!host)
+    return;
+  host->WorkerContextDestroyed();
 }
 
 void SharedWorkerServiceImpl::WorkerScriptLoaded(
     int worker_route_id,
     SharedWorkerMessageFilter* filter) {
-  // TODO(horo): implement this.
-  NOTIMPLEMENTED();
+  if (SharedWorkerHost* host = FindSharedWorkerHost(filter, worker_route_id))
+    host->WorkerScriptLoaded();
 }
 
 void SharedWorkerServiceImpl::WorkerScriptLoadFailed(
     int worker_route_id,
     SharedWorkerMessageFilter* filter) {
-  // TODO(horo): implement this.
-  NOTIMPLEMENTED();
+  scoped_ptr<SharedWorkerHost> host =
+      worker_hosts_.take_and_erase(std::make_pair(filter->render_process_id(),
+                                                  worker_route_id));
+  if (!host)

[kinuko, 2014/02/28 06:52:20]

if (!host || !host->instance()) ?

[horo, 2014/02/28 07:16:16]

Added null check of instance in SharedWorkerHost.

+    return;
+  host->WorkerScriptLoadFailed();
 }
 
 void SharedWorkerServiceImpl::WorkerConnected(
     int message_port_id,
     int worker_route_id,
     SharedWorkerMessageFilter* filter) {
-  // TODO(horo): implement this.
-  NOTIMPLEMENTED();
+  if (SharedWorkerHost* host = FindSharedWorkerHost(filter, worker_route_id))
+    host->WorkerConnected(message_port_id);
 }
 
 void SharedWorkerServiceImpl::AllowDatabase(
     int worker_route_id,
     const GURL& url,
     const base::string16& name,
     const base::string16& display_name,
     unsigned long estimated_size,
     bool* result,
     SharedWorkerMessageFilter* filter) {
-  // TODO(horo): implement this.
-  NOTIMPLEMENTED();
+  if (SharedWorkerHost* host = FindSharedWorkerHost(filter, worker_route_id))
+    host->AllowDatabase(url, name, display_name, estimated_size, result);
 }
 
 void SharedWorkerServiceImpl::AllowFileSystem(
     int worker_route_id,
     const GURL& url,
     bool* result,
     SharedWorkerMessageFilter* filter) {
-  // TODO(horo): implement this.
-  NOTIMPLEMENTED();
+  if (SharedWorkerHost* host = FindSharedWorkerHost(filter, worker_route_id))
+    host->AllowFileSystem(url, result);
 }
 
 void SharedWorkerServiceImpl::AllowIndexedDB(
     int worker_route_id,
     const GURL& url,
     const base::string16& name,
     bool* result,
     SharedWorkerMessageFilter* filter) {
-  // TODO(horo): implement this.
-  NOTIMPLEMENTED();
+  if (SharedWorkerHost* host = FindSharedWorkerHost(filter, worker_route_id))
+    host->AllowIndexedDB(url, name, result);
 }
 
 void SharedWorkerServiceImpl::OnSharedWorkerMessageFilterClosing(
     SharedWorkerMessageFilter* filter) {
-  // TODO(horo): implement this.
-  NOTIMPLEMENTED();
+  std::vector<ProcessRouteIdPair> remove_list;
+  for (WorkerHostMap::iterator iter = worker_hosts_.begin();
+       iter != worker_hosts_.end();
+       ++iter) {
+    iter->second->FilterShutdown(filter);
+    if (iter->first.first == filter->render_process_id())
+      remove_list.push_back(iter->first);
+  }
+  for (size_t i = 0; i < remove_list.size(); ++i)
+    worker_hosts_.erase(remove_list[i]);
+}
+
+SharedWorkerHost* SharedWorkerServiceImpl::FindSharedWorkerHost(
+    SharedWorkerMessageFilter* filter,
+    int worker_route_id) {
+  return worker_hosts_.get(std::make_pair(filter->render_process_id(),
+                                          worker_route_id));

[kinuko, 2014/02/28 06:52:20]

In the previous implementation we also used to check if (*iter)->instance() is
non-null. Is it safe to remove the condition?

[horo, 2014/02/28 07:16:16]

Added null check of instance in SharedWorkerHost.

 }
 
 SharedWorkerInstance* SharedWorkerServiceImpl::FindSharedWorkerInstance(
     const GURL& url,
     const base::string16& name,
     const WorkerStoragePartition& partition,
     ResourceContext* resource_context) {
-  for (ScopedVector<SharedWorkerHost>::const_iterator iter =
-           worker_hosts_.begin();
+  for (WorkerHostMap::const_iterator iter = worker_hosts_.begin();
        iter != worker_hosts_.end();
        ++iter) {
-    SharedWorkerInstance* instance = (*iter)->instance();
-    if (instance &&
-        instance->Matches(url, name, partition, resource_context))
+    SharedWorkerInstance* instance = iter->second->instance();
+    if (instance && instance->Matches(url, name, partition, resource_context))
       return instance;
   }
   return NULL;
 }
 
 }  // namespace content